- support adding cameras via interface

- support qr code auth
- support resolution change
- support h265
- refactor code

internal/tuya/tuya.go

@@ -1,13 +1,214 @@
 package tuya
 
 import (
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+	"net/url"
+	"strings"
+	"time"
+
+	"github.com/AlexxIT/go2rtc/internal/api"
 	"github.com/AlexxIT/go2rtc/internal/streams"
 	"github.com/AlexxIT/go2rtc/pkg/core"
 	"github.com/AlexxIT/go2rtc/pkg/tuya"
 )
 
+var users = make(map[string]tuya.LoginResponse)
+
 func Init() {
 	streams.HandleFunc("tuya", func(source string) (core.Producer, error) {
 		return tuya.Dial(source)
 	})
+
+	api.HandleFunc("api/tuya", apiTuya)
+}
+
+func apiTuya(w http.ResponseWriter, r *http.Request) {
+	query := r.URL.Query()
+	userCode := query.Get("user_code")
+	token := query.Get("token")
+
+	if userCode == "" {
+		http.Error(w, "user_code is required", http.StatusBadRequest)
+		return
+	}
+
+	var auth *tuya.LoginResponse
+	if loginResponse, ok := users[userCode]; ok {
+		expireTime := loginResponse.Timestamp + loginResponse.Result.ExpireTime
+
+		if expireTime > time.Now().Unix() {
+			auth = &loginResponse
+		} else {
+			delete(users, userCode)
+			token = ""
+		}
+	}
+
+	if auth == nil && token == "" {
+		qrCode, err := getQRCode(userCode)
+		if err != nil {
+			http.Error(w, err.Error(), http.StatusInternalServerError)
+			return
+		}
+
+		// response qrCode
+		json.NewEncoder(w).Encode(map[string]interface{}{
+			"qrCode": qrCode,
+		})
+
+		return
+	}
+
+	if auth == nil && token != "" {
+		authResponse, err := login(userCode, token)
+		if err != nil {
+			http.Error(w, err.Error(), http.StatusInternalServerError)
+			return
+		}
+
+		auth = authResponse
+	}
+
+	if auth == nil {
+		http.Error(w, "failed to get auth", http.StatusInternalServerError)
+		return
+	}
+
+	tokenInfo := tuya.TokenInfo{
+		AccessToken:  auth.Result.AccessToken,
+		ExpireTime:   auth.Timestamp + auth.Result.ExpireTime,
+		RefreshToken: auth.Result.RefreshToken,
+	}
+
+	tokenInfoBase64, err := tuya.ToBase64(&tokenInfo)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	tuyaAPI, err := tuya.NewTuyaOpenApiClient(
+		strings.Replace(auth.Result.Endpoint, "https://", "", 1),
+		auth.Result.UID,
+		"",
+		auth.Result.TerminalID,
+		tokenInfo,
+		"",
+	)
+
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	devices, err := tuyaAPI.GetAllDevices()
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	var items []*api.Source
+	for _, device := range devices {
+		cleanQuery := url.Values{}
+		cleanQuery.Set("uid", auth.Result.UID)
+		cleanQuery.Set("token", tokenInfoBase64)
+		cleanQuery.Set("terminal_id", auth.Result.TerminalID)
+		cleanQuery.Set("device_id", device.ID)
+
+		endpoint := strings.Replace(auth.Result.Endpoint, "https://", "tuya://", 1)
+		url := fmt.Sprintf("%s?%s", endpoint, cleanQuery.Encode())
+
+		items = append(items, &api.Source{
+			Name: device.Name,
+			URL:  url,
+		})
+	}
+
+	api.ResponseSources(w, items)
+}
+
+func login(userCode string, qrCode string) (*tuya.LoginResponse, error) {
+	url := fmt.Sprintf("https://%s/v1.0/m/life/home-assistant/qrcode/tokens/%s?clientid=%s&usercode=%s", tuya.TUYA_HOST, qrCode, tuya.TUYA_CLIENT_ID, userCode)
+
+	req, err := http.NewRequest("GET", url, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	httpClient := &http.Client{
+		Timeout: 10 * time.Second,
+	}
+
+	response, err := httpClient.Do(req)
+	if err != nil {
+		return nil, err
+	}
+	defer response.Body.Close()
+
+	res, err := io.ReadAll(response.Body)
+	if err != nil {
+		return nil, err
+	}
+
+	if response.StatusCode != http.StatusOK {
+		return nil, fmt.Errorf("failed to get QR code: %s", string(res))
+	}
+
+	var loginResponse tuya.LoginResponse
+	err = json.Unmarshal(res, &loginResponse)
+	if err != nil {
+		return nil, err
+	}
+
+	if !loginResponse.Success {
+		return nil, fmt.Errorf("failed to login: %s", loginResponse.Msg)
+	}
+
+	users[userCode] = loginResponse
+
+	return &loginResponse, nil
+}
+
+func getQRCode(userCode string) (string, error) {
+	url := fmt.Sprintf("https://%s/v1.0/m/life/home-assistant/qrcode/tokens?clientid=%s&schema=%s&usercode=%s", tuya.TUYA_HOST, tuya.TUYA_CLIENT_ID, tuya.TUYA_SCHEMA, userCode)
+
+	req, err := http.NewRequest("POST", url, nil)
+	if err != nil {
+		return "", err
+	}
+
+	req.Header.Set("Content-Type", "text/plain")
+
+	httpClient := &http.Client{
+		Timeout: 10 * time.Second,
+	}
+
+	response, err := httpClient.Do(req)
+	if err != nil {
+		return "", err
+	}
+	defer response.Body.Close()
+
+	res, err := io.ReadAll(response.Body)
+	if err != nil {
+		return "", err
+	}
+
+	if response.StatusCode != http.StatusOK {
+		return "", err
+	}
+
+	var qrResponse tuya.QRResponse
+	err = json.Unmarshal(res, &qrResponse)
+	if err != nil {
+		return "", err
+	}
+
+	if !qrResponse.Success {
+		return "", fmt.Errorf("failed to get QR code: %s", qrResponse.Msg)
+	}
+
+	return qrResponse.Result.Code, nil
 }