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
|
use anyhow::{Context, Result};
use clap::Parser;
use derivative::Derivative;
use tracing::{debug, metadata::LevelFilter, Level};
use tracing_subscriber::{fmt::format::FmtSpan, prelude::*, EnvFilter};
use crate::repo::{Aggregator, Repos};
pub mod config;
pub mod forge;
pub mod git;
pub mod repo;
mod list;
mod sync;
mod update;
#[cfg(test)]
mod tests;
#[derive(Derivative)]
#[derivative(Debug)]
struct GTree {
figment: figment::Figment,
config: config::Config,
args: config::args::Args,
forge: forge::Forge,
}
impl GTree {
#[tracing::instrument(level = "trace")]
pub async fn new() -> Result<GTree> {
let args = config::args::Args::parse();
let figment = config::Config::figment()?;
let config: config::Config = figment.extract()?;
let (_name, forge_config) = config
.iter()
.next()
.context("No Forge configured, please setup a forge")?;
let forge = forge::Forge::new(forge_config).await?;
Ok(GTree {
figment,
config,
args,
forge,
})
}
#[tracing::instrument(level = "trace")]
pub async fn run(self) -> Result<()> {
let scope = self.args.scope.as_ref().map_or("", |x| x);
// TODO select a specific forge
let (_name, forge) = self
.config
.iter()
.next()
.context("No Forge configured, please setup a forge")?;
let (local, remote) = tokio::join!(
Repos::from_local(forge.root(), scope),
Repos::from_forge(forge.root(), self.forge.projects(scope).await?)
);
let repos = Repos::aggregate(local, remote).await;
match self.args.command {
config::args::Commands::Sync => self.sync(repos).await,
config::args::Commands::Update => self.update(repos).await,
config::args::Commands::List => self.list(repos).await?,
};
Ok(())
}
}
#[tokio::main]
async fn main() -> Result<()> {
use tracing_flame::FlameLayer;
let filter = tracing_subscriber::filter::Targets::new()
.with_default(Level::TRACE)
.with_target("hyper", LevelFilter::OFF)
.with_target("reqwest", LevelFilter::OFF);
let env_filter = EnvFilter::from_default_env();
let (flame_layer, _guard) = FlameLayer::with_file("./tracing.folded").unwrap();
let flameguard = flame_layer.flush_on_drop();
tracing_subscriber::registry()
.with(tracing_subscriber::fmt::layer().with_span_events(FmtSpan::ACTIVE))
.with(filter)
.with(env_filter)
.with(flame_layer)
.init();
debug!("starting");
let gtree = GTree::new().await?;
gtree.run().await?;
flameguard.flush()?;
Ok(())
}
|
