[Hub] fetch project id by key

Signed-off-by: Vishal Rana <[email protected]>

Author: vishr

hub/hub.go

@@ -2,54 +2,80 @@ package hub
 
 import (
 	"fmt"
+	"os"
 	"strings"
 	"time"
 
-	mqtt "github.com/eclipse/paho.mqtt.golang"
+	"github.com/labstack/gommon/log"
+
+	"github.com/eclipse/paho.mqtt.golang"
+	"github.com/go-resty/resty"
 )
 
 type (
 	Hub struct {
 		Options
-		accountID string
-		apiKey    string
-		deviceID  string
-		client    mqtt.Client
+		resty  *resty.Client
+		key    *Key
+		client mqtt.Client
+		logger *log.Logger
 	}
 
 	Options struct {
+		DeviceID       string
 		MessageHandler MessageHandler
 	}
 
+	Key struct {
+		ID        string `json:"id"`
+		ProjectID string `json:"project_id"`
+	}
+
 	ConnectHandler func()
 
 	MessageHandler func(topic string, message []byte)
 )
 
-func New(accountID, apiKey, deviceID string) *Hub {
-	return NewWithOptions(accountID, apiKey, deviceID, Options{})
+func New(apiKey string) *Hub {
+	return NewWithOptions(apiKey, Options{})
 }
 
-func NewWithOptions(accountID, apiKey, deviceID string, options Options) *Hub {
-	h := &Hub{
-		accountID: accountID,
-		apiKey:    apiKey,
+func NewWithOptions(apiKey string, options Options) (h *Hub) {
+	h = &Hub{
+		key: &Key{
+			ID: apiKey,
+		},
+		resty:  resty.New().SetHostURL("https://api.labstack.com").SetAuthToken(apiKey),
+		logger: log.New("hub"),
 	}
-	h.deviceID = h.normalizeDeviceID(deviceID)
 	h.Options = options
-	return h
+	if h.DeviceID == "" {
+		h.DeviceID, _ = os.Hostname()
+	}
+	res, err := h.resty.R().
+		SetResult(h.key).
+		Get("/keys/" + h.key.ID)
+	if err != nil {
+		h.logger.Fatal(err)
+	}
+	if res.StatusCode() < 200 || res.StatusCode() >= 300 {
+		h.logger.Fatal(err)
+	}
+	h.DeviceID = h.normalizeDeviceID(h.DeviceID)
+	return
 }
 
 func (h *Hub) normalizeDeviceID(id string) string {
-	return fmt.Sprintf("%s:%s", h.accountID, id)
+	return fmt.Sprintf("%s:%s", h.key.ProjectID, id)
+
 }
 
 func (h *Hub) normalizeTopic(name string) string {
-	return fmt.Sprintf("%s/%s", h.accountID, name)
+	return fmt.Sprintf("%s/%s", h.key.ProjectID, name)
 }
 
 func (h *Hub) denormalizeTopic(name string) string {
-	return strings.TrimPrefix(name, h.accountID+"/")
+	return strings.TrimPrefix(name, h.key.ProjectID+"/")
 }
 
 func (h *Hub) Connect() error {
@@ -59,9 +85,9 @@ func (h *Hub) Connect() error {
 func (h *Hub) ConnectWithHandler(handler ConnectHandler) error {
 	o := mqtt.NewClientOptions().
 		AddBroker("tcp://hub.labstack.com:1883").
-		SetUsername(h.accountID).
-		SetPassword(h.apiKey).
-		SetClientID(h.deviceID)
+		SetUsername(h.key.ProjectID).
+		SetPassword(h.key.ID).
+		SetClientID(h.DeviceID)
 	if handler != nil {
 		o.OnConnect = func(_ mqtt.Client) {
 			handler()

labstack.go

@@ -2,4 +2,9 @@ package labstack
 
 type (
 	Map map[string]interface{}
+
+	APIError struct {
+		Code    int    `json:"code"`
+		Message string `json:"message"`
+	}
 )