105 lines
2.6 KiB
Go
105 lines
2.6 KiB
Go
|
package placement
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"bytes"
|
|||
|
|
"context"
|
|||
|
|
"encoding/json"
|
|||
|
|
"fmt"
|
|||
|
|
"os"
|
|||
|
|
"os/exec"
|
|||
|
|
"path/filepath"
|
|||
|
|
"strings"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type resolveMediaInput struct {
|
|||
|
|
Permalink string `json:"permalink"`
|
|||
|
|
StorageState string `json:"storage_state,omitempty"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type resolveMediaOutput struct {
|
|||
|
|
MediaID string `json:"mediaId"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// RunResolveMediaFromPermalink opens a Threads post page via Playwright and extracts numeric media id.
|
|||
|
|
func RunResolveMediaFromPermalink(ctx context.Context, permalink, storageState string) (string, error) {
|
|||
|
|
permalink = strings.TrimSpace(permalink)
|
|||
|
|
if permalink == "" {
|
|||
|
|
return "", fmt.Errorf("permalink is required")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
repoRoot, cliPath, err := resolveMediaCLI()
|
|||
|
|
if err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
payload, err := json.Marshal(resolveMediaInput{
|
|||
|
|
Permalink: permalink,
|
|||
|
|
StorageState: strings.TrimSpace(storageState),
|
|||
|
|
})
|
|||
|
|
if err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
runCtx, cancel := context.WithTimeout(ctx, 90*time.Second)
|
|||
|
|
defer cancel()
|
|||
|
|
|
|||
|
|
cmd := exec.CommandContext(runCtx, "npx", "tsx", cliPath)
|
|||
|
|
cmd.Dir = repoRoot
|
|||
|
|
cmd.Stdin = bytes.NewReader(payload)
|
|||
|
|
var stdout, stderr bytes.Buffer
|
|||
|
|
cmd.Stdout = &stdout
|
|||
|
|
cmd.Stderr = &stderr
|
|||
|
|
if err := cmd.Run(); err != nil {
|
|||
|
|
msg := strings.TrimSpace(stderr.String())
|
|||
|
|
if msg == "" {
|
|||
|
|
msg = err.Error()
|
|||
|
|
}
|
|||
|
|
return "", fmt.Errorf("permalink media resolve failed: %s", msg)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var out resolveMediaOutput
|
|||
|
|
if err := json.Unmarshal(stdout.Bytes(), &out); err != nil {
|
|||
|
|
return "", fmt.Errorf("permalink media resolve output parse failed: %w", err)
|
|||
|
|
}
|
|||
|
|
mediaID := strings.TrimSpace(out.MediaID)
|
|||
|
|
if mediaID == "" {
|
|||
|
|
return "", fmt.Errorf("permalink media resolve returned empty media id")
|
|||
|
|
}
|
|||
|
|
return mediaID, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func resolveMediaCLI() (repoRoot, cliPath string, err error) {
|
|||
|
|
if root := strings.TrimSpace(os.Getenv("HAIXUN_REPO_ROOT")); root != "" {
|
|||
|
|
cli := filepath.Join(root, "haixun-backend", "worker", "threads-resolve-media-cli.ts")
|
|||
|
|
if fileExists(cli) {
|
|||
|
|
return root, cli, nil
|
|||
|
|
}
|
|||
|
|
cli = filepath.Join(root, "worker", "threads-resolve-media-cli.ts")
|
|||
|
|
if fileExists(cli) {
|
|||
|
|
return root, cli, nil
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
cwd, err := os.Getwd()
|
|||
|
|
if err != nil {
|
|||
|
|
return "", "", fmt.Errorf("resolve media cli: %w", err)
|
|||
|
|
}
|
|||
|
|
dir := cwd
|
|||
|
|
for i := 0; i < 6; i++ {
|
|||
|
|
cli := filepath.Join(dir, "haixun-backend", "worker", "threads-resolve-media-cli.ts")
|
|||
|
|
if fileExists(cli) {
|
|||
|
|
return dir, cli, nil
|
|||
|
|
}
|
|||
|
|
cli = filepath.Join(dir, "worker", "threads-resolve-media-cli.ts")
|
|||
|
|
if fileExists(cli) {
|
|||
|
|
return dir, cli, nil
|
|||
|
|
}
|
|||
|
|
parent := filepath.Dir(dir)
|
|||
|
|
if parent == dir {
|
|||
|
|
break
|
|||
|
|
}
|
|||
|
|
dir = parent
|
|||
|
|
}
|
|||
|
|
return "", "", fmt.Errorf("找不到 threads-resolve-media-cli.ts，請設定 HAIXUN_REPO_ROOT")
|
|||
|
|
}
|