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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114

use crate::analysis::parser::Language;
use crate::analysis::query::{call_target_name, node_range, CallSite};

pub(super) fn collect_calls(
    source: &str,
    root: tree_sitter::Node,
    language: Language,
    target: &str,
) -> Vec<CallSite> {
    let mut results = Vec::new();
    match language {
        Language::Rust => collect_rust_calls(source, root, target, &mut results),
        Language::JavaScript => collect_js_calls(source, root, target, &mut results),
        Language::Python => collect_python_calls(source, root, target, &mut results),
        Language::Go => collect_go_calls(source, root, target, &mut results),
    }
    results
}

fn collect_rust_calls(
    source: &str,
    node: tree_sitter::Node,
    target: &str,
    results: &mut Vec<CallSite>,
) {
    if node.kind() == "call_expression" {
        if let Some(func) = node.child_by_field_name("function") {
            let name = call_target_name(source, func);
            if name == target {
                results.push(CallSite {
                    name,
                    span: node_range(node),
                });
            }
        }
    }
    for i in 0..node.child_count() {
        if let Some(child) = node.child(i as u32) {
            collect_rust_calls(source, child, target, results);
        }
    }
}

fn collect_js_calls(
    source: &str,
    node: tree_sitter::Node,
    target: &str,
    results: &mut Vec<CallSite>,
) {
    if node.kind() == "call_expression" {
        if let Some(func) = node.child_by_field_name("function") {
            let name = call_target_name(source, func);
            if name == target {
                results.push(CallSite {
                    name,
                    span: node_range(node),
                });
            }
        }
    }
    for i in 0..node.child_count() {
        if let Some(child) = node.child(i as u32) {
            collect_js_calls(source, child, target, results);
        }
    }
}

fn collect_python_calls(
    source: &str,
    node: tree_sitter::Node,
    target: &str,
    results: &mut Vec<CallSite>,
) {
    if node.kind() == "call" {
        if let Some(func) = node.child_by_field_name("function") {
            let name = call_target_name(source, func);
            if name == target {
                results.push(CallSite {
                    name,
                    span: node_range(node),
                });
            }
        }
    }
    for i in 0..node.child_count() {
        if let Some(child) = node.child(i as u32) {
            collect_python_calls(source, child, target, results);
        }
    }
}

fn collect_go_calls(
    source: &str,
    node: tree_sitter::Node,
    target: &str,
    results: &mut Vec<CallSite>,
) {
    if node.kind() == "call_expression" {
        if let Some(func) = node.child_by_field_name("function") {
            let name = call_target_name(source, func);
            if name == target {
                results.push(CallSite {
                    name,
                    span: node_range(node),
                });
            }
        }
    }
    for i in 0..node.child_count() {
        if let Some(child) = node.child(i as u32) {
            collect_go_calls(source, child, target, results);
        }
    }
}