Support more data from webrtc stats

icetransportstate.go

@@ -49,22 +49,53 @@
 	ICETransportStateClosed
 )
 
+const (
+	iceTransportStateNewStr          = "new"
+	iceTransportStateCheckingStr     = "checking"
+	iceTransportStateConnectedStr    = "connected"
+	iceTransportStateCompletedStr    = "completed"
+	iceTransportStateFailedStr       = "failed"
+	iceTransportStateDisconnectedStr = "disconnected"
+	iceTransportStateClosedStr       = "closed"
+)
+
+func newICETransportState(raw string) ICETransportState {
+	switch raw {
+	case iceTransportStateNewStr:
+		return ICETransportStateNew
+	case iceTransportStateCheckingStr:
+		return ICETransportStateChecking
+	case iceTransportStateConnectedStr:
+		return ICETransportStateConnected
+	case iceTransportStateCompletedStr:
+		return ICETransportStateCompleted
+	case iceTransportStateFailedStr:
+		return ICETransportStateFailed
+	case iceTransportStateDisconnectedStr:
+		return ICETransportStateDisconnected
+	case iceTransportStateClosedStr:
+		return ICETransportStateClosed
+	default:
+		return ICETransportStateUnknown
+	}
+}
+
 func (c ICETransportState) String() string {
 	switch c {
 	case ICETransportStateNew:
-		return "new"
+		return iceTransportStateNewStr
 	case ICETransportStateChecking:
-		return "checking"
+		return iceTransportStateCheckingStr
 	case ICETransportStateConnected:
-		return "connected"
+		return iceTransportStateConnectedStr
 	case ICETransportStateCompleted:
-		return "completed"
+		return iceTransportStateCompletedStr
 	case ICETransportStateFailed:
-		return "failed"
+		return iceTransportStateFailedStr
 	case ICETransportStateDisconnected:
-		return "disconnected"
+		return iceTransportStateDisconnectedStr
 	case ICETransportStateClosed:
-		return "closed"
+		return iceTransportStateClosedStr
 	default:
 		return ErrUnknownType.Error()
 	}
@@ -111,3 +142,14 @@
 		return ice.ConnectionStateUnknown
 	}
 }
+
+// MarshalText implements encoding.TextMarshaler
+func (c ICETransportState) MarshalText() ([]byte, error) {
+	return []byte(c.String()), nil
+}
+
+// UnmarshalText implements encoding.TextUnmarshaler
+func (c *ICETransportState) UnmarshalText(b []byte) error {
+	*c = newICETransportState(string(b))
+	return nil
+}