fix: improve ai table file handling

internal/httpapi/ai_export_oss.go

@@ -8,7 +8,10 @@ import (
 	"encoding/base64"
 	"encoding/binary"
 	"fmt"
+	"io"
 	"net/http"
+	"net/url"
+	"os"
 	"path"
 	"strconv"
 	"strings"
@@ -88,6 +91,98 @@ func (a *API) uploadAITableAnalysisExport(ctx context.Context, r *http.Request,
 	}, nil
 }
 
+func downloadWorkbenchFileToTemp(r *http.Request, fileID, fileType string) (string, func(), error) {
+	fileID = strings.TrimSpace(fileID)
+	if fileID == "" {
+		return "", func() {}, fmt.Errorf("empty file id")
+	}
+	req, err := http.NewRequestWithContext(r.Context(), http.MethodGet, absoluteURL(r, "/v0/file/s/"+url.PathEscape(fileID)), nil)
+	if err != nil {
+		return "", func() {}, err
+	}
+	copyHeader(req, r, "Authorization")
+	copyHeader(req, r, "X-MeetingIM-APIKey")
+	copyHeader(req, r, "X-Tenant-Code")
+	copyHeader(req, r, "X-IM-Schema")
+	copyHeader(req, r, "X-Request-ID")
+	if auth := strings.TrimSpace(r.Header.Get("X-MeetingIM-Auth")); auth != "" {
+		req.Header.Set("X-MeetingIM-Auth", auth)
+	} else if token := bearerTokenFromHeader(r.Header.Get("Authorization")); token != "" {
+		req.Header.Set("X-MeetingIM-Auth", "token "+token)
+	}
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return "", func() {}, err
+	}
+	defer resp.Body.Close()
+	if resp.StatusCode < 200 || resp.StatusCode >= 300 {
+		return "", func() {}, fmt.Errorf("file service returned %d", resp.StatusCode)
+	}
+	return responseBodyToTemp(resp.Body, fileType)
+}
+
+func (a *API) downloadOSSObjectToTemp(ctx context.Context, objectKey, fileType string) (string, func(), error) {
+	objectKey = strings.Trim(strings.TrimSpace(objectKey), "/")
+	if objectKey == "" {
+		return "", func() {}, fmt.Errorf("empty object key")
+	}
+	cfg, err := a.ossUploadConfig(ctx)
+	if err != nil {
+		return "", func() {}, err
+	}
+	client, err := aliyunoss.New(cfg.Endpoint, cfg.AccessKeyID, cfg.AccessKeySecret)
+	if err != nil {
+		return "", func() {}, err
+	}
+	bucket, err := client.Bucket(cfg.Bucket)
+	if err != nil {
+		return "", func() {}, err
+	}
+	reader, err := bucket.GetObject(objectKey)
+	if err != nil {
+		return "", func() {}, err
+	}
+	defer reader.Close()
+	return responseBodyToTemp(reader, fileType)
+}
+
+func responseBodyToTemp(reader io.Reader, fileType string) (string, func(), error) {
+	suffix := "." + strings.TrimLeft(strings.ToLower(strings.TrimSpace(fileType)), ".")
+	if suffix == "." {
+		suffix = ".table"
+	}
+	tmp, err := os.CreateTemp("", "im-workbench-ai-table-*"+suffix)
+	if err != nil {
+		return "", func() {}, err
+	}
+	tmpPath := tmp.Name()
+	cleanup := func() { _ = os.Remove(tmpPath) }
+	if _, err := io.Copy(tmp, reader); err != nil {
+		_ = tmp.Close()
+		cleanup()
+		return "", func() {}, err
+	}
+	if err := tmp.Close(); err != nil {
+		cleanup()
+		return "", func() {}, err
+	}
+	return tmpPath, cleanup, nil
+}
+
+func copyHeader(dst *http.Request, src *http.Request, key string) {
+	if value := strings.TrimSpace(src.Header.Get(key)); value != "" {
+		dst.Header.Set(key, value)
+	}
+}
+
+func bearerTokenFromHeader(value string) string {
+	token, ok := strings.CutPrefix(strings.TrimSpace(value), "Bearer ")
+	if !ok {
+		return ""
+	}
+	return strings.TrimSpace(token)
+}
+
 func (a *API) ossUploadConfig(ctx context.Context) (ossUploadConfig, error) {
 	values, err := managerconfig.New(a.manager).ThirdPartyConfig(ctx, "im_third_party",
 		"oss_enabled",