import (
    "github.com/resolute-sh/resolute-bitbucket"
)

// Register with worker
bitbucket.RegisterActivities(w)

// Or use Provider() for introspection
provider := bitbucket.Provider()

type PRMetadata struct {
    Workspace    string
    RepoSlug     string
    Branch       string
    PRID         int
    TargetBranch string
    CommitSHA    string
    Author       string
    Title        string
    PRURL        string
}

type PRCreatedEvent struct {
    PullRequest PullRequest `json:"pullrequest"`
    Repository  Repository  `json:"repository"`
    Actor       Actor       `json:"actor"`
}

type ParseWebhookInput struct {
    RawPayload string  // Raw JSON webhook payload
}

type ParseWebhookOutput struct {
    PR PRMetadata
}

func ParseWebhook(input ParseWebhookInput) *core.Node[ParseWebhookInput, ParseWebhookOutput]

parseNode := bitbucket.ParseWebhook(bitbucket.ParseWebhookInput{
    RawPayload: core.InputData("webhook_payload"),
})

type AddCommentInput struct {
    Username    string  // Bitbucket username
    AppPassword string  // Bitbucket App Password
    Workspace   string  // Workspace slug
    RepoSlug    string  // Repository slug
    PRID        string  // Pull request ID
    Body        string  // Comment body (Markdown supported)
}

type AddCommentOutput struct {
    Posted bool
}

func AddComment(input AddCommentInput) *core.Node[AddCommentInput, AddCommentOutput]

commentNode := bitbucket.AddComment(bitbucket.AddCommentInput{
    Username:    os.Getenv("BITBUCKET_USERNAME"),
    AppPassword: os.Getenv("BITBUCKET_APP_PASSWORD"),
    Workspace:   core.Output("parse-webhook.PR.Workspace"),
    RepoSlug:    core.Output("parse-webhook.PR.RepoSlug"),
    PRID:        core.Output("parse-webhook.PR.PRID"),
    Body:        core.Output("review.Summary"),
})

func ExtractTicketID(branch string) string

ticket := bitbucket.ExtractTicketID("feature/PLATFORM-123-add-auth")
// ticket = "PLATFORM-123"

ticket := bitbucket.ExtractTicketID("hotfix/no-ticket")
// ticket = ""

flow := core.NewFlow("pr-review").
    TriggeredBy(core.Webhook("/bitbucket")).
    Then(bitbucket.ParseWebhook(bitbucket.ParseWebhookInput{
        RawPayload: core.InputData("webhook_payload"),
    }).As("pr")).
    Then(fetchDiffNode).
    Then(analyzeCodeNode).
    Then(bitbucket.AddComment(bitbucket.AddCommentInput{
        Username:    os.Getenv("BITBUCKET_USERNAME"),
        AppPassword: os.Getenv("BITBUCKET_APP_PASSWORD"),
        Workspace:   core.Output("pr.PR.Workspace"),
        RepoSlug:    core.Output("pr.PR.RepoSlug"),
        PRID:        core.Output("pr.PR.PRID"),
        Body:        core.Output("analyze-code.ReviewComment"),
    })).
    Build()

// In an InputFunc, extract Jira ticket from branch name
processNode := core.NewNode("process", processActivity, ProcessInput{}).
    WithInputFunc(func(state *core.FlowState) ProcessInput {
        pr := core.Get[bitbucket.ParseWebhookOutput](state, "pr")
        ticket := bitbucket.ExtractTicketID(pr.PR.Branch)
        return ProcessInput{
            PRID:     pr.PR.PRID,
            TicketID: ticket,
        }
    })