package main

import (
	"context"
	"fmt"
	"os"

	"ai_script/common_util/feishu_util"
)

func main() {
	client, err := feishu_util.DefaultClient()
	if err != nil {
		fail("加载配置失败", err)
	}

	url := feishu_util.DefaultOptimizationIssueSheetURL
	title := feishu_util.DefaultOptimizationIssueSheetTitle
	if len(os.Args) > 1 {
		url = os.Args[1]
	}
	if len(os.Args) > 2 {
		title = os.Args[2]
	}

	ctx := context.Background()
	meta, token, err := client.CheckReadPermission(ctx, url, title)
	if err != nil {
		fail("读取权限校验失败", err)
	}
	fmt.Printf("OK: 具备读取权限\n")
	fmt.Printf("  spreadsheet_token: %s\n", token)
	fmt.Printf("  sheet_id: %s\n", meta.SheetID)
	fmt.Printf("  sheet_title: %s\n", meta.Title)

	values, err := client.ReadSpreadsheetURL(ctx, url, title)
	if err != nil {
		fail("读取表格失败", err)
	}
	fmt.Printf("  rows: %d\n", len(values.Values))
	if len(values.Values) > 0 {
		fmt.Printf("  first_row_cols: %d\n", len(values.Values[0]))
	}
}

func fail(msg string, err error) {
	fmt.Fprintf(os.Stderr, "%s: %v\n", msg, err)
	os.Exit(1)
}