aipack 0.7.7-WIP

Command Agent runner to accelerate production coding with genai.
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

use crate::Result;
use crate::dir_context::{AipackPaths, AipackWksDir, DirContext, find_wks_dir};
use crate::hub::get_hub;
use crate::init::assets;
use crate::support::files::current_dir;
use simple_fs::{SPath, ensure_dir};
use std::fs::write;

// -- Doc Content
/// Note: The `show_info_always` will ensure that even if the `.aipack/` is found, it will print the message
///       This is useful for the `aip init` to always show the status
pub async fn init_wks(ref_dir: Option<&str>, show_info_always: bool) -> Result<DirContext> {
	let hub = get_hub();

	let wks_dir = if let Some(dir) = ref_dir {
		SPath::new(dir)
	} else if let Some(path) = find_wks_dir(current_dir()?)? {
		path
	} else {
		current_dir()?
	};

	let wks_dir = wks_dir.canonicalize()?;

	let aipack_paths = AipackPaths::from_wks_dir(&wks_dir)?;

	let aipack_wks_dir = aipack_paths
		.aipack_wks_dir()
		.ok_or_else(|| format!("Cannot Initialize Workspace because .aipack/ folder was not computed for {wks_dir}"))?;

	// -- Display the heading
	if aipack_wks_dir.exists() {
		if show_info_always {
			hub.publish("\n=== Initializing .aipack/").await;
			hub.publish(format!(
				"-- Parent path: '{}'\n   (`.aipack/` already exists. Will create missing files)",
				wks_dir
			))
			.await;
		}
	} else {
		hub.publish("\n=== Initializing .aipack/").await;
		hub.publish(format!(
			"-- Parent path: '{}'\n   (`.aipack/` will be created at this path)",
			wks_dir
		))
		.await;
	}

	// -- Init or refresh
	create_or_refresh_wks_files(aipack_wks_dir).await?;

	if show_info_always {
		hub.publish("=== DONE\n").await;
	}

	// -- Return
	let dir_context = DirContext::new(aipack_paths)?;

	Ok(dir_context)
}

/// Create or refresh missing files in a aipack directory
/// - create `.aipack/config.toml` if not present.
/// - ensure `.aipack/pack/custom/` to show use how to create per workspace agent pack
async fn create_or_refresh_wks_files(aipack_wks_dir: &AipackWksDir) -> Result<()> {
	let hub = get_hub();

	ensure_dir(aipack_wks_dir.path())?;

	// -- Create the config file
	let config_path = aipack_wks_dir.get_config_toml_path()?;

	if !config_path.exists() {
		let config_zfile = assets::extract_workspace_config_toml_zfile()?;
		write(&config_path, config_zfile.content)?;
		hub.publish(format!(
			"-> {:<18} '{}'",
			"Create config file",
			config_path.try_diff(aipack_wks_dir.parent().ok_or("Should have parent dir")?)?
		))
		.await;
	}

	// NOTE: Currently, we do not create the workspace .aipack/pack/custom directory because users can use their own paths to run agents.
	//       Eventually, we might support installing packs in the workspace using `aip install pro@coder --workspace`.
	//       These will be placed in `.aipack/pack/installed/` and will take precedence over the base custom & installed packs.

	// NOTE: For now, the workspace .aipack/pack/custom/ directory is still added to the pack resolution, which is beneficial for advanced users.

	Ok(())
}