Fix passing Component to StreamManager

0a4acd12c3, which fixes #160, introduced
a regression as it assumed only Client may implement StreamClient, and
passing a component triggers an unconditional "client is not
disconnected" error.

stanza/commands.go

@@ -23,7 +23,7 @@ const (
 type Command struct {
 	XMLName xml.Name `xml:"http://jabber.org/protocol/commands command"`
 
-	CommandElements []CommandElement
+	CommandElement CommandElement
 
 	BadAction       *struct{} `xml:"bad-action,omitempty"`
 	BadLocale       *struct{} `xml:"bad-locale,omitempty"`
@@ -56,8 +56,6 @@ type CommandElement interface {
 }
 
 type Actions struct {
-	XMLName xml.Name `xml:"actions"`
-
 	Prev     *struct{} `xml:"prev,omitempty"`
 	Next     *struct{} `xml:"next,omitempty"`
 	Complete *struct{} `xml:"complete,omitempty"`
@@ -70,8 +68,6 @@ func (a *Actions) Ref() string {
 }
 
 type Note struct {
-	XMLName xml.Name `xml:"note"`
-
 	Text string `xml:",cdata"`
 	Type string `xml:"type,attr,omitempty"`
 }
@@ -121,22 +117,22 @@ func (c *Command) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
 			var err error
 			switch tt.Name.Local {
 
-			case "actions":
+			case "affiliations":
 				a := Actions{}
 				err = d.DecodeElement(&a, &tt)
-				c.CommandElements = append(c.CommandElements, &a)
-			case "note":
+				c.CommandElement = &a
+			case "configure":
 				nt := Note{}
 				err = d.DecodeElement(&nt, &tt)
-				c.CommandElements = append(c.CommandElements, &nt)
+				c.CommandElement = &nt
 			case "x":
 				f := Form{}
 				err = d.DecodeElement(&f, &tt)
-				c.CommandElements = append(c.CommandElements, &f)
+				c.CommandElement = &f
 			default:
 				n := Node{}
 				err = d.DecodeElement(&n, &tt)
-				c.CommandElements = append(c.CommandElements, &n)
+				c.CommandElement = &n
 				if err != nil {
 					return err
 				}

stanza/pubsub_owner.go

@@ -237,10 +237,10 @@ func NewApprovePendingSubRequest(serviceId, sessionId, nodeId string) (*IQ, erro
 	}
 	iq.Payload = &Command{
 		//  the command name ('node' attribute of the command element) MUST have a value of "http://jabber.org/protocol/pubsub#get-pending"
-		Node:            "http://jabber.org/protocol/pubsub#get-pending",
-		Action:          CommandActionExecute,
-		SessionId:       sessionId,
-		CommandElements: []CommandElement{&n},
+		Node:           "http://jabber.org/protocol/pubsub#get-pending",
+		Action:         CommandActionExecute,
+		SessionId:      sessionId,
+		CommandElement: &n,
 	}
 	return iq, nil
 }
@@ -353,18 +353,11 @@ func (iq *IQ) GetFormFields() (map[string]*Field, error) {
 
 	case *Command:
 		fieldMap := make(map[string]*Field)
-		var form *Form
-		for _, ce := range payload.CommandElements {
-			fo, ok := ce.(*Form)
-			if ok {
-				form = fo
-				break
-			}
-		}
-		if form == nil {
+		co, ok := payload.CommandElement.(*Form)
+		if !ok {
 			return nil, errors.New("this IQ does not contain a command payload with a form")
 		}
-		for _, elt := range form.Fields {
+		for _, elt := range co.Fields {
 			fieldMap[elt.Var] = elt
 		}
 		return fieldMap, nil

stream_manager.go

@@ -114,18 +114,23 @@ func (sm *StreamManager) Stop() {
 
 func (sm *StreamManager) connect() error {
 	if sm.client != nil {
-		if c, ok := sm.client.(*Client); ok {
-			if c.CurrentState.getState() == StateDisconnected {
-				sm.Metrics = initMetrics()
-				err := c.Connect()
-				if err != nil {
-					return err
-				}
-				if sm.PostConnect != nil {
-					sm.PostConnect(sm.client)
-				}
-				return nil
+		var scs *SyncConnState
+		if client, ok := sm.client.(*Client); ok {
+			scs = &client.CurrentState
+		}
+		if component, ok := sm.client.(*Component); ok {
+			scs = &component.CurrentState
+		}
+		if scs != nil && scs.getState() == StateDisconnected {
+			sm.Metrics = initMetrics()
+			err := sm.client.Connect()
+			if err != nil {
+				return err
 			}
+			if sm.PostConnect != nil {
+				sm.PostConnect(sm.client)
+			}
+			return nil
 		}
 	}
 	return errors.New("client is not disconnected")