omk 0.5.0

A Rust runtime for Kimi CLI. Turns prompts into proof-backed engineering runs with gates, worktrees, and replay.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99

use anyhow::{Context, Result};

use super::{
    GoalDeliveryPolicy, GoalGithubPrClient, GoalGithubPrDeliveryOptions,
    GoalGithubPrDeliveryOutcome, GoalGithubPrOperation, GoalGithubPrRequest,
};
use crate::runtime::goal::open_pr::{render_goal_open_pr, GoalOpenPrDraft};

pub async fn open_goal_pr_with_client<C>(
    goal_id: &str,
    options: GoalGithubPrDeliveryOptions,
    client: &mut C,
) -> Result<GoalGithubPrDeliveryOutcome>
where
    C: GoalGithubPrClient,
{
    let render_as_draft = options.draft || options.policy == GoalDeliveryPolicy::DraftPr;
    let draft = render_goal_open_pr(goal_id, render_as_draft, options.dry_run).await?;
    deliver_goal_open_pr_with_client(&draft, options, client).await
}

pub async fn deliver_goal_open_pr_with_client<C>(
    draft: &GoalOpenPrDraft,
    options: GoalGithubPrDeliveryOptions,
    client: &mut C,
) -> Result<GoalGithubPrDeliveryOutcome>
where
    C: GoalGithubPrClient,
{
    if options.dry_run {
        return Ok(skipped_outcome(
            &options,
            draft.existing_pr_url.clone(),
            "dry-run delivery requested",
        ));
    }
    if !options.policy.permits_github_mutation() {
        return Ok(skipped_outcome(
            &options,
            draft.existing_pr_url.clone(),
            "local delivery policy does not permit GitHub mutation",
        ));
    }

    let request = github_request_from_draft(draft, &options)?;
    let operation = if request.existing_pr_url.is_some() {
        GoalGithubPrOperation::Update
    } else {
        GoalGithubPrOperation::Create
    };
    let mutation = match operation {
        GoalGithubPrOperation::Create => client.create_pr(request).await?,
        GoalGithubPrOperation::Update => client.update_pr(request).await?,
    };

    Ok(GoalGithubPrDeliveryOutcome {
        policy: options.policy,
        dry_run: options.dry_run,
        mutated: true,
        operation: Some(mutation.operation),
        pr_url: mutation.url,
        reason: format!("GitHub PR {} completed", mutation.operation.as_str()),
    })
}

fn github_request_from_draft(
    draft: &GoalOpenPrDraft,
    options: &GoalGithubPrDeliveryOptions,
) -> Result<GoalGithubPrRequest> {
    let head_branch = draft
        .head_branch
        .as_deref()
        .filter(|branch| !branch.trim().is_empty())
        .context("GitHub PR delivery requires a head branch")?
        .to_string();
    Ok(GoalGithubPrRequest {
        title: draft.title.clone(),
        body: draft.body.clone(),
        head_branch,
        base_branch: options.base_branch.clone(),
        draft: options.draft || options.policy == GoalDeliveryPolicy::DraftPr,
        existing_pr_url: draft.existing_pr_url.clone(),
    })
}

fn skipped_outcome(
    options: &GoalGithubPrDeliveryOptions,
    pr_url: Option<String>,
    reason: &str,
) -> GoalGithubPrDeliveryOutcome {
    GoalGithubPrDeliveryOutcome {
        policy: options.policy,
        dry_run: options.dry_run,
        mutated: false,
        operation: None,
        pr_url,
        reason: reason.to_string(),
    }
}