Add convenience functions

gometal · Jul 8, 2014 · ba63c57 · ba63c57
1 parent 3f3444f
commit ba63c57
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/gobot.go b/gobot.go
@@ -58,6 +58,10 @@ func (g *Gobot) Robots() *robots {
 	return g.robots
 }
 
+func (g *Gobot) AddRobot(r *Robot) *Robot {
+	return g.Robots().Add(r)
+}
+
 func (g *Gobot) Robot(name string) *Robot {
 	for _, robot := range g.Robots().robots {
 		if robot.Name == name {

diff --git a/robot.go b/robot.go
@@ -113,6 +113,10 @@ func (r *Robot) Devices() *devices {
 	return r.devices
 }
 
+func (r *Robot) AddDevice(d Device) Device {
+	return r.Devices().Add(d)
+}
+
 func (r *Robot) Device(name string) Device {
 	if r == nil {
 		return nil
@@ -129,6 +133,10 @@ func (r *Robot) Connections() *connections {
 	return r.connections
 }
 
+func (r *Robot) AddConnection(c Connection) Connection {
+	return r.Connections().Add(c)
+}
+
 func (r *Robot) Connection(name string) Connection {
 	if r == nil {
 		return nil