hybridgroup.gobot/connection.go

package gobot

import (
	"errors"
	"fmt"
	"log"
	"reflect"
)

type Connection interface {
	Connect() bool
	Finalize() bool
	port() string
	name() string
	setName(string)
	params() map[string]interface{}
}

type JSONConnection struct {
	Name    string `json:"name"`
	Port    string `json:"port"`
	Adaptor string `json:"adaptor"`
}

type connection struct {
	Name    string
	Type    string
	Adaptor AdaptorInterface
	Robot   *Robot
}

type connections []*connection

// Start() starts all the connections.
func (c connections) Start() error {
	var err error
	log.Println("Starting connections...")
	for _, connection := range c {
		log.Println("Starting connection " + connection.Name + "...")
		if connection.Connect() == false {
			err = errors.New("Could not start connection")
			break
		}
	}
	return err
}

// Filanize() finalizes all the connections.
func (c connections) Finalize() {
	for _, connection := range c {
		connection.Finalize()
	}
}

func NewConnection(adaptor AdaptorInterface, r *Robot) *connection {
	if adaptor.name() == "" {
		adaptor.setName(fmt.Sprintf("%X", Rand(int(^uint(0)>>1))))
	}
	t := reflect.ValueOf(adaptor).Type().String()
	return &connection{
		Type:    t[1:len(t)],
		Name:    adaptor.name(),
		Robot:   r,
		Adaptor: adaptor,
	}
}

func (c *connection) Connect() bool {
	log.Println("Connecting to " + c.Name + " on port " + c.port() + "...")
	return c.Adaptor.Connect()
}

func (c *connection) Finalize() bool {
	log.Println("Finalizing " + c.Name + "...")
	return c.Adaptor.Finalize()
}

func (c *connection) ToJSON() *JSONConnection {
	return &JSONConnection{
		Name:    c.Name,
		Port:    c.port(),
		Adaptor: c.Type,
	}
}

func (c *connection) port() string {
	return c.Adaptor.port()
}

func (c *connection) name() string {
	return c.Name
}

func (c *connection) setName(s string) {
	c.Name = s
}

func (c *connection) params() map[string]interface{} {
	return c.Adaptor.params()
}