balls 0.3.0

Git-native task tracker for parallel agent workflows
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

mod limits;
mod runner;
mod types;

pub use runner::Plugin;
pub use types::{PushResponse, SyncCreate, SyncDelete, SyncReport, SyncUpdate};

use crate::config::Config;
use crate::error::Result;
use crate::store::{task_lock, Store};
use crate::task::Task;
use serde_json::Value;
use std::collections::BTreeMap;

/// Run plugin push for all active plugins. Returns a map of
/// plugin_name -> PushResponse for plugins that returned data.
pub fn run_plugin_push(store: &Store, task: &Task) -> Result<BTreeMap<String, PushResponse>> {
    let cfg = store.load_config()?;
    let mut results = BTreeMap::new();
    for (name, entry) in active_plugins(&cfg) {
        if entry.sync_on_change {
            let plugin = Plugin::resolve(store, name, entry);
            if !plugin.auth_check() {
                continue;
            }
            if let Ok(Some(result)) = plugin.push(task) {
                results.insert(name.clone(), result);
            }
        }
    }
    Ok(results)
}

/// Run plugin sync for all active plugins. Returns (plugin_name, SyncReport) pairs.
pub fn run_plugin_sync(
    store: &Store,
    filter: Option<&str>,
) -> Result<Vec<(String, SyncReport)>> {
    let cfg = store.load_config()?;
    let tasks = store.all_tasks()?;
    let mut reports = Vec::new();
    for (name, entry) in active_plugins(&cfg) {
        let plugin = Plugin::resolve(store, name, entry);
        if !plugin.auth_check() {
            continue;
        }
        if let Ok(Some(report)) = plugin.sync(&tasks, filter) {
            reports.push((name.clone(), report));
        }
    }
    Ok(reports)
}

/// Merge push results into task.external and save+commit.
pub fn apply_push_response(
    store: &Store,
    task_id: &str,
    results: &BTreeMap<String, PushResponse>,
) -> Result<()> {
    if results.is_empty() {
        return Ok(());
    }
    let _g = task_lock(store, task_id)?;
    let mut task = store.load_task(task_id)?;
    for (plugin_name, response) in results {
        let ext_value = Value::Object(response.0.clone());
        task.external.insert(plugin_name.clone(), ext_value);
    }
    task.touch();
    store.save_task(&task)?;
    store.commit_task(task_id, &format!("balls: update external for {task_id}"))?;
    Ok(())
}

fn active_plugins(cfg: &Config) -> impl Iterator<Item = (&String, &crate::config::PluginEntry)> {
    cfg.plugins.iter().filter(|(_, e)| e.enabled)
}