selfware 0.2.2

Your personal AI workshop — software you own, software that lasts
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
115
116
117
118
119
120
121
122
123
124
125
126
127

/// Planner generates structured prompts for task planning
pub struct Planner;

impl Planner {
    /// Create a planning prompt with task and context
    pub fn create_plan(task: &str, context: &str) -> String {
        format!(
            r#"
         <task>
        {}
         </task>

         <context>
        {}
         </context>

         Create a step-by-step plan to accomplish this task. Analyze the codebase first if needed, then determine the specific files to modify and changes to make.
         "#,
            task, context
        )
    }

    /// Create a prompt for analyzing codebase structure
    pub fn analyze_prompt(path: &str) -> String {
        format!(
            r#"
Analyze the codebase at {} and provide:
1. Directory structure overview
2. Key files and their purposes
3. Dependencies (Cargo.toml, package.json, etc.)
4. Architecture patterns used
5. Entry points and main functionality

Be thorough but concise.
"#,
            path
        )
    }

    /// Create a prompt for code review
    pub fn review_prompt(file_path: &str, content: &str) -> String {
        format!(
            r#"
Review the following code from {}:

```
{}
```

Identify:
1. Potential bugs or issues
2. Code quality improvements
3. Security concerns
4. Performance optimizations
5. Documentation needs
"#,
            file_path, content
        )
    }
}

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

    #[test]
    fn test_create_plan_includes_task() {
        let plan = Planner::create_plan("Fix the bug", "Some context");
        assert!(plan.contains("<task>"));
        assert!(plan.contains("Fix the bug"));
        assert!(plan.contains("</task>"));
    }

    #[test]
    fn test_create_plan_includes_context() {
        let plan = Planner::create_plan("Some task", "Important context info");
        assert!(plan.contains("<context>"));
        assert!(plan.contains("Important context info"));
        assert!(plan.contains("</context>"));
    }

    #[test]
    fn test_create_plan_includes_instructions() {
        let plan = Planner::create_plan("Task", "Context");
        assert!(plan.contains("step-by-step plan"));
        assert!(plan.contains("Analyze the codebase"));
    }

    #[test]
    fn test_create_plan_with_empty_task() {
        let plan = Planner::create_plan("", "Context");
        assert!(plan.contains("<task>"));
        assert!(plan.contains("</task>"));
    }

    #[test]
    fn test_create_plan_with_empty_context() {
        let plan = Planner::create_plan("Task", "");
        assert!(plan.contains("<context>"));
        assert!(plan.contains("</context>"));
    }

    #[test]
    fn test_create_plan_with_special_characters() {
        let plan =
            Planner::create_plan("Fix <xml> &amp; \"quotes\"", "Context with 'special' chars");
        assert!(plan.contains("Fix <xml> &amp; \"quotes\""));
        assert!(plan.contains("Context with 'special' chars"));
    }

    #[test]
    fn test_analyze_prompt_includes_path() {
        let prompt = Planner::analyze_prompt("./src");
        assert!(prompt.contains("./src"));
        assert!(prompt.contains("Directory structure"));
        assert!(prompt.contains("Dependencies"));
    }

    #[test]
    fn test_review_prompt_includes_content() {
        let prompt = Planner::review_prompt("src/main.rs", "fn main() {}");
        assert!(prompt.contains("src/main.rs"));
        assert!(prompt.contains("fn main() {}"));
        assert!(prompt.contains("bugs"));
        assert!(prompt.contains("Security"));
    }
}