Add support Xiaomi Home source

Author: AlexxIT

internal/accounts.go

@@ -11,6 +11,16 @@ import (
 	"github.com/AlexxIT/SmartScaleConnect/pkg/zepp"
 )
 
+const (
+	AccGarmin     = "garmin"
+	AccMiFitness  = "mifitness"
+	AccPicooc     = "picooc"
+	AccTanita     = "tanita"
+	AccXiaomi     = "xiaomi"
+	AccXiaomiHome = "xiaomihome"
+	AccZeppXiaomi = "zepp/xiaomi"
+)
+
 var accounts = map[string]core.Account{}
 
 func GetAccount(fields []string) (core.Account, error) {
@@ -33,15 +43,17 @@ func getAccount(fields []string, key string) (core.Account, error) {
 	var acc core.Account
 
 	switch fields[0] {
-	case "garmin":
+	case AccGarmin:
 		acc = garmin.NewClient()
-	case "picooc":
+	case AccPicooc:
 		acc = picooc.NewClient()
-	case "tanita":
+	case AccTanita:
 		acc = tanita.NewClient()
-	case "xiaomi":
+	case AccXiaomi, AccMiFitness:
 		acc = xiaomi.NewClient(xiaomi.AppMiFitness)
-	case "zepp/xiaomi":
+	case AccXiaomiHome:
+		acc = xiaomi.NewClient(xiaomi.AppXiaomiHome)
+	case AccZeppXiaomi:
 		acc = zepp.NewClient()
 	default:
 		return nil, errors.New("unsupported type: " + fields[0])

internal/tokens.go

@@ -3,11 +3,14 @@ package internal
 import (
 	"encoding/json"
 	"os"
+	"strings"
 )
 
 var tokens = map[string]string{}
 
 func LoadToken(key string) string {
+	key = replaceKey(key)
+
 	if len(tokens) == 0 {
 		f, err := os.Open("scaleconnect.json")
 		if err != nil {
@@ -22,6 +25,8 @@ func LoadToken(key string) string {
 }
 
 func SaveToken(key string, value string) {
+	key = replaceKey(key)
+
 	tokens[key] = value
 
 	f, err := os.Create("scaleconnect.json")
@@ -32,3 +37,12 @@ func SaveToken(key string, value string) {
 
 	_ = json.NewEncoder(f).Encode(&tokens)
 }
+
+func replaceKey(key string) string {
+	key, value, _ := strings.Cut(key, ":")
+	switch key {
+	case AccMiFitness, AccXiaomiHome:
+		return AccXiaomi + ":" + value
+	}
+	return key
+}

internal/weights.go

@@ -15,6 +15,7 @@ import (
 	"github.com/AlexxIT/SmartScaleConnect/pkg/core"
 	"github.com/AlexxIT/SmartScaleConnect/pkg/csv"
 	"github.com/AlexxIT/SmartScaleConnect/pkg/fitbit"
+	"github.com/AlexxIT/SmartScaleConnect/pkg/xiaomi"
 )
 
 func GetWeights(from any) ([]*core.Weight, error) {
@@ -80,14 +81,14 @@ func getWeights(config string) ([]*core.Weight, error) {
 	case "fitbit":
 		return fitbit.Read(fields[1])
 
-	case "garmin", "tanita":
+	case AccGarmin, AccTanita:
 		acc, err := GetAccount(fields)
 		if err != nil {
 			return nil, err
 		}
 		return acc.GetAllWeights()
 
-	case "picooc", "xiaomi", "zepp/xiaomi":
+	case AccMiFitness, AccPicooc, AccXiaomi, AccZeppXiaomi:
 		acc, err := GetAccount(fields)
 		if err != nil {
 			return nil, err
@@ -99,6 +100,14 @@ func getWeights(config string) ([]*core.Weight, error) {
 
 		return acc.(core.AccountWithFilter).GetFilterWeights(fields[3])
 
+	case AccXiaomiHome:
+		acc, err := GetAccount(fields)
+		if err != nil {
+			return nil, err
+		}
+
+		return acc.(*xiaomi.Client).GetModelWeights(fields[3], fields[4])
+
 	default:
 		return nil, errors.New("unsupported type: " + fields[0])
 	}
@@ -109,7 +118,7 @@ func SetWeights(config string, src []*core.Weight) error {
 	case "csv", "json":
 		return writeFile(config, src)
 
-	case "garmin", "zepp/xiaomi":
+	case AccGarmin, AccZeppXiaomi:
 		return appendAccount(config, src)
 
 	case "json/latest":

pkg/xiaomi/auth.go

@@ -271,7 +271,7 @@ func (c *Client) Request(baseURL, apiURL, params string) ([]byte, error) {
 	}
 	defer res.Body.Close()
 
-	if res.StatusCode == http.StatusUnauthorized {
+	if res.StatusCode != http.StatusOK {
 		return nil, errors.New(res.Status)
 	}