kwaak 0.1.2

Run a team of autonomous agents on your code, right from your terminal
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

use crate::{
    config::defaults::{default_main_branch, default_owner_and_repo, default_project_name},
    templates::Templates,
};
use anyhow::{Context as _, Result};
use serde_json::json;
use swiftide::integrations::treesitter::SupportedLanguages;

pub fn create_template_config() -> Result<String> {
    let mut context = tera::Context::new();

    context.insert(
        "language",
        &naive_lang_detect().map_or_else(|| "REQUIRED".to_string(), |l| l.to_string()),
    );
    context.insert("project_name", &default_project_name());

    let (owner, repository) = default_owner_and_repo();
    context.insert(
        "github",
        &json!({
            "owner": owner,
            "repository": repository,
            "main_branch": default_main_branch(),

        }),
    );

    let config =
        Templates::render("kwaak.toml", &context).context("Failed to render default config")?;

    // Since we want the template annotated with comments, just return the template
    Ok(config)
}

fn naive_lang_detect() -> Option<SupportedLanguages> {
    // Check for major package manager files to detect the language
    // Then return the first language found
    let language_files = [
        ("Cargo.toml", "Rust"),
        ("Gemfile", "Ruby"),
        ("tsconfig.json", "Typescript"),
        ("package.json", "Javascript"),
        ("pyproject.toml", "Python"),
        ("requirements.txt", "Python"),
        ("Pipfile", "Python"),
        ("build.gradle", "Java"),
        ("pom.xml", "Java"),
    ];

    // Iterate through the files and detect the language
    for (file, language) in language_files {
        if std::fs::metadata(file).is_ok() {
            return language.parse().ok();
        }
    }

    None
}

#[cfg(test)]
mod test {
    use super::*;

    #[test]
    fn test_valid_template() {
        // Clean up env variables for a pure test
        std::env::vars().for_each(|(key, _)| {
            if key.starts_with("KWAAK") {
                std::env::remove_var(key);
            }
        });
        std::env::set_var("KWAAK_OPENAI_API_KEY", "test");
        std::env::set_var("KWAAK_GITHUB_TOKEN", "test");
        let config = create_template_config().unwrap();

        toml::from_str::<crate::config::Config>(&config).unwrap();
    }
}