package rpc

import (
	"github.com/mitchellh/packer/packer"
	"net/rpc"
)

// A ClientCommand is an implementation of the Command interface where the
// command is actually executed over an RPC connection.
type ClientCommand struct {
	client *rpc.Client
}

// A ServerCommand wraps a Command and makes it exportable as part
// of a Golang RPC server.
type ServerCommand struct {
	command packer.Command
}

type CommandRunArgs struct {
	Env packer.Environment
	Args []string
}

type CommandSynopsisArgs byte

func (c *ClientCommand) Run(env packer.Environment, args []string) (result int) {
	// TODO: Environment
	rpcArgs := &CommandRunArgs{nil, args}
	c.client.Call("Command.Run", rpcArgs, &result)
	return
}

func (c *ClientCommand) Synopsis() (result string) {
	c.client.Call("Command.Synopsis", CommandSynopsisArgs(0), &result)
	return
}

func (c *ServerCommand) Run(args *CommandRunArgs, reply *int) error {
	*reply = c.command.Run(args.Env, args.Args)
	return nil
}

func (c *ServerCommand) Synopsis(args *CommandSynopsisArgs, reply *string) error {
	*reply = c.command.Synopsis()
	return nil
}