64 lines
2.0 KiB
Go
64 lines
2.0 KiB
Go
|
package job
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
libthreads "haixun-backend/internal/library/threadsapi"
|
||
|
|
jobdom "haixun-backend/internal/model/job/domain/usecase"
|
||
|
|
threadsaccountdomain "haixun-backend/internal/model/threads_account/domain/usecase"
|
||
|
|
)
|
||
|
|
|
||
|
|
type RefreshThreadsTokenDeps struct {
|
||
|
|
Jobs jobdom.UseCase
|
||
|
|
ThreadsAccount threadsaccountdomain.UseCase
|
||
|
|
}
|
||
|
|
|
||
|
|
func RegisterRefreshThreadsTokenHandler(runner *Runner, deps RefreshThreadsTokenDeps) {
|
||
|
|
if runner == nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
runner.RegisterStepHandler("refresh", func(ctx context.Context, step StepContext) error {
|
||
|
|
return runRefreshThreadsToken(ctx, step, deps)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func runRefreshThreadsToken(ctx context.Context, step StepContext, deps RefreshThreadsTokenDeps) error {
|
||
|
|
payload := step.Run.Payload
|
||
|
|
tenantID, ownerUID := runActorFromPayload(payload, step.Run)
|
||
|
|
accountID := strings.TrimSpace(stringField(payload, "account_id"))
|
||
|
|
if tenantID == "" || ownerUID == "" || accountID == "" {
|
||
|
|
return fmt.Errorf("refresh-threads-token payload missing tenant_id, owner_uid, or account_id")
|
||
|
|
}
|
||
|
|
|
||
|
|
cred, err := deps.ThreadsAccount.ResolveMemberThreadsPublishCredential(ctx, tenantID, ownerUID)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("resolve publish credential: %w", err)
|
||
|
|
}
|
||
|
|
if cred == nil || cred.AccessToken == "" {
|
||
|
|
return fmt.Errorf("no access token found for account %s", accountID)
|
||
|
|
}
|
||
|
|
|
||
|
|
result, err := libthreads.RefreshAccessToken(ctx, cred.AccessToken)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("refresh token: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
expiresAt := libthreads.NsFromNow(result.ExpiresIn)
|
||
|
|
if _, err := deps.ThreadsAccount.SaveAPIAccessToken(ctx, accountID, result.AccessToken, expiresAt); err != nil {
|
||
|
|
return fmt.Errorf("save refreshed token: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
_, err = deps.Jobs.CompleteRun(ctx, jobdom.CompleteRunRequest{
|
||
|
|
JobID: step.JobID,
|
||
|
|
WorkerID: step.WorkerID,
|
||
|
|
Result: map[string]any{
|
||
|
|
"account_id": accountID,
|
||
|
|
"expires_at": expiresAt,
|
||
|
|
"token_short": result.AccessToken[:min(len(result.AccessToken), 8)] + "...",
|
||
|
|
},
|
||
|
|
})
|
||
|
|
return err
|
||
|
|
}
|