vtcode-core 0.104.1

Core library for VT Code - a Rust-based terminal coding agent
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

//! Test for configuration loading with home directory support

use std::fs;
use std::path::{Path, PathBuf};
use std::sync::Arc;
use tempfile::TempDir;
use vtcode_commons::paths::WorkspacePaths;
use vtcode_core::config::VTCodeConfig;
use vtcode_core::config::defaults::{self, ConfigDefaultsProvider};

struct TestDefaults {
    inner: Arc<dyn ConfigDefaultsProvider>,
    home_dir: PathBuf,
}

impl std::fmt::Debug for TestDefaults {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("TestDefaults")
            .field("home_dir", &self.home_dir)
            .finish()
    }
}

impl TestDefaults {
    fn new(inner: Arc<dyn ConfigDefaultsProvider>, home_dir: PathBuf) -> Self {
        Self { inner, home_dir }
    }
}

impl ConfigDefaultsProvider for TestDefaults {
    fn config_file_name(&self) -> &str {
        self.inner.config_file_name()
    }

    fn workspace_paths_for(&self, workspace_root: &Path) -> Box<dyn WorkspacePaths> {
        self.inner.workspace_paths_for(workspace_root)
    }

    fn home_config_paths(&self, config_file_name: &str) -> Vec<PathBuf> {
        vec![self.home_dir.join(".vtcode").join(config_file_name)]
    }

    fn syntax_theme(&self) -> String {
        self.inner.syntax_theme()
    }

    fn syntax_languages(&self) -> Vec<String> {
        self.inner.syntax_languages()
    }
}

#[test]
fn test_load_config_from_home_directory() {
    // Create a temporary directory to simulate home directory
    let temp_dir = TempDir::new().expect("Failed to create temp directory");
    let home_dir = temp_dir.path();
    let vtcode_dir = home_dir.join(".vtcode");
    fs::create_dir_all(&vtcode_dir).expect("Failed to create .vtcode directory");

    // Create a sample config file in the home directory
    let config_content = r#"
[agent]
default_model = "test-model"
provider = "test-provider"
max_conversation_turns = 50

[security]
human_in_the_loop = false
"#;

    let config_path = vtcode_dir.join("vtcode.toml");
    fs::write(&config_path, config_content).expect("Failed to write config file");

    // Create a mock workspace directory
    let workspace_dir = temp_dir.path().join("workspace");
    fs::create_dir_all(&workspace_dir).expect("Failed to create workspace directory");

    // Test that we can load the config from home directory
    // We need to mock the home directory detection for this test
    // Since we can't easily mock environment variables in a test,
    // we'll directly test the bootstrap function with our temp directory
    let created_files = defaults::provider::with_config_defaults_provider_for_test(
        Arc::new(TestDefaults::new(
            defaults::current_config_defaults(),
            home_dir.to_path_buf(),
        )),
        || {
            VTCodeConfig::bootstrap_project_with_options(
                &workspace_dir,
                true, // force
                true, // use home directory
            )
        },
    )
    .expect("Failed to bootstrap project with home directory");

    assert!(!created_files.is_empty());
    assert!(created_files.contains(&"vtcode.toml".to_string()));

    // Check that the files were created in the home directory
    let home_config_path = vtcode_dir.join("vtcode.toml");
    let home_gitignore_path = vtcode_dir.join(".vtcodegitignore");
    assert!(home_config_path.exists());
    assert!(home_gitignore_path.exists());
}