增加命令功能

cmd/saastool/read.go

@@ -1,5 +1,95 @@
 package main
 
+import (
+	"flag"
+	"fmt"
+	"log/slog"
+	"net/http"
+	"strings"
+
+	"e.coding.net/rta/public/saasapi"
+	"e.coding.net/rta/public/saasapi/pkg/saashttp"
+	"google.golang.org/protobuf/encoding/protojson"
+)
+
+const (
+	getIdsMax = 100
+)
+
+type readParams struct {
+	cfg      *Config
+	appid    string
+	userids  []string
+	saasHttp *saashttp.SaasClient
+}
+
 func RunRead(args ...string) error {
+	fs := flag.NewFlagSet("read", flag.ExitOnError)
+	cfgFile := paramConfig(fs)
+	appid := paramAppid(fs)
+	userids := paramUserids(fs)
+
+	if err := fs.Parse(args); err != nil {
+		fmt.Println("command line parse error", "err", err)
+		return err
+	}
+
+	// 切割字符串
+	idsSlice := strings.Split(*userids, ",")
+
+	if fs.NArg() > 0 || len(idsSlice) == 0 || (len(idsSlice) == 1 && idsSlice[0] == "") || len(idsSlice) > getIdsMax {
+		fs.PrintDefaults()
+		return nil
+	}
+
+	cfg, err := LoadConfigFile(*cfgFile)
+	if err != nil {
+		slog.Error("LoadConfigFile error", "err", err)
+		return err
+	}
+
+	readParams := readParams{
+		cfg:     cfg,
+		userids: idsSlice,
+		appid:   *appid,
+		saasHttp: &saashttp.SaasClient{
+			Client:  http.Client{},
+			ApiUrls: cfg.ApiUrls,
+			Auth:    cfg.Auth,
+		},
+	}
+
+	return doRead(readParams)
+}
+
+func doRead(readParams readParams) error {
+	saasReq := &saasapi.SaasReq{
+		Cmd: &saasapi.SaasReq_Read{
+			Read: &saasapi.Read{},
+		},
+	}
+
+	if readParams.appid != "" {
+		saasReq.UseridType = saasapi.UserIdType_OPENID
+		saasReq.Appid = readParams.appid
+	}
+	saasReadItems := []*saasapi.ReadItem{}
+	for _, userid := range readParams.userids {
+		saasReadItems = append(saasReadItems, &saasapi.ReadItem{
+			Userid: userid,
+		})
+	}
+
+	saasReq.Cmd.(*saasapi.SaasReq_Read).Read.ReadItems = saasReadItems
+
+	res, err := readParams.saasHttp.Read(saasReq)
+
+	if err != nil {
+		slog.Error("submitRead error", "err", err)
+		return err
+	}
+
+	fmt.Println(protojson.Format(res))
+
 	return nil
 }