fix: apply different network policies based on current target #97
@ -9,7 +9,7 @@ jobs:
|
|||||||
check:
|
check:
|
||||||
runs-on: docker
|
runs-on: docker
|
||||||
container:
|
container:
|
||||||
image: hub.nationtech.io/harmony/harmony_composer:latest@sha256:eb0406fcb95c63df9b7c4b19bc50ad7914dd8232ce98e9c9abef628e07c69386
|
image: hub.nationtech.io/harmony/harmony_composer:latest
|
||||||
steps:
|
steps:
|
||||||
- name: Checkout code
|
- name: Checkout code
|
||||||
uses: actions/checkout@v4
|
uses: actions/checkout@v4
|
||||||
|
|||||||
@ -7,7 +7,7 @@ on:
|
|||||||
jobs:
|
jobs:
|
||||||
package_harmony_composer:
|
package_harmony_composer:
|
||||||
container:
|
container:
|
||||||
image: hub.nationtech.io/harmony/harmony_composer:latest@sha256:eb0406fcb95c63df9b7c4b19bc50ad7914dd8232ce98e9c9abef628e07c69386
|
image: hub.nationtech.io/harmony/harmony_composer:latest
|
||||||
runs-on: dind
|
runs-on: dind
|
||||||
steps:
|
steps:
|
||||||
- name: Checkout code
|
- name: Checkout code
|
||||||
@ -45,14 +45,14 @@ jobs:
|
|||||||
-H "Authorization: token ${{ secrets.GITEATOKEN }}" \
|
-H "Authorization: token ${{ secrets.GITEATOKEN }}" \
|
||||||
"https://git.nationtech.io/api/v1/repos/nationtech/harmony/releases/tags/snapshot-latest" \
|
"https://git.nationtech.io/api/v1/repos/nationtech/harmony/releases/tags/snapshot-latest" \
|
||||||
| jq -r '.id // empty')
|
| jq -r '.id // empty')
|
||||||
|
|
||||||
if [ -n "$RELEASE_ID" ]; then
|
if [ -n "$RELEASE_ID" ]; then
|
||||||
# Delete existing release
|
# Delete existing release
|
||||||
curl -X DELETE \
|
curl -X DELETE \
|
||||||
-H "Authorization: token ${{ secrets.GITEATOKEN }}" \
|
-H "Authorization: token ${{ secrets.GITEATOKEN }}" \
|
||||||
"https://git.nationtech.io/api/v1/repos/nationtech/harmony/releases/$RELEASE_ID"
|
"https://git.nationtech.io/api/v1/repos/nationtech/harmony/releases/$RELEASE_ID"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
# Create new release
|
# Create new release
|
||||||
RESPONSE=$(curl -X POST \
|
RESPONSE=$(curl -X POST \
|
||||||
-H "Authorization: token ${{ secrets.GITEATOKEN }}" \
|
-H "Authorization: token ${{ secrets.GITEATOKEN }}" \
|
||||||
@ -65,7 +65,7 @@ jobs:
|
|||||||
"prerelease": true
|
"prerelease": true
|
||||||
}' \
|
}' \
|
||||||
"https://git.nationtech.io/api/v1/repos/nationtech/harmony/releases")
|
"https://git.nationtech.io/api/v1/repos/nationtech/harmony/releases")
|
||||||
|
|
||||||
echo "RELEASE_ID=$(echo $RESPONSE | jq -r '.id')" >> $GITHUB_ENV
|
echo "RELEASE_ID=$(echo $RESPONSE | jq -r '.id')" >> $GITHUB_ENV
|
||||||
|
|
||||||
- name: Upload Linux binary
|
- name: Upload Linux binary
|
||||||
|
|||||||
@ -5,6 +5,9 @@ version.workspace = true
|
|||||||
readme.workspace = true
|
readme.workspace = true
|
||||||
license.workspace = true
|
license.workspace = true
|
||||||
|
|
||||||
|
[features]
|
||||||
|
testing = []
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
rand = "0.9"
|
rand = "0.9"
|
||||||
hex = "0.4"
|
hex = "0.4"
|
||||||
|
|||||||
@ -11,5 +11,5 @@ lazy_static! {
|
|||||||
pub static ref REGISTRY_PROJECT: String =
|
pub static ref REGISTRY_PROJECT: String =
|
||||||
std::env::var("HARMONY_REGISTRY_PROJECT").unwrap_or_else(|_| "harmony".to_string());
|
std::env::var("HARMONY_REGISTRY_PROJECT").unwrap_or_else(|_| "harmony".to_string());
|
||||||
pub static ref DRY_RUN: bool =
|
pub static ref DRY_RUN: bool =
|
||||||
std::env::var("HARMONY_DRY_RUN").map_or(true, |value| value.parse().unwrap_or(true));
|
std::env::var("HARMONY_DRY_RUN").is_ok_and(|value| value.parse().unwrap_or(false));
|
||||||
}
|
}
|
||||||
|
|||||||
@ -2,28 +2,34 @@ use log::debug;
|
|||||||
use once_cell::sync::Lazy;
|
use once_cell::sync::Lazy;
|
||||||
use tokio::sync::broadcast;
|
use tokio::sync::broadcast;
|
||||||
|
|
||||||
use super::interpret::{InterpretError, Outcome};
|
use super::{
|
||||||
|
interpret::{InterpretError, Outcome},
|
||||||
|
topology::TopologyStatus,
|
||||||
|
};
|
||||||
|
|
||||||
#[derive(Debug, Clone)]
|
#[derive(Debug, Clone)]
|
||||||
pub enum HarmonyEvent {
|
pub enum HarmonyEvent {
|
||||||
HarmonyStarted,
|
HarmonyStarted,
|
||||||
PrepareTopologyStarted {
|
HarmonyFinished,
|
||||||
topology: String,
|
|
||||||
},
|
|
||||||
TopologyPrepared {
|
|
||||||
topology: String,
|
|
||||||
outcome: Outcome,
|
|
||||||
},
|
|
||||||
InterpretExecutionStarted {
|
InterpretExecutionStarted {
|
||||||
|
execution_id: String,
|
||||||
topology: String,
|
topology: String,
|
||||||
interpret: String,
|
interpret: String,
|
||||||
|
score: String,
|
||||||
message: String,
|
message: String,
|
||||||
},
|
},
|
||||||
InterpretExecutionFinished {
|
InterpretExecutionFinished {
|
||||||
|
execution_id: String,
|
||||||
topology: String,
|
topology: String,
|
||||||
interpret: String,
|
interpret: String,
|
||||||
|
score: String,
|
||||||
outcome: Result<Outcome, InterpretError>,
|
outcome: Result<Outcome, InterpretError>,
|
||||||
},
|
},
|
||||||
|
TopologyStateChanged {
|
||||||
|
topology: String,
|
||||||
|
status: TopologyStatus,
|
||||||
|
message: Option<String>,
|
||||||
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
static HARMONY_EVENT_BUS: Lazy<broadcast::Sender<HarmonyEvent>> = Lazy::new(|| {
|
static HARMONY_EVENT_BUS: Lazy<broadcast::Sender<HarmonyEvent>> = Lazy::new(|| {
|
||||||
@ -33,9 +39,14 @@ static HARMONY_EVENT_BUS: Lazy<broadcast::Sender<HarmonyEvent>> = Lazy::new(|| {
|
|||||||
});
|
});
|
||||||
|
|
||||||
pub fn instrument(event: HarmonyEvent) -> Result<(), &'static str> {
|
pub fn instrument(event: HarmonyEvent) -> Result<(), &'static str> {
|
||||||
match HARMONY_EVENT_BUS.send(event) {
|
if cfg!(any(test, feature = "testing")) {
|
||||||
Ok(_) => Ok(()),
|
let _ = event; // Suppress the "unused variable" warning for `event`
|
||||||
Err(_) => Err("send error: no subscribers"),
|
Ok(())
|
||||||
|
} else {
|
||||||
|
match HARMONY_EVENT_BUS.send(event) {
|
||||||
|
Ok(_) => Ok(()),
|
||||||
|
Err(_) => Err("send error: no subscribers"),
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@ -7,6 +7,7 @@ use super::{
|
|||||||
data::{Id, Version},
|
data::{Id, Version},
|
||||||
executors::ExecutorError,
|
executors::ExecutorError,
|
||||||
inventory::Inventory,
|
inventory::Inventory,
|
||||||
|
topology::PreparationError,
|
||||||
};
|
};
|
||||||
|
|
||||||
pub enum InterpretName {
|
pub enum InterpretName {
|
||||||
@ -23,6 +24,14 @@ pub enum InterpretName {
|
|||||||
TenantInterpret,
|
TenantInterpret,
|
||||||
Application,
|
Application,
|
||||||
ArgoCD,
|
ArgoCD,
|
||||||
|
Alerting,
|
||||||
|
Ntfy,
|
||||||
|
HelmChart,
|
||||||
|
HelmCommand,
|
||||||
|
K8sResource,
|
||||||
|
Lamp,
|
||||||
|
ApplicationMonitoring,
|
||||||
|
K8sPrometheusCrdAlerting,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl std::fmt::Display for InterpretName {
|
impl std::fmt::Display for InterpretName {
|
||||||
@ -41,6 +50,14 @@ impl std::fmt::Display for InterpretName {
|
|||||||
InterpretName::TenantInterpret => f.write_str("Tenant"),
|
InterpretName::TenantInterpret => f.write_str("Tenant"),
|
||||||
InterpretName::Application => f.write_str("Application"),
|
InterpretName::Application => f.write_str("Application"),
|
||||||
InterpretName::ArgoCD => f.write_str("ArgoCD"),
|
InterpretName::ArgoCD => f.write_str("ArgoCD"),
|
||||||
|
InterpretName::Alerting => f.write_str("Alerting"),
|
||||||
|
InterpretName::Ntfy => f.write_str("Ntfy"),
|
||||||
|
InterpretName::HelmChart => f.write_str("HelmChart"),
|
||||||
|
InterpretName::HelmCommand => f.write_str("HelmCommand"),
|
||||||
|
InterpretName::K8sResource => f.write_str("K8sResource"),
|
||||||
|
InterpretName::Lamp => f.write_str("LAMP"),
|
||||||
|
InterpretName::ApplicationMonitoring => f.write_str("ApplicationMonitoring"),
|
||||||
|
InterpretName::K8sPrometheusCrdAlerting => f.write_str("K8sPrometheusCrdAlerting"),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -113,6 +130,14 @@ impl std::fmt::Display for InterpretError {
|
|||||||
}
|
}
|
||||||
impl Error for InterpretError {}
|
impl Error for InterpretError {}
|
||||||
|
|
||||||
|
impl From<PreparationError> for InterpretError {
|
||||||
|
fn from(value: PreparationError) -> Self {
|
||||||
|
Self {
|
||||||
|
msg: format!("InterpretError : {value}"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
impl From<ExecutorError> for InterpretError {
|
impl From<ExecutorError> for InterpretError {
|
||||||
fn from(value: ExecutorError) -> Self {
|
fn from(value: ExecutorError) -> Self {
|
||||||
Self {
|
Self {
|
||||||
|
|||||||
@ -1,14 +1,14 @@
|
|||||||
use std::sync::{Arc, Mutex, RwLock};
|
use std::sync::{Arc, RwLock};
|
||||||
|
|
||||||
use log::{debug, warn};
|
use log::{debug, warn};
|
||||||
|
|
||||||
use crate::instrumentation::{self, HarmonyEvent};
|
use crate::topology::TopologyStatus;
|
||||||
|
|
||||||
use super::{
|
use super::{
|
||||||
interpret::{InterpretError, InterpretStatus, Outcome},
|
interpret::{InterpretError, Outcome},
|
||||||
inventory::Inventory,
|
inventory::Inventory,
|
||||||
score::Score,
|
score::Score,
|
||||||
topology::Topology,
|
topology::{PreparationError, PreparationOutcome, Topology, TopologyState},
|
||||||
};
|
};
|
||||||
|
|
||||||
type ScoreVec<T> = Vec<Box<dyn Score<T>>>;
|
type ScoreVec<T> = Vec<Box<dyn Score<T>>>;
|
||||||
@ -17,7 +17,7 @@ pub struct Maestro<T: Topology> {
|
|||||||
inventory: Inventory,
|
inventory: Inventory,
|
||||||
topology: T,
|
topology: T,
|
||||||
scores: Arc<RwLock<ScoreVec<T>>>,
|
scores: Arc<RwLock<ScoreVec<T>>>,
|
||||||
topology_preparation_result: Mutex<Option<Outcome>>,
|
topology_state: TopologyState,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<T: Topology> Maestro<T> {
|
impl<T: Topology> Maestro<T> {
|
||||||
@ -25,41 +25,46 @@ impl<T: Topology> Maestro<T> {
|
|||||||
///
|
///
|
||||||
/// This should rarely be used. Most of the time Maestro::initialize should be used instead.
|
/// This should rarely be used. Most of the time Maestro::initialize should be used instead.
|
||||||
pub fn new_without_initialization(inventory: Inventory, topology: T) -> Self {
|
pub fn new_without_initialization(inventory: Inventory, topology: T) -> Self {
|
||||||
|
let topology_name = topology.name().to_string();
|
||||||
|
|
||||||
Self {
|
Self {
|
||||||
inventory,
|
inventory,
|
||||||
topology,
|
topology,
|
||||||
scores: Arc::new(RwLock::new(Vec::new())),
|
scores: Arc::new(RwLock::new(Vec::new())),
|
||||||
topology_preparation_result: None.into(),
|
topology_state: TopologyState::new(topology_name),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn initialize(inventory: Inventory, topology: T) -> Result<Self, InterpretError> {
|
pub async fn initialize(inventory: Inventory, topology: T) -> Result<Self, PreparationError> {
|
||||||
let instance = Self::new_without_initialization(inventory, topology);
|
let mut instance = Self::new_without_initialization(inventory, topology);
|
||||||
instance.prepare_topology().await?;
|
instance.prepare_topology().await?;
|
||||||
Ok(instance)
|
Ok(instance)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Ensures the associated Topology is ready for operations.
|
/// Ensures the associated Topology is ready for operations.
|
||||||
/// Delegates the readiness check and potential setup actions to the Topology.
|
/// Delegates the readiness check and potential setup actions to the Topology.
|
||||||
pub async fn prepare_topology(&self) -> Result<Outcome, InterpretError> {
|
async fn prepare_topology(&mut self) -> Result<PreparationOutcome, PreparationError> {
|
||||||
instrumentation::instrument(HarmonyEvent::PrepareTopologyStarted {
|
self.topology_state.prepare();
|
||||||
topology: self.topology.name().to_string(),
|
|
||||||
})
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
let outcome = self.topology.ensure_ready().await?;
|
let result = self.topology.ensure_ready().await;
|
||||||
|
|
||||||
instrumentation::instrument(HarmonyEvent::TopologyPrepared {
|
match result {
|
||||||
topology: self.topology.name().to_string(),
|
Ok(outcome) => {
|
||||||
outcome: outcome.clone(),
|
match outcome.clone() {
|
||||||
})
|
PreparationOutcome::Success { details } => {
|
||||||
.unwrap();
|
self.topology_state.success(details);
|
||||||
|
}
|
||||||
self.topology_preparation_result
|
PreparationOutcome::Noop => {
|
||||||
.lock()
|
self.topology_state.noop();
|
||||||
.unwrap()
|
}
|
||||||
.replace(outcome.clone());
|
};
|
||||||
Ok(outcome)
|
Ok(outcome)
|
||||||
|
}
|
||||||
|
Err(err) => {
|
||||||
|
self.topology_state.error(err.to_string());
|
||||||
|
Err(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn register_all(&mut self, mut scores: ScoreVec<T>) {
|
pub fn register_all(&mut self, mut scores: ScoreVec<T>) {
|
||||||
@ -68,12 +73,7 @@ impl<T: Topology> Maestro<T> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn is_topology_initialized(&self) -> bool {
|
fn is_topology_initialized(&self) -> bool {
|
||||||
let result = self.topology_preparation_result.lock().unwrap();
|
self.topology_state.status == TopologyStatus::Success
|
||||||
if let Some(outcome) = result.as_ref() {
|
|
||||||
matches!(outcome.status, InterpretStatus::SUCCESS)
|
|
||||||
} else {
|
|
||||||
false
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn interpret(&self, score: Box<dyn Score<T>>) -> Result<Outcome, InterpretError> {
|
pub async fn interpret(&self, score: Box<dyn Score<T>>) -> Result<Outcome, InterpretError> {
|
||||||
@ -84,10 +84,8 @@ impl<T: Topology> Maestro<T> {
|
|||||||
self.topology.name(),
|
self.topology.name(),
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
debug!("Running score {score:?}");
|
debug!("Interpreting score {score:?}");
|
||||||
let interpret = score.create_interpret();
|
let result = score.interpret(&self.inventory, &self.topology).await;
|
||||||
debug!("Launching interpret {interpret:?}");
|
|
||||||
let result = interpret.execute(&self.inventory, &self.topology).await;
|
|
||||||
debug!("Got result {result:?}");
|
debug!("Got result {result:?}");
|
||||||
result
|
result
|
||||||
}
|
}
|
||||||
|
|||||||
@ -1,15 +1,55 @@
|
|||||||
use std::collections::BTreeMap;
|
use std::collections::BTreeMap;
|
||||||
|
|
||||||
|
use async_trait::async_trait;
|
||||||
use serde::Serialize;
|
use serde::Serialize;
|
||||||
use serde_value::Value;
|
use serde_value::Value;
|
||||||
|
|
||||||
use super::{interpret::Interpret, topology::Topology};
|
use super::{
|
||||||
|
data::Id,
|
||||||
|
instrumentation::{self, HarmonyEvent},
|
||||||
|
interpret::{Interpret, InterpretError, Outcome},
|
||||||
|
inventory::Inventory,
|
||||||
|
topology::Topology,
|
||||||
|
};
|
||||||
|
|
||||||
|
#[async_trait]
|
||||||
pub trait Score<T: Topology>:
|
pub trait Score<T: Topology>:
|
||||||
std::fmt::Debug + ScoreToString<T> + Send + Sync + CloneBoxScore<T> + SerializeScore<T>
|
std::fmt::Debug + ScoreToString<T> + Send + Sync + CloneBoxScore<T> + SerializeScore<T>
|
||||||
{
|
{
|
||||||
fn create_interpret(&self) -> Box<dyn Interpret<T>>;
|
async fn interpret(
|
||||||
|
&self,
|
||||||
|
inventory: &Inventory,
|
||||||
|
topology: &T,
|
||||||
|
) -> Result<Outcome, InterpretError> {
|
||||||
|
let id = Id::default();
|
||||||
|
let interpret = self.create_interpret();
|
||||||
|
|
||||||
|
instrumentation::instrument(HarmonyEvent::InterpretExecutionStarted {
|
||||||
|
execution_id: id.clone().to_string(),
|
||||||
|
topology: topology.name().into(),
|
||||||
|
interpret: interpret.get_name().to_string(),
|
||||||
|
score: self.name(),
|
||||||
|
message: format!("{} running...", interpret.get_name()),
|
||||||
|
})
|
||||||
|
.unwrap();
|
||||||
|
let result = interpret.execute(inventory, topology).await;
|
||||||
|
|
||||||
|
instrumentation::instrument(HarmonyEvent::InterpretExecutionFinished {
|
||||||
|
execution_id: id.clone().to_string(),
|
||||||
|
topology: topology.name().into(),
|
||||||
|
interpret: interpret.get_name().to_string(),
|
||||||
|
score: self.name(),
|
||||||
|
outcome: result.clone(),
|
||||||
|
})
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
result
|
||||||
|
}
|
||||||
|
|
||||||
fn name(&self) -> String;
|
fn name(&self) -> String;
|
||||||
|
|
||||||
|
#[doc(hidden)]
|
||||||
|
fn create_interpret(&self) -> Box<dyn Interpret<T>>;
|
||||||
}
|
}
|
||||||
|
|
||||||
pub trait SerializeScore<T: Topology> {
|
pub trait SerializeScore<T: Topology> {
|
||||||
|
|||||||
@ -4,8 +4,6 @@ use harmony_types::net::MacAddress;
|
|||||||
use log::info;
|
use log::info;
|
||||||
|
|
||||||
use crate::executors::ExecutorError;
|
use crate::executors::ExecutorError;
|
||||||
use crate::interpret::InterpretError;
|
|
||||||
use crate::interpret::Outcome;
|
|
||||||
|
|
||||||
use super::DHCPStaticEntry;
|
use super::DHCPStaticEntry;
|
||||||
use super::DhcpServer;
|
use super::DhcpServer;
|
||||||
@ -19,6 +17,8 @@ use super::K8sclient;
|
|||||||
use super::LoadBalancer;
|
use super::LoadBalancer;
|
||||||
use super::LoadBalancerService;
|
use super::LoadBalancerService;
|
||||||
use super::LogicalHost;
|
use super::LogicalHost;
|
||||||
|
use super::PreparationError;
|
||||||
|
use super::PreparationOutcome;
|
||||||
use super::Router;
|
use super::Router;
|
||||||
use super::TftpServer;
|
use super::TftpServer;
|
||||||
|
|
||||||
@ -48,7 +48,7 @@ impl Topology for HAClusterTopology {
|
|||||||
fn name(&self) -> &str {
|
fn name(&self) -> &str {
|
||||||
"HAClusterTopology"
|
"HAClusterTopology"
|
||||||
}
|
}
|
||||||
async fn ensure_ready(&self) -> Result<Outcome, InterpretError> {
|
async fn ensure_ready(&self) -> Result<PreparationOutcome, PreparationError> {
|
||||||
todo!(
|
todo!(
|
||||||
"ensure_ready, not entirely sure what it should do here, probably something like verify that the hosts are reachable and all services are up and ready."
|
"ensure_ready, not entirely sure what it should do here, probably something like verify that the hosts are reachable and all services are up and ready."
|
||||||
)
|
)
|
||||||
@ -244,10 +244,12 @@ impl Topology for DummyInfra {
|
|||||||
todo!()
|
todo!()
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn ensure_ready(&self) -> Result<Outcome, InterpretError> {
|
async fn ensure_ready(&self) -> Result<PreparationOutcome, PreparationError> {
|
||||||
let dummy_msg = "This is a dummy infrastructure that does nothing";
|
let dummy_msg = "This is a dummy infrastructure that does nothing";
|
||||||
info!("{dummy_msg}");
|
info!("{dummy_msg}");
|
||||||
Ok(Outcome::success(dummy_msg.to_string()))
|
Ok(PreparationOutcome::Success {
|
||||||
|
details: dummy_msg.into(),
|
||||||
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@ -7,7 +7,7 @@ use tokio::sync::OnceCell;
|
|||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
executors::ExecutorError,
|
executors::ExecutorError,
|
||||||
interpret::{InterpretError, InterpretStatus, Outcome},
|
interpret::InterpretStatus,
|
||||||
inventory::Inventory,
|
inventory::Inventory,
|
||||||
modules::{
|
modules::{
|
||||||
k3d::K3DInstallationScore,
|
k3d::K3DInstallationScore,
|
||||||
@ -24,7 +24,8 @@ use crate::{
|
|||||||
};
|
};
|
||||||
|
|
||||||
use super::{
|
use super::{
|
||||||
DeploymentTarget, HelmCommand, K8sclient, MultiTargetTopology, Topology,
|
DeploymentTarget, HelmCommand, K8sclient, MultiTargetTopology, PreparationError,
|
||||||
|
PreparationOutcome, Topology,
|
||||||
k8s::K8sClient,
|
k8s::K8sClient,
|
||||||
oberservability::monitoring::AlertReceiver,
|
oberservability::monitoring::AlertReceiver,
|
||||||
tenant::{
|
tenant::{
|
||||||
@ -80,22 +81,30 @@ impl PrometheusApplicationMonitoring<CRDPrometheus> for K8sAnywhereTopology {
|
|||||||
sender: &CRDPrometheus,
|
sender: &CRDPrometheus,
|
||||||
inventory: &Inventory,
|
inventory: &Inventory,
|
||||||
receivers: Option<Vec<Box<dyn AlertReceiver<CRDPrometheus>>>>,
|
receivers: Option<Vec<Box<dyn AlertReceiver<CRDPrometheus>>>>,
|
||||||
) -> Result<Outcome, InterpretError> {
|
) -> Result<PreparationOutcome, PreparationError> {
|
||||||
let po_result = self.ensure_prometheus_operator(sender).await?;
|
let po_result = self.ensure_prometheus_operator(sender).await?;
|
||||||
|
|
||||||
if po_result.status == InterpretStatus::NOOP {
|
if po_result == PreparationOutcome::Noop {
|
||||||
debug!("Skipping Prometheus CR installation due to missing operator.");
|
debug!("Skipping Prometheus CR installation due to missing operator.");
|
||||||
return Ok(Outcome::noop());
|
return Ok(po_result);
|
||||||
}
|
}
|
||||||
self.get_k8s_prometheus_application_score(sender.clone(), receivers)
|
|
||||||
.await
|
|
||||||
.create_interpret()
|
|
||||||
.execute(inventory, self)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
Ok(Outcome::success(
|
let result = self
|
||||||
"No action, working on cluster ".to_string(),
|
.get_k8s_prometheus_application_score(sender.clone(), receivers)
|
||||||
))
|
.await
|
||||||
|
.interpret(inventory, self)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
match result {
|
||||||
|
Ok(outcome) => match outcome.status {
|
||||||
|
InterpretStatus::SUCCESS => Ok(PreparationOutcome::Success {
|
||||||
|
details: outcome.message,
|
||||||
|
}),
|
||||||
|
InterpretStatus::NOOP => Ok(PreparationOutcome::Noop),
|
||||||
|
_ => Err(PreparationError::new(outcome.message)),
|
||||||
|
},
|
||||||
|
Err(err) => Err(PreparationError::new(err.to_string())),
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -166,15 +175,23 @@ impl K8sAnywhereTopology {
|
|||||||
K3DInstallationScore::default()
|
K3DInstallationScore::default()
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn try_install_k3d(&self) -> Result<(), InterpretError> {
|
async fn try_install_k3d(&self) -> Result<(), PreparationError> {
|
||||||
self.get_k3d_installation_score()
|
let result = self
|
||||||
.create_interpret()
|
.get_k3d_installation_score()
|
||||||
.execute(&Inventory::empty(), self)
|
.interpret(&Inventory::empty(), self)
|
||||||
.await?;
|
.await;
|
||||||
Ok(())
|
|
||||||
|
match result {
|
||||||
|
Ok(outcome) => match outcome.status {
|
||||||
|
InterpretStatus::SUCCESS => Ok(()),
|
||||||
|
InterpretStatus::NOOP => Ok(()),
|
||||||
|
_ => Err(PreparationError::new(outcome.message)),
|
||||||
|
},
|
||||||
|
Err(err) => Err(PreparationError::new(err.to_string())),
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn try_get_or_install_k8s_client(&self) -> Result<Option<K8sState>, InterpretError> {
|
async fn try_get_or_install_k8s_client(&self) -> Result<Option<K8sState>, PreparationError> {
|
||||||
let k8s_anywhere_config = &self.config;
|
let k8s_anywhere_config = &self.config;
|
||||||
|
|
||||||
// TODO this deserves some refactoring, it is becoming a bit hard to figure out
|
// TODO this deserves some refactoring, it is becoming a bit hard to figure out
|
||||||
@ -193,7 +210,7 @@ impl K8sAnywhereTopology {
|
|||||||
}));
|
}));
|
||||||
}
|
}
|
||||||
None => {
|
None => {
|
||||||
return Err(InterpretError::new(format!(
|
return Err(PreparationError::new(format!(
|
||||||
"Failed to load kubeconfig from {kubeconfig}"
|
"Failed to load kubeconfig from {kubeconfig}"
|
||||||
)));
|
)));
|
||||||
}
|
}
|
||||||
@ -272,11 +289,11 @@ impl K8sAnywhereTopology {
|
|||||||
async fn ensure_prometheus_operator(
|
async fn ensure_prometheus_operator(
|
||||||
&self,
|
&self,
|
||||||
sender: &CRDPrometheus,
|
sender: &CRDPrometheus,
|
||||||
) -> Result<Outcome, InterpretError> {
|
) -> Result<PreparationOutcome, PreparationError> {
|
||||||
let status = Command::new("sh")
|
let status = Command::new("sh")
|
||||||
.args(["-c", "kubectl get crd -A | grep -i prometheuses"])
|
.args(["-c", "kubectl get crd -A | grep -i prometheuses"])
|
||||||
.status()
|
.status()
|
||||||
.map_err(|e| InterpretError::new(format!("could not connect to cluster: {}", e)))?;
|
.map_err(|e| PreparationError::new(format!("could not connect to cluster: {}", e)))?;
|
||||||
|
|
||||||
if !status.success() {
|
if !status.success() {
|
||||||
if let Some(Some(k8s_state)) = self.k8s_state.get() {
|
if let Some(Some(k8s_state)) = self.k8s_state.get() {
|
||||||
@ -285,30 +302,37 @@ impl K8sAnywhereTopology {
|
|||||||
debug!("installing prometheus operator");
|
debug!("installing prometheus operator");
|
||||||
let op_score =
|
let op_score =
|
||||||
prometheus_operator_helm_chart_score(sender.namespace.clone());
|
prometheus_operator_helm_chart_score(sender.namespace.clone());
|
||||||
op_score
|
let result = op_score.interpret(&Inventory::empty(), self).await;
|
||||||
.create_interpret()
|
|
||||||
.execute(&Inventory::empty(), self)
|
return match result {
|
||||||
.await?;
|
Ok(outcome) => match outcome.status {
|
||||||
return Ok(Outcome::success(
|
InterpretStatus::SUCCESS => Ok(PreparationOutcome::Success {
|
||||||
"installed prometheus operator".to_string(),
|
details: "installed prometheus operator".into(),
|
||||||
));
|
}),
|
||||||
|
InterpretStatus::NOOP => Ok(PreparationOutcome::Noop),
|
||||||
|
_ => Err(PreparationError::new(
|
||||||
|
"failed to install prometheus operator (unknown error)".into(),
|
||||||
|
)),
|
||||||
|
},
|
||||||
|
Err(err) => Err(PreparationError::new(err.to_string())),
|
||||||
|
};
|
||||||
}
|
}
|
||||||
K8sSource::Kubeconfig => {
|
K8sSource::Kubeconfig => {
|
||||||
debug!("unable to install prometheus operator, contact cluster admin");
|
debug!("unable to install prometheus operator, contact cluster admin");
|
||||||
return Ok(Outcome::noop());
|
return Ok(PreparationOutcome::Noop);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
warn!("Unable to detect k8s_state. Skipping Prometheus Operator install.");
|
warn!("Unable to detect k8s_state. Skipping Prometheus Operator install.");
|
||||||
return Ok(Outcome::noop());
|
return Ok(PreparationOutcome::Noop);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
debug!("Prometheus operator is already present, skipping install");
|
debug!("Prometheus operator is already present, skipping install");
|
||||||
|
|
||||||
Ok(Outcome::success(
|
Ok(PreparationOutcome::Success {
|
||||||
"prometheus operator present in cluster".to_string(),
|
details: "prometheus operator present in cluster".into(),
|
||||||
))
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -367,26 +391,25 @@ impl Topology for K8sAnywhereTopology {
|
|||||||
"K8sAnywhereTopology"
|
"K8sAnywhereTopology"
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn ensure_ready(&self) -> Result<Outcome, InterpretError> {
|
async fn ensure_ready(&self) -> Result<PreparationOutcome, PreparationError> {
|
||||||
let k8s_state = self
|
let k8s_state = self
|
||||||
.k8s_state
|
.k8s_state
|
||||||
.get_or_try_init(|| self.try_get_or_install_k8s_client())
|
.get_or_try_init(|| self.try_get_or_install_k8s_client())
|
||||||
.await?;
|
.await?;
|
||||||
|
|
||||||
let k8s_state: &K8sState = k8s_state.as_ref().ok_or(InterpretError::new(
|
let k8s_state: &K8sState = k8s_state.as_ref().ok_or(PreparationError::new(
|
||||||
"No K8s client could be found or installed".to_string(),
|
"no K8s client could be found or installed".to_string(),
|
||||||
))?;
|
))?;
|
||||||
|
|
||||||
self.ensure_k8s_tenant_manager(k8s_state)
|
self.ensure_k8s_tenant_manager(k8s_state)
|
||||||
.await
|
.await
|
||||||
.map_err(InterpretError::new)?;
|
.map_err(PreparationError::new)?;
|
||||||
|
|
||||||
match self.is_helm_available() {
|
match self.is_helm_available() {
|
||||||
Ok(()) => Ok(Outcome::success(format!(
|
Ok(()) => Ok(PreparationOutcome::Success {
|
||||||
"{} + helm available",
|
details: format!("{} + helm available", k8s_state.message.clone()),
|
||||||
k8s_state.message.clone()
|
}),
|
||||||
))),
|
Err(e) => Err(PreparationError::new(format!("helm unavailable: {}", e))),
|
||||||
Err(e) => Err(InterpretError::new(format!("helm unavailable: {}", e))),
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@ -1,9 +1,7 @@
|
|||||||
use async_trait::async_trait;
|
use async_trait::async_trait;
|
||||||
use derive_new::new;
|
use derive_new::new;
|
||||||
|
|
||||||
use crate::interpret::{InterpretError, Outcome};
|
use super::{HelmCommand, PreparationError, PreparationOutcome, Topology};
|
||||||
|
|
||||||
use super::{HelmCommand, Topology};
|
|
||||||
|
|
||||||
#[derive(new)]
|
#[derive(new)]
|
||||||
pub struct LocalhostTopology;
|
pub struct LocalhostTopology;
|
||||||
@ -14,10 +12,10 @@ impl Topology for LocalhostTopology {
|
|||||||
"LocalHostTopology"
|
"LocalHostTopology"
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn ensure_ready(&self) -> Result<Outcome, InterpretError> {
|
async fn ensure_ready(&self) -> Result<PreparationOutcome, PreparationError> {
|
||||||
Ok(Outcome::success(
|
Ok(PreparationOutcome::Success {
|
||||||
"Localhost is Chuck Norris, always ready.".to_string(),
|
details: "Localhost is Chuck Norris, always ready.".into(),
|
||||||
))
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@ -6,6 +6,7 @@ mod k8s_anywhere;
|
|||||||
mod localhost;
|
mod localhost;
|
||||||
pub mod oberservability;
|
pub mod oberservability;
|
||||||
pub mod tenant;
|
pub mod tenant;
|
||||||
|
use derive_new::new;
|
||||||
pub use k8s_anywhere::*;
|
pub use k8s_anywhere::*;
|
||||||
pub use localhost::*;
|
pub use localhost::*;
|
||||||
pub mod k8s;
|
pub mod k8s;
|
||||||
@ -26,10 +27,13 @@ pub use tftp::*;
|
|||||||
mod helm_command;
|
mod helm_command;
|
||||||
pub use helm_command::*;
|
pub use helm_command::*;
|
||||||
|
|
||||||
|
use super::{
|
||||||
|
executors::ExecutorError,
|
||||||
|
instrumentation::{self, HarmonyEvent},
|
||||||
|
};
|
||||||
|
use std::error::Error;
|
||||||
use std::net::IpAddr;
|
use std::net::IpAddr;
|
||||||
|
|
||||||
use super::interpret::{InterpretError, Outcome};
|
|
||||||
|
|
||||||
/// Represents a logical view of an infrastructure environment providing specific capabilities.
|
/// Represents a logical view of an infrastructure environment providing specific capabilities.
|
||||||
///
|
///
|
||||||
/// A Topology acts as a self-contained "package" responsible for managing access
|
/// A Topology acts as a self-contained "package" responsible for managing access
|
||||||
@ -57,9 +61,128 @@ pub trait Topology: Send + Sync {
|
|||||||
/// * **Internal Orchestration:** For complex topologies, this method might manage dependencies on other sub-topologies, ensuring *their* `ensure_ready` is called first. Using nested `Maestros` to run setup `Scores` against these sub-topologies is the recommended pattern for non-trivial bootstrapping, allowing reuse of Harmony's core orchestration logic.
|
/// * **Internal Orchestration:** For complex topologies, this method might manage dependencies on other sub-topologies, ensuring *their* `ensure_ready` is called first. Using nested `Maestros` to run setup `Scores` against these sub-topologies is the recommended pattern for non-trivial bootstrapping, allowing reuse of Harmony's core orchestration logic.
|
||||||
///
|
///
|
||||||
/// # Returns
|
/// # Returns
|
||||||
/// - `Ok(Outcome)`: Indicates the topology is now ready. The `Outcome` status might be `SUCCESS` if actions were taken, or `NOOP` if it was already ready. The message should provide context.
|
/// - `Ok(PreparationOutcome)`: Indicates the topology is now ready. The `Outcome` status might be `SUCCESS` if actions were taken, or `NOOP` if it was already ready. The message should provide context.
|
||||||
/// - `Err(TopologyError)`: Indicates the topology could not reach a ready state due to configuration issues, discovery failures, bootstrap errors, or unsupported environments.
|
/// - `Err(PreparationError)`: Indicates the topology could not reach a ready state due to configuration issues, discovery failures, bootstrap errors, or unsupported environments.
|
||||||
async fn ensure_ready(&self) -> Result<Outcome, InterpretError>;
|
async fn ensure_ready(&self) -> Result<PreparationOutcome, PreparationError>;
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, PartialEq, Eq)]
|
||||||
|
pub enum PreparationOutcome {
|
||||||
|
Success { details: String },
|
||||||
|
Noop,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, new)]
|
||||||
|
pub struct PreparationError {
|
||||||
|
msg: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl std::fmt::Display for PreparationError {
|
||||||
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
|
f.write_str(&self.msg)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Error for PreparationError {}
|
||||||
|
|
||||||
|
impl From<ExecutorError> for PreparationError {
|
||||||
|
fn from(value: ExecutorError) -> Self {
|
||||||
|
Self {
|
||||||
|
msg: format!("InterpretError : {value}"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<kube::Error> for PreparationError {
|
||||||
|
fn from(value: kube::Error) -> Self {
|
||||||
|
Self {
|
||||||
|
msg: format!("PreparationError : {value}"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<String> for PreparationError {
|
||||||
|
fn from(value: String) -> Self {
|
||||||
|
Self {
|
||||||
|
msg: format!("PreparationError : {value}"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, Debug, PartialEq)]
|
||||||
|
pub enum TopologyStatus {
|
||||||
|
Queued,
|
||||||
|
Preparing,
|
||||||
|
Success,
|
||||||
|
Noop,
|
||||||
|
Error,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct TopologyState {
|
||||||
|
pub topology: String,
|
||||||
|
pub status: TopologyStatus,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl TopologyState {
|
||||||
|
pub fn new(topology: String) -> Self {
|
||||||
|
let instance = Self {
|
||||||
|
topology,
|
||||||
|
status: TopologyStatus::Queued,
|
||||||
|
};
|
||||||
|
|
||||||
|
instrumentation::instrument(HarmonyEvent::TopologyStateChanged {
|
||||||
|
topology: instance.topology.clone(),
|
||||||
|
status: instance.status.clone(),
|
||||||
|
message: None,
|
||||||
|
})
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
instance
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn prepare(&mut self) {
|
||||||
|
self.status = TopologyStatus::Preparing;
|
||||||
|
|
||||||
|
instrumentation::instrument(HarmonyEvent::TopologyStateChanged {
|
||||||
|
topology: self.topology.clone(),
|
||||||
|
status: self.status.clone(),
|
||||||
|
message: None,
|
||||||
|
})
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn success(&mut self, message: String) {
|
||||||
|
self.status = TopologyStatus::Success;
|
||||||
|
|
||||||
|
instrumentation::instrument(HarmonyEvent::TopologyStateChanged {
|
||||||
|
topology: self.topology.clone(),
|
||||||
|
status: self.status.clone(),
|
||||||
|
message: Some(message),
|
||||||
|
})
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn noop(&mut self) {
|
||||||
|
self.status = TopologyStatus::Noop;
|
||||||
|
|
||||||
|
instrumentation::instrument(HarmonyEvent::TopologyStateChanged {
|
||||||
|
topology: self.topology.clone(),
|
||||||
|
status: self.status.clone(),
|
||||||
|
message: None,
|
||||||
|
})
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn error(&mut self, message: String) {
|
||||||
|
self.status = TopologyStatus::Error;
|
||||||
|
|
||||||
|
instrumentation::instrument(HarmonyEvent::TopologyStateChanged {
|
||||||
|
topology: self.topology.clone(),
|
||||||
|
status: self.status.clone(),
|
||||||
|
message: Some(message),
|
||||||
|
})
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
|
|||||||
@ -45,7 +45,7 @@ impl<S: AlertSender + Installable<T>, T: Topology> Interpret<T> for AlertingInte
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn get_name(&self) -> InterpretName {
|
fn get_name(&self) -> InterpretName {
|
||||||
todo!()
|
InterpretName::Alerting
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_version(&self) -> Version {
|
fn get_version(&self) -> Version {
|
||||||
|
|||||||
@ -15,7 +15,7 @@ use k8s_openapi::{
|
|||||||
apimachinery::pkg::util::intstr::IntOrString,
|
apimachinery::pkg::util::intstr::IntOrString,
|
||||||
};
|
};
|
||||||
use kube::Resource;
|
use kube::Resource;
|
||||||
use log::{debug, info, warn};
|
use log::debug;
|
||||||
use serde::de::DeserializeOwned;
|
use serde::de::DeserializeOwned;
|
||||||
use serde_json::json;
|
use serde_json::json;
|
||||||
use tokio::sync::OnceCell;
|
use tokio::sync::OnceCell;
|
||||||
@ -46,8 +46,7 @@ impl K8sTenantManager {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn ensure_constraints(&self, _namespace: &Namespace) -> Result<(), ExecutorError> {
|
fn ensure_constraints(&self, _namespace: &Namespace) -> Result<(), ExecutorError> {
|
||||||
warn!("Validate that when tenant already exists (by id) that name has not changed");
|
// TODO: Ensure constraints are applied to namespace (https://git.nationtech.io/NationTech/harmony/issues/98)
|
||||||
warn!("Make sure other Tenant constraints are respected by this k8s implementation");
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -437,13 +436,14 @@ impl TenantManager for K8sTenantManager {
|
|||||||
debug!("Creating network_policy for tenant {}", config.name);
|
debug!("Creating network_policy for tenant {}", config.name);
|
||||||
self.apply_resource(network_policy, config).await?;
|
self.apply_resource(network_policy, config).await?;
|
||||||
|
|
||||||
info!(
|
debug!(
|
||||||
"Success provisionning K8s tenant id {} name {}",
|
"Success provisionning K8s tenant id {} name {}",
|
||||||
config.id, config.name
|
config.id, config.name
|
||||||
);
|
);
|
||||||
self.store_config(config);
|
self.store_config(config);
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn get_tenant_config(&self) -> Option<TenantConfig> {
|
async fn get_tenant_config(&self) -> Option<TenantConfig> {
|
||||||
self.k8s_tenant_config.get().cloned()
|
self.k8s_tenant_config.get().cloned()
|
||||||
}
|
}
|
||||||
|
|||||||
@ -193,8 +193,7 @@ impl<
|
|||||||
})],
|
})],
|
||||||
};
|
};
|
||||||
score
|
score
|
||||||
.create_interpret()
|
.interpret(&Inventory::empty(), topology)
|
||||||
.execute(&Inventory::empty(), topology)
|
|
||||||
.await
|
.await
|
||||||
.unwrap();
|
.unwrap();
|
||||||
}
|
}
|
||||||
|
|||||||
@ -51,10 +51,7 @@ impl<T: Topology + K8sclient + HelmCommand> Interpret<T> for ArgoInterpret {
|
|||||||
topology: &T,
|
topology: &T,
|
||||||
) -> Result<Outcome, InterpretError> {
|
) -> Result<Outcome, InterpretError> {
|
||||||
error!("Uncomment below, only disabled for debugging");
|
error!("Uncomment below, only disabled for debugging");
|
||||||
self.score
|
self.score.interpret(inventory, topology).await?;
|
||||||
.create_interpret()
|
|
||||||
.execute(inventory, topology)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
let k8s_client = topology.k8s_client().await?;
|
let k8s_client = topology.k8s_client().await?;
|
||||||
k8s_client
|
k8s_client
|
||||||
@ -62,7 +59,7 @@ impl<T: Topology + K8sclient + HelmCommand> Interpret<T> for ArgoInterpret {
|
|||||||
.await
|
.await
|
||||||
.unwrap();
|
.unwrap();
|
||||||
Ok(Outcome::success(format!(
|
Ok(Outcome::success(format!(
|
||||||
"Successfully installed ArgoCD and {} Applications",
|
"ArgoCD installed with {} applications",
|
||||||
self.argo_apps.len()
|
self.argo_apps.len()
|
||||||
)))
|
)))
|
||||||
}
|
}
|
||||||
|
|||||||
@ -57,8 +57,7 @@ impl<
|
|||||||
namespace: namespace.clone(),
|
namespace: namespace.clone(),
|
||||||
host: "localhost".to_string(),
|
host: "localhost".to_string(),
|
||||||
};
|
};
|
||||||
ntfy.create_interpret()
|
ntfy.interpret(&Inventory::empty(), topology)
|
||||||
.execute(&Inventory::empty(), topology)
|
|
||||||
.await
|
.await
|
||||||
.expect("couldn't create interpret for ntfy");
|
.expect("couldn't create interpret for ntfy");
|
||||||
|
|
||||||
@ -95,8 +94,7 @@ impl<
|
|||||||
|
|
||||||
alerting_score.receivers.push(Box::new(ntfy_receiver));
|
alerting_score.receivers.push(Box::new(ntfy_receiver));
|
||||||
alerting_score
|
alerting_score
|
||||||
.create_interpret()
|
.interpret(&Inventory::empty(), topology)
|
||||||
.execute(&Inventory::empty(), topology)
|
|
||||||
.await
|
.await
|
||||||
.unwrap();
|
.unwrap();
|
||||||
Ok(())
|
Ok(())
|
||||||
|
|||||||
@ -60,7 +60,7 @@ impl<A: Application, T: Topology + std::fmt::Debug> Interpret<T> for Application
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
Ok(Outcome::success("successfully created app".to_string()))
|
Ok(Outcome::success("Application created".to_string()))
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_name(&self) -> InterpretName {
|
fn get_name(&self) -> InterpretName {
|
||||||
|
|||||||
@ -46,7 +46,7 @@ where
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn name(&self) -> String {
|
fn name(&self) -> String {
|
||||||
format!("Application: {}", self.application.name())
|
format!("{} [ApplicationScore]", self.application.name())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@ -55,7 +55,7 @@ impl<T: Topology + HelmCommand> Score<T> for HelmChartScore {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn name(&self) -> String {
|
fn name(&self) -> String {
|
||||||
format!("{} {} HelmChartScore", self.release_name, self.chart_name)
|
format!("{} [HelmChartScore]", self.release_name)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -225,24 +225,27 @@ impl<T: Topology + HelmCommand> Interpret<T> for HelmChartInterpret {
|
|||||||
match status {
|
match status {
|
||||||
helm_wrapper_rs::HelmDeployStatus::Deployed => Ok(Outcome::new(
|
helm_wrapper_rs::HelmDeployStatus::Deployed => Ok(Outcome::new(
|
||||||
InterpretStatus::SUCCESS,
|
InterpretStatus::SUCCESS,
|
||||||
"Helm Chart deployed".to_string(),
|
format!("Helm Chart {} deployed", self.score.release_name),
|
||||||
)),
|
)),
|
||||||
helm_wrapper_rs::HelmDeployStatus::PendingInstall => Ok(Outcome::new(
|
helm_wrapper_rs::HelmDeployStatus::PendingInstall => Ok(Outcome::new(
|
||||||
InterpretStatus::RUNNING,
|
InterpretStatus::RUNNING,
|
||||||
"Helm Chart Pending install".to_string(),
|
format!("Helm Chart {} pending install...", self.score.release_name),
|
||||||
)),
|
)),
|
||||||
helm_wrapper_rs::HelmDeployStatus::PendingUpgrade => Ok(Outcome::new(
|
helm_wrapper_rs::HelmDeployStatus::PendingUpgrade => Ok(Outcome::new(
|
||||||
InterpretStatus::RUNNING,
|
InterpretStatus::RUNNING,
|
||||||
"Helm Chart pending upgrade".to_string(),
|
format!("Helm Chart {} pending upgrade...", self.score.release_name),
|
||||||
)),
|
|
||||||
helm_wrapper_rs::HelmDeployStatus::Failed => Err(InterpretError::new(
|
|
||||||
"Failed to install helm chart".to_string(),
|
|
||||||
)),
|
)),
|
||||||
|
helm_wrapper_rs::HelmDeployStatus::Failed => Err(InterpretError::new(format!(
|
||||||
|
"Helm Chart {} installation failed",
|
||||||
|
self.score.release_name
|
||||||
|
))),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_name(&self) -> InterpretName {
|
fn get_name(&self) -> InterpretName {
|
||||||
todo!()
|
InterpretName::HelmChart
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_version(&self) -> Version {
|
fn get_version(&self) -> Version {
|
||||||
todo!()
|
todo!()
|
||||||
}
|
}
|
||||||
|
|||||||
@ -349,7 +349,7 @@ impl<T: Topology + K8sclient + HelmCommand> Interpret<T> for HelmChartInterpretV
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn get_name(&self) -> InterpretName {
|
fn get_name(&self) -> InterpretName {
|
||||||
todo!()
|
InterpretName::HelmCommand
|
||||||
}
|
}
|
||||||
fn get_version(&self) -> Version {
|
fn get_version(&self) -> Version {
|
||||||
todo!()
|
todo!()
|
||||||
|
|||||||
@ -7,7 +7,6 @@ use serde::Serialize;
|
|||||||
use crate::{
|
use crate::{
|
||||||
config::HARMONY_DATA_DIR,
|
config::HARMONY_DATA_DIR,
|
||||||
data::{Id, Version},
|
data::{Id, Version},
|
||||||
instrumentation::{self, HarmonyEvent},
|
|
||||||
interpret::{Interpret, InterpretError, InterpretName, InterpretStatus, Outcome},
|
interpret::{Interpret, InterpretError, InterpretName, InterpretStatus, Outcome},
|
||||||
inventory::Inventory,
|
inventory::Inventory,
|
||||||
score::Score,
|
score::Score,
|
||||||
@ -30,14 +29,14 @@ impl Default for K3DInstallationScore {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl<T: Topology> Score<T> for K3DInstallationScore {
|
impl<T: Topology> Score<T> for K3DInstallationScore {
|
||||||
fn create_interpret(&self) -> Box<dyn crate::interpret::Interpret<T>> {
|
fn create_interpret(&self) -> Box<dyn Interpret<T>> {
|
||||||
Box::new(K3dInstallationInterpret {
|
Box::new(K3dInstallationInterpret {
|
||||||
score: self.clone(),
|
score: self.clone(),
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
fn name(&self) -> String {
|
fn name(&self) -> String {
|
||||||
todo!()
|
"K3dInstallationScore".into()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -51,20 +50,14 @@ impl<T: Topology> Interpret<T> for K3dInstallationInterpret {
|
|||||||
async fn execute(
|
async fn execute(
|
||||||
&self,
|
&self,
|
||||||
_inventory: &Inventory,
|
_inventory: &Inventory,
|
||||||
topology: &T,
|
_topology: &T,
|
||||||
) -> Result<Outcome, InterpretError> {
|
) -> Result<Outcome, InterpretError> {
|
||||||
instrumentation::instrument(HarmonyEvent::InterpretExecutionStarted {
|
|
||||||
topology: topology.name().into(),
|
|
||||||
interpret: "k3d-installation".into(),
|
|
||||||
message: "installing k3d...".into(),
|
|
||||||
})
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
let k3d = k3d_rs::K3d::new(
|
let k3d = k3d_rs::K3d::new(
|
||||||
self.score.installation_path.clone(),
|
self.score.installation_path.clone(),
|
||||||
Some(self.score.cluster_name.clone()),
|
Some(self.score.cluster_name.clone()),
|
||||||
);
|
);
|
||||||
let outcome = match k3d.ensure_installed().await {
|
|
||||||
|
match k3d.ensure_installed().await {
|
||||||
Ok(_client) => {
|
Ok(_client) => {
|
||||||
let msg = format!("k3d cluster '{}' installed ", self.score.cluster_name);
|
let msg = format!("k3d cluster '{}' installed ", self.score.cluster_name);
|
||||||
debug!("{msg}");
|
debug!("{msg}");
|
||||||
@ -73,16 +66,7 @@ impl<T: Topology> Interpret<T> for K3dInstallationInterpret {
|
|||||||
Err(msg) => Err(InterpretError::new(format!(
|
Err(msg) => Err(InterpretError::new(format!(
|
||||||
"failed to ensure k3d is installed : {msg}"
|
"failed to ensure k3d is installed : {msg}"
|
||||||
))),
|
))),
|
||||||
};
|
}
|
||||||
|
|
||||||
instrumentation::instrument(HarmonyEvent::InterpretExecutionFinished {
|
|
||||||
topology: topology.name().into(),
|
|
||||||
interpret: "k3d-installation".into(),
|
|
||||||
outcome: outcome.clone(),
|
|
||||||
})
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
outcome
|
|
||||||
}
|
}
|
||||||
fn get_name(&self) -> InterpretName {
|
fn get_name(&self) -> InterpretName {
|
||||||
InterpretName::K3dInstallation
|
InterpretName::K3dInstallation
|
||||||
|
|||||||
@ -89,7 +89,7 @@ where
|
|||||||
))
|
))
|
||||||
}
|
}
|
||||||
fn get_name(&self) -> InterpretName {
|
fn get_name(&self) -> InterpretName {
|
||||||
todo!()
|
InterpretName::K8sResource
|
||||||
}
|
}
|
||||||
fn get_version(&self) -> Version {
|
fn get_version(&self) -> Version {
|
||||||
todo!()
|
todo!()
|
||||||
|
|||||||
@ -128,10 +128,7 @@ impl<T: Topology + K8sclient + HelmCommand> Interpret<T> for LAMPInterpret {
|
|||||||
|
|
||||||
info!("Deploying score {deployment_score:#?}");
|
info!("Deploying score {deployment_score:#?}");
|
||||||
|
|
||||||
deployment_score
|
deployment_score.interpret(inventory, topology).await?;
|
||||||
.create_interpret()
|
|
||||||
.execute(inventory, topology)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
info!("LAMP deployment_score {deployment_score:?}");
|
info!("LAMP deployment_score {deployment_score:?}");
|
||||||
|
|
||||||
@ -153,10 +150,7 @@ impl<T: Topology + K8sclient + HelmCommand> Interpret<T> for LAMPInterpret {
|
|||||||
.map(|nbs| fqdn!(nbs.to_string().as_str())),
|
.map(|nbs| fqdn!(nbs.to_string().as_str())),
|
||||||
};
|
};
|
||||||
|
|
||||||
lamp_ingress
|
lamp_ingress.interpret(inventory, topology).await?;
|
||||||
.create_interpret()
|
|
||||||
.execute(inventory, topology)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
info!("LAMP lamp_ingress {lamp_ingress:?}");
|
info!("LAMP lamp_ingress {lamp_ingress:?}");
|
||||||
|
|
||||||
@ -166,7 +160,7 @@ impl<T: Topology + K8sclient + HelmCommand> Interpret<T> for LAMPInterpret {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn get_name(&self) -> InterpretName {
|
fn get_name(&self) -> InterpretName {
|
||||||
todo!()
|
InterpretName::Lamp
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_version(&self) -> Version {
|
fn get_version(&self) -> Version {
|
||||||
@ -215,7 +209,7 @@ impl LAMPInterpret {
|
|||||||
repository: None,
|
repository: None,
|
||||||
};
|
};
|
||||||
|
|
||||||
score.create_interpret().execute(inventory, topology).await
|
score.interpret(inventory, topology).await
|
||||||
}
|
}
|
||||||
fn build_dockerfile(&self, score: &LAMPScore) -> Result<PathBuf, Box<dyn std::error::Error>> {
|
fn build_dockerfile(&self, score: &LAMPScore) -> Result<PathBuf, Box<dyn std::error::Error>> {
|
||||||
let mut dockerfile = Dockerfile::new();
|
let mut dockerfile = Dockerfile::new();
|
||||||
|
|||||||
@ -13,7 +13,7 @@ use crate::{
|
|||||||
prometheus::prometheus::PrometheusApplicationMonitoring,
|
prometheus::prometheus::PrometheusApplicationMonitoring,
|
||||||
},
|
},
|
||||||
score::Score,
|
score::Score,
|
||||||
topology::{Topology, oberservability::monitoring::AlertReceiver},
|
topology::{PreparationOutcome, Topology, oberservability::monitoring::AlertReceiver},
|
||||||
};
|
};
|
||||||
|
|
||||||
#[derive(Debug, Clone, Serialize)]
|
#[derive(Debug, Clone, Serialize)]
|
||||||
@ -33,7 +33,10 @@ impl<T: Topology + PrometheusApplicationMonitoring<CRDPrometheus>> Score<T>
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn name(&self) -> String {
|
fn name(&self) -> String {
|
||||||
"ApplicationMonitoringScore".to_string()
|
format!(
|
||||||
|
"{} monitoring [ApplicationMonitoringScore]",
|
||||||
|
self.application.name()
|
||||||
|
)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -51,17 +54,27 @@ impl<T: Topology + PrometheusApplicationMonitoring<CRDPrometheus>> Interpret<T>
|
|||||||
inventory: &Inventory,
|
inventory: &Inventory,
|
||||||
topology: &T,
|
topology: &T,
|
||||||
) -> Result<Outcome, InterpretError> {
|
) -> Result<Outcome, InterpretError> {
|
||||||
topology
|
let result = topology
|
||||||
.install_prometheus(
|
.install_prometheus(
|
||||||
&self.score.sender,
|
&self.score.sender,
|
||||||
inventory,
|
inventory,
|
||||||
Some(self.score.receivers.clone()),
|
Some(self.score.receivers.clone()),
|
||||||
)
|
)
|
||||||
.await
|
.await;
|
||||||
|
|
||||||
|
match result {
|
||||||
|
Ok(outcome) => match outcome {
|
||||||
|
PreparationOutcome::Success { details: _ } => {
|
||||||
|
Ok(Outcome::success("Prometheus installed".into()))
|
||||||
|
}
|
||||||
|
PreparationOutcome::Noop => Ok(Outcome::noop()),
|
||||||
|
},
|
||||||
|
Err(err) => Err(InterpretError::from(err)),
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_name(&self) -> InterpretName {
|
fn get_name(&self) -> InterpretName {
|
||||||
todo!()
|
InterpretName::ApplicationMonitoring
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_version(&self) -> Version {
|
fn get_version(&self) -> Version {
|
||||||
|
|||||||
@ -119,8 +119,7 @@ impl KubePrometheus {
|
|||||||
topology: &T,
|
topology: &T,
|
||||||
) -> Result<Outcome, InterpretError> {
|
) -> Result<Outcome, InterpretError> {
|
||||||
kube_prometheus_helm_chart_score(self.config.clone())
|
kube_prometheus_helm_chart_score(self.config.clone())
|
||||||
.create_interpret()
|
.interpret(inventory, topology)
|
||||||
.execute(inventory, topology)
|
|
||||||
.await
|
.await
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@ -28,7 +28,7 @@ impl<T: Topology + HelmCommand + K8sclient> Score<T> for NtfyScore {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn name(&self) -> String {
|
fn name(&self) -> String {
|
||||||
"Ntfy".to_string()
|
"alert receiver [NtfyScore]".into()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -96,8 +96,7 @@ impl<T: Topology + HelmCommand + K8sclient> Interpret<T> for NtfyInterpret {
|
|||||||
topology: &T,
|
topology: &T,
|
||||||
) -> Result<Outcome, InterpretError> {
|
) -> Result<Outcome, InterpretError> {
|
||||||
ntfy_helm_chart_score(self.score.namespace.clone(), self.score.host.clone())
|
ntfy_helm_chart_score(self.score.namespace.clone(), self.score.host.clone())
|
||||||
.create_interpret()
|
.interpret(inventory, topology)
|
||||||
.execute(inventory, topology)
|
|
||||||
.await?;
|
.await?;
|
||||||
|
|
||||||
debug!("installed ntfy helm chart");
|
debug!("installed ntfy helm chart");
|
||||||
@ -120,12 +119,13 @@ impl<T: Topology + HelmCommand + K8sclient> Interpret<T> for NtfyInterpret {
|
|||||||
|
|
||||||
debug!("exec into pod done");
|
debug!("exec into pod done");
|
||||||
|
|
||||||
Ok(Outcome::success("installed ntfy".to_string()))
|
Ok(Outcome::success("Ntfy installed".to_string()))
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_name(&self) -> InterpretName {
|
fn get_name(&self) -> InterpretName {
|
||||||
todo!()
|
InterpretName::Ntfy
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_version(&self) -> Version {
|
fn get_version(&self) -> Version {
|
||||||
todo!()
|
todo!()
|
||||||
}
|
}
|
||||||
|
|||||||
@ -100,8 +100,7 @@ impl Prometheus {
|
|||||||
topology: &T,
|
topology: &T,
|
||||||
) -> Result<Outcome, InterpretError> {
|
) -> Result<Outcome, InterpretError> {
|
||||||
prometheus_helm_chart_score(self.config.clone())
|
prometheus_helm_chart_score(self.config.clone())
|
||||||
.create_interpret()
|
.interpret(inventory, topology)
|
||||||
.execute(inventory, topology)
|
|
||||||
.await
|
.await
|
||||||
}
|
}
|
||||||
pub async fn install_grafana<T: Topology + HelmCommand + Send + Sync>(
|
pub async fn install_grafana<T: Topology + HelmCommand + Send + Sync>(
|
||||||
@ -116,8 +115,7 @@ impl Prometheus {
|
|||||||
|
|
||||||
if let Some(ns) = namespace.as_deref() {
|
if let Some(ns) = namespace.as_deref() {
|
||||||
grafana_helm_chart_score(ns)
|
grafana_helm_chart_score(ns)
|
||||||
.create_interpret()
|
.interpret(inventory, topology)
|
||||||
.execute(inventory, topology)
|
|
||||||
.await
|
.await
|
||||||
} else {
|
} else {
|
||||||
Err(InterpretError::new(
|
Err(InterpretError::new(
|
||||||
|
|||||||
@ -61,7 +61,7 @@ impl<T: Topology + K8sclient + PrometheusApplicationMonitoring<CRDPrometheus>> S
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn name(&self) -> String {
|
fn name(&self) -> String {
|
||||||
"CRDApplicationAlertingScore".into()
|
"prometheus alerting [CRDAlertingScore]".into()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -94,12 +94,12 @@ impl<T: Topology + K8sclient + PrometheusApplicationMonitoring<CRDPrometheus>> I
|
|||||||
self.install_monitors(self.service_monitors.clone(), &client)
|
self.install_monitors(self.service_monitors.clone(), &client)
|
||||||
.await?;
|
.await?;
|
||||||
Ok(Outcome::success(
|
Ok(Outcome::success(
|
||||||
"deployed application monitoring composants".to_string(),
|
"K8s monitoring components installed".to_string(),
|
||||||
))
|
))
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_name(&self) -> InterpretName {
|
fn get_name(&self) -> InterpretName {
|
||||||
todo!()
|
InterpretName::K8sPrometheusCrdAlerting
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_version(&self) -> Version {
|
fn get_version(&self) -> Version {
|
||||||
@ -118,7 +118,7 @@ impl<T: Topology + K8sclient + PrometheusApplicationMonitoring<CRDPrometheus>> I
|
|||||||
impl K8sPrometheusCRDAlertingInterpret {
|
impl K8sPrometheusCRDAlertingInterpret {
|
||||||
async fn crd_exists(&self, crd: &str) -> bool {
|
async fn crd_exists(&self, crd: &str) -> bool {
|
||||||
let status = Command::new("sh")
|
let status = Command::new("sh")
|
||||||
.args(["-c", "kubectl get crd -A | grep -i", crd])
|
.args(["-c", &format!("kubectl get crd -A | grep -i {crd}")])
|
||||||
.status()
|
.status()
|
||||||
.map_err(|e| InterpretError::new(format!("could not connect to cluster: {}", e)))
|
.map_err(|e| InterpretError::new(format!("could not connect to cluster: {}", e)))
|
||||||
.unwrap();
|
.unwrap();
|
||||||
|
|||||||
@ -1,9 +1,11 @@
|
|||||||
use async_trait::async_trait;
|
use async_trait::async_trait;
|
||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
interpret::{InterpretError, Outcome},
|
|
||||||
inventory::Inventory,
|
inventory::Inventory,
|
||||||
topology::oberservability::monitoring::{AlertReceiver, AlertSender},
|
topology::{
|
||||||
|
PreparationError, PreparationOutcome,
|
||||||
|
oberservability::monitoring::{AlertReceiver, AlertSender},
|
||||||
|
},
|
||||||
};
|
};
|
||||||
|
|
||||||
#[async_trait]
|
#[async_trait]
|
||||||
@ -13,5 +15,5 @@ pub trait PrometheusApplicationMonitoring<S: AlertSender> {
|
|||||||
sender: &S,
|
sender: &S,
|
||||||
inventory: &Inventory,
|
inventory: &Inventory,
|
||||||
receivers: Option<Vec<Box<dyn AlertReceiver<S>>>>,
|
receivers: Option<Vec<Box<dyn AlertReceiver<S>>>>,
|
||||||
) -> Result<Outcome, InterpretError>;
|
) -> Result<PreparationOutcome, PreparationError>;
|
||||||
}
|
}
|
||||||
|
|||||||
@ -17,7 +17,7 @@ impl<T: Topology + TenantCredentialManager> Score<T> for TenantCredentialScore {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn name(&self) -> String {
|
fn name(&self) -> String {
|
||||||
todo!()
|
"TenantCredentialScore".into()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@ -28,7 +28,7 @@ impl<T: Topology + TenantManager> Score<T> for TenantScore {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn name(&self) -> String {
|
fn name(&self) -> String {
|
||||||
format!("{} TenantScore", self.config.name)
|
format!("{} [TenantScore]", self.config.name)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -47,8 +47,8 @@ impl<T: Topology + TenantManager> Interpret<T> for TenantInterpret {
|
|||||||
topology.provision_tenant(&self.tenant_config).await?;
|
topology.provision_tenant(&self.tenant_config).await?;
|
||||||
|
|
||||||
Ok(Outcome::success(format!(
|
Ok(Outcome::success(format!(
|
||||||
"Successfully provisioned tenant {} with id {}",
|
"Tenant provisioned with id '{}'",
|
||||||
self.tenant_config.name, self.tenant_config.id
|
self.tenant_config.id
|
||||||
)))
|
)))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@ -5,6 +5,10 @@ version.workspace = true
|
|||||||
readme.workspace = true
|
readme.workspace = true
|
||||||
license.workspace = true
|
license.workspace = true
|
||||||
|
|
||||||
|
[features]
|
||||||
|
default = ["tui"]
|
||||||
|
tui = ["dep:harmony_tui"]
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
assert_cmd = "2.0.17"
|
assert_cmd = "2.0.17"
|
||||||
clap = { version = "4.5.35", features = ["derive"] }
|
clap = { version = "4.5.35", features = ["derive"] }
|
||||||
@ -19,7 +23,5 @@ lazy_static = "1.5.0"
|
|||||||
log.workspace = true
|
log.workspace = true
|
||||||
indicatif-log-bridge = "0.2.3"
|
indicatif-log-bridge = "0.2.3"
|
||||||
|
|
||||||
|
[dev-dependencies]
|
||||||
[features]
|
harmony = { path = "../harmony", features = ["testing"] }
|
||||||
default = ["tui"]
|
|
||||||
tui = ["dep:harmony_tui"]
|
|
||||||
|
|||||||
@ -1,16 +1,16 @@
|
|||||||
use harmony::instrumentation::{self, HarmonyEvent};
|
use harmony::{
|
||||||
use indicatif::{MultiProgress, ProgressBar};
|
instrumentation::{self, HarmonyEvent},
|
||||||
use indicatif_log_bridge::LogWrapper;
|
topology::TopologyStatus,
|
||||||
use std::{
|
|
||||||
collections::HashMap,
|
|
||||||
sync::{Arc, Mutex},
|
|
||||||
};
|
};
|
||||||
|
use indicatif::MultiProgress;
|
||||||
|
use indicatif_log_bridge::LogWrapper;
|
||||||
|
use std::sync::{Arc, Mutex};
|
||||||
|
|
||||||
use crate::progress;
|
use crate::progress::{IndicatifProgressTracker, ProgressTracker};
|
||||||
|
|
||||||
pub fn init() -> tokio::task::JoinHandle<()> {
|
pub fn init() -> tokio::task::JoinHandle<()> {
|
||||||
configure_logger();
|
let base_progress = configure_logger();
|
||||||
let handle = tokio::spawn(handle_events());
|
let handle = tokio::spawn(handle_events(base_progress));
|
||||||
|
|
||||||
loop {
|
loop {
|
||||||
if instrumentation::instrument(HarmonyEvent::HarmonyStarted).is_ok() {
|
if instrumentation::instrument(HarmonyEvent::HarmonyStarted).is_ok() {
|
||||||
@ -21,91 +21,144 @@ pub fn init() -> tokio::task::JoinHandle<()> {
|
|||||||
handle
|
handle
|
||||||
}
|
}
|
||||||
|
|
||||||
fn configure_logger() {
|
fn configure_logger() -> MultiProgress {
|
||||||
let logger =
|
let logger =
|
||||||
env_logger::Builder::from_env(env_logger::Env::default().default_filter_or("info")).build();
|
env_logger::Builder::from_env(env_logger::Env::default().default_filter_or("info")).build();
|
||||||
let level = logger.filter();
|
let level = logger.filter();
|
||||||
let multi = MultiProgress::new();
|
let progress = MultiProgress::new();
|
||||||
LogWrapper::new(multi.clone(), logger).try_init().unwrap();
|
|
||||||
|
LogWrapper::new(progress.clone(), logger)
|
||||||
|
.try_init()
|
||||||
|
.unwrap();
|
||||||
log::set_max_level(level);
|
log::set_max_level(level);
|
||||||
|
|
||||||
|
progress
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn handle_events() {
|
async fn handle_events(base_progress: MultiProgress) {
|
||||||
instrumentation::subscribe("Harmony CLI Logger", {
|
let progress_tracker = Arc::new(IndicatifProgressTracker::new(base_progress.clone()));
|
||||||
let sections: Arc<Mutex<HashMap<String, MultiProgress>>> =
|
let preparing_topology = Arc::new(Mutex::new(false));
|
||||||
Arc::new(Mutex::new(HashMap::new()));
|
let current_score: Arc<Mutex<Option<String>>> = Arc::new(Mutex::new(None));
|
||||||
let progress_bars: Arc<Mutex<HashMap<String, ProgressBar>>> =
|
|
||||||
Arc::new(Mutex::new(HashMap::new()));
|
|
||||||
|
|
||||||
|
instrumentation::subscribe("Harmony CLI Logger", {
|
||||||
move |event| {
|
move |event| {
|
||||||
let sections_clone = Arc::clone(§ions);
|
let progress_tracker = Arc::clone(&progress_tracker);
|
||||||
let progress_bars_clone = Arc::clone(&progress_bars);
|
let preparing_topology = Arc::clone(&preparing_topology);
|
||||||
|
let current_score = Arc::clone(¤t_score);
|
||||||
|
|
||||||
async move {
|
async move {
|
||||||
let mut sections = sections_clone.lock().unwrap();
|
let mut preparing_topology = preparing_topology.lock().unwrap();
|
||||||
let mut progress_bars = progress_bars_clone.lock().unwrap();
|
let mut current_score = current_score.lock().unwrap();
|
||||||
|
|
||||||
match event {
|
match event {
|
||||||
HarmonyEvent::HarmonyStarted => {}
|
HarmonyEvent::HarmonyStarted => {}
|
||||||
HarmonyEvent::PrepareTopologyStarted { topology: name } => {
|
HarmonyEvent::HarmonyFinished => {
|
||||||
let section = progress::new_section(format!(
|
progress_tracker.add_section(
|
||||||
"{} Preparing environment: {name}...",
|
"harmony-summary",
|
||||||
crate::theme::EMOJI_TOPOLOGY,
|
&format!("\n{} Harmony completed\n\n", crate::theme::EMOJI_HARMONY),
|
||||||
));
|
);
|
||||||
(*sections).insert(name, section);
|
progress_tracker.add_section("harmony-finished", "\n\n");
|
||||||
|
return false;
|
||||||
}
|
}
|
||||||
HarmonyEvent::TopologyPrepared {
|
HarmonyEvent::TopologyStateChanged {
|
||||||
topology: name,
|
topology,
|
||||||
outcome,
|
status,
|
||||||
|
message,
|
||||||
} => {
|
} => {
|
||||||
let section = (*sections).get(&name).unwrap();
|
let section_key = topology_key(&topology);
|
||||||
let progress = progress::add_spinner(section, "".into());
|
|
||||||
|
|
||||||
match outcome.status {
|
match status {
|
||||||
harmony::interpret::InterpretStatus::SUCCESS => {
|
TopologyStatus::Queued => {}
|
||||||
progress::success(section, Some(progress), outcome.message);
|
TopologyStatus::Preparing => {
|
||||||
|
progress_tracker.add_section(
|
||||||
|
§ion_key,
|
||||||
|
&format!(
|
||||||
|
"\n{} Preparing environment: {topology}...",
|
||||||
|
crate::theme::EMOJI_TOPOLOGY
|
||||||
|
),
|
||||||
|
);
|
||||||
|
(*preparing_topology) = true;
|
||||||
}
|
}
|
||||||
harmony::interpret::InterpretStatus::FAILURE => {
|
TopologyStatus::Success => {
|
||||||
progress::error(section, Some(progress), outcome.message);
|
(*preparing_topology) = false;
|
||||||
|
progress_tracker.add_task(§ion_key, "topology-success", "");
|
||||||
|
progress_tracker
|
||||||
|
.finish_task("topology-success", &message.unwrap_or("".into()));
|
||||||
}
|
}
|
||||||
harmony::interpret::InterpretStatus::RUNNING => todo!(),
|
TopologyStatus::Noop => {
|
||||||
harmony::interpret::InterpretStatus::QUEUED => todo!(),
|
(*preparing_topology) = false;
|
||||||
harmony::interpret::InterpretStatus::BLOCKED => todo!(),
|
progress_tracker.add_task(§ion_key, "topology-skip", "");
|
||||||
harmony::interpret::InterpretStatus::NOOP => {
|
progress_tracker
|
||||||
progress::skip(section, Some(progress), outcome.message);
|
.skip_task("topology-skip", &message.unwrap_or("".into()));
|
||||||
|
}
|
||||||
|
TopologyStatus::Error => {
|
||||||
|
progress_tracker.add_task(§ion_key, "topology-error", "");
|
||||||
|
(*preparing_topology) = false;
|
||||||
|
progress_tracker
|
||||||
|
.fail_task("topology-error", &message.unwrap_or("".into()));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
HarmonyEvent::InterpretExecutionStarted {
|
HarmonyEvent::InterpretExecutionStarted {
|
||||||
interpret: name,
|
execution_id: task_key,
|
||||||
topology,
|
topology,
|
||||||
|
interpret: _,
|
||||||
|
score,
|
||||||
message,
|
message,
|
||||||
} => {
|
} => {
|
||||||
let section = (*sections).get(&topology).unwrap();
|
let is_key_topology = (*preparing_topology)
|
||||||
let progress_bar = progress::add_spinner(section, message);
|
&& progress_tracker.contains_section(&topology_key(&topology));
|
||||||
|
let is_key_current_score = current_score.is_some()
|
||||||
|
&& progress_tracker
|
||||||
|
.contains_section(&score_key(¤t_score.clone().unwrap()));
|
||||||
|
let is_key_score = progress_tracker.contains_section(&score_key(&score));
|
||||||
|
|
||||||
(*progress_bars).insert(name, progress_bar);
|
let section_key = if is_key_topology {
|
||||||
|
topology_key(&topology)
|
||||||
|
} else if is_key_current_score {
|
||||||
|
score_key(¤t_score.clone().unwrap())
|
||||||
|
} else if is_key_score {
|
||||||
|
score_key(&score)
|
||||||
|
} else {
|
||||||
|
(*current_score) = Some(score.clone());
|
||||||
|
let key = score_key(&score);
|
||||||
|
progress_tracker.add_section(
|
||||||
|
&key,
|
||||||
|
&format!(
|
||||||
|
"{} Interpreting score: {score}...",
|
||||||
|
crate::theme::EMOJI_SCORE
|
||||||
|
),
|
||||||
|
);
|
||||||
|
key
|
||||||
|
};
|
||||||
|
|
||||||
|
progress_tracker.add_task(§ion_key, &task_key, &message);
|
||||||
}
|
}
|
||||||
HarmonyEvent::InterpretExecutionFinished {
|
HarmonyEvent::InterpretExecutionFinished {
|
||||||
topology,
|
execution_id: task_key,
|
||||||
interpret: name,
|
topology: _,
|
||||||
|
interpret: _,
|
||||||
|
score,
|
||||||
outcome,
|
outcome,
|
||||||
} => {
|
} => {
|
||||||
let section = (*sections).get(&topology).unwrap();
|
if current_score.is_some() && current_score.clone().unwrap() == score {
|
||||||
let progress_bar = (*progress_bars).get(&name).cloned();
|
(*current_score) = None;
|
||||||
|
|
||||||
let _ = section.clear();
|
|
||||||
|
|
||||||
match outcome {
|
|
||||||
Ok(outcome) => {
|
|
||||||
progress::success(section, progress_bar, outcome.message);
|
|
||||||
}
|
|
||||||
Err(err) => {
|
|
||||||
progress::error(section, progress_bar, err.to_string());
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
(*progress_bars).remove(&name);
|
match outcome {
|
||||||
|
Ok(outcome) => match outcome.status {
|
||||||
|
harmony::interpret::InterpretStatus::SUCCESS => {
|
||||||
|
progress_tracker.finish_task(&task_key, &outcome.message);
|
||||||
|
}
|
||||||
|
harmony::interpret::InterpretStatus::NOOP => {
|
||||||
|
progress_tracker.skip_task(&task_key, &outcome.message);
|
||||||
|
}
|
||||||
|
_ => progress_tracker.fail_task(&task_key, &outcome.message),
|
||||||
|
},
|
||||||
|
Err(err) => {
|
||||||
|
progress_tracker.fail_task(&task_key, &err.to_string());
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
true
|
true
|
||||||
@ -114,3 +167,11 @@ async fn handle_events() {
|
|||||||
})
|
})
|
||||||
.await;
|
.await;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn topology_key(topology: &str) -> String {
|
||||||
|
format!("topology-{topology}")
|
||||||
|
}
|
||||||
|
|
||||||
|
fn score_key(score: &str) -> String {
|
||||||
|
format!("score-{score}")
|
||||||
|
}
|
||||||
|
|||||||
@ -1,5 +1,6 @@
|
|||||||
use clap::Parser;
|
use clap::Parser;
|
||||||
use clap::builder::ArgPredicate;
|
use clap::builder::ArgPredicate;
|
||||||
|
use harmony::instrumentation;
|
||||||
use harmony::inventory::Inventory;
|
use harmony::inventory::Inventory;
|
||||||
use harmony::maestro::Maestro;
|
use harmony::maestro::Maestro;
|
||||||
use harmony::{score::Score, topology::Topology};
|
use harmony::{score::Score, topology::Topology};
|
||||||
@ -97,6 +98,7 @@ pub async fn run<T: Topology + Send + Sync + 'static>(
|
|||||||
|
|
||||||
let result = init(maestro, args_struct).await;
|
let result = init(maestro, args_struct).await;
|
||||||
|
|
||||||
|
instrumentation::instrument(instrumentation::HarmonyEvent::HarmonyFinished).unwrap();
|
||||||
let _ = tokio::try_join!(cli_logger_handle);
|
let _ = tokio::try_join!(cli_logger_handle);
|
||||||
result
|
result
|
||||||
}
|
}
|
||||||
@ -163,7 +165,7 @@ async fn init<T: Topology + Send + Sync + 'static>(
|
|||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod test {
|
mod tests {
|
||||||
use harmony::{
|
use harmony::{
|
||||||
inventory::Inventory,
|
inventory::Inventory,
|
||||||
maestro::Maestro,
|
maestro::Maestro,
|
||||||
|
|||||||
@ -1,50 +1,163 @@
|
|||||||
|
use indicatif::{MultiProgress, ProgressBar};
|
||||||
|
use std::collections::HashMap;
|
||||||
|
use std::sync::{Arc, Mutex};
|
||||||
use std::time::Duration;
|
use std::time::Duration;
|
||||||
|
|
||||||
use indicatif::{MultiProgress, ProgressBar};
|
pub trait ProgressTracker: Send + Sync {
|
||||||
|
fn contains_section(&self, id: &str) -> bool;
|
||||||
pub fn new_section(title: String) -> MultiProgress {
|
fn add_section(&self, id: &str, message: &str);
|
||||||
let multi_progress = MultiProgress::new();
|
fn add_task(&self, section_id: &str, task_id: &str, message: &str);
|
||||||
let _ = multi_progress.println(title);
|
fn finish_task(&self, id: &str, message: &str);
|
||||||
|
fn fail_task(&self, id: &str, message: &str);
|
||||||
multi_progress
|
fn skip_task(&self, id: &str, message: &str);
|
||||||
|
fn clear(&self);
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn add_spinner(multi_progress: &MultiProgress, message: String) -> ProgressBar {
|
struct Section {
|
||||||
let progress = multi_progress.add(ProgressBar::new_spinner());
|
header_index: usize,
|
||||||
|
task_count: usize,
|
||||||
progress.set_style(crate::theme::SPINNER_STYLE.clone());
|
pb: ProgressBar,
|
||||||
progress.set_message(message);
|
|
||||||
progress.enable_steady_tick(Duration::from_millis(100));
|
|
||||||
|
|
||||||
progress
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn success(multi_progress: &MultiProgress, progress: Option<ProgressBar>, message: String) {
|
struct IndicatifProgressTrackerState {
|
||||||
if let Some(progress) = progress {
|
sections: HashMap<String, Section>,
|
||||||
multi_progress.remove(&progress)
|
tasks: HashMap<String, ProgressBar>,
|
||||||
|
pb_count: usize,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
|
pub struct IndicatifProgressTracker {
|
||||||
|
mp: MultiProgress,
|
||||||
|
state: Arc<Mutex<IndicatifProgressTrackerState>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl IndicatifProgressTracker {
|
||||||
|
pub fn new(base: MultiProgress) -> Self {
|
||||||
|
// The indicatif log bridge will insert a progress bar at the top.
|
||||||
|
// To prevent our first section from being erased, we need to create
|
||||||
|
// a dummy progress bar as our first progress bar.
|
||||||
|
let _ = base.clear();
|
||||||
|
let log_pb = base.add(ProgressBar::new(1));
|
||||||
|
|
||||||
|
let mut sections = HashMap::new();
|
||||||
|
sections.insert(
|
||||||
|
"__log__".into(),
|
||||||
|
Section {
|
||||||
|
header_index: 0,
|
||||||
|
task_count: 0,
|
||||||
|
pb: log_pb.clone(),
|
||||||
|
},
|
||||||
|
);
|
||||||
|
|
||||||
|
let mut tasks = HashMap::new();
|
||||||
|
tasks.insert("__log__".into(), log_pb);
|
||||||
|
|
||||||
|
let state = Arc::new(Mutex::new(IndicatifProgressTrackerState {
|
||||||
|
sections,
|
||||||
|
tasks,
|
||||||
|
pb_count: 1,
|
||||||
|
}));
|
||||||
|
|
||||||
|
Self { mp: base, state }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ProgressTracker for IndicatifProgressTracker {
|
||||||
|
fn add_section(&self, id: &str, message: &str) {
|
||||||
|
let mut state = self.state.lock().unwrap();
|
||||||
|
|
||||||
|
let header_pb = self
|
||||||
|
.mp
|
||||||
|
.add(ProgressBar::new(1).with_style(crate::theme::SECTION_STYLE.clone()));
|
||||||
|
header_pb.finish_with_message(message.to_string());
|
||||||
|
|
||||||
|
let header_index = state.pb_count;
|
||||||
|
state.pb_count += 1;
|
||||||
|
|
||||||
|
state.sections.insert(
|
||||||
|
id.to_string(),
|
||||||
|
Section {
|
||||||
|
header_index,
|
||||||
|
task_count: 0,
|
||||||
|
pb: header_pb,
|
||||||
|
},
|
||||||
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
let progress = multi_progress.add(ProgressBar::new_spinner());
|
fn add_task(&self, section_id: &str, task_id: &str, message: &str) {
|
||||||
progress.set_style(crate::theme::SUCCESS_SPINNER_STYLE.clone());
|
let mut state = self.state.lock().unwrap();
|
||||||
progress.finish_with_message(message);
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn error(multi_progress: &MultiProgress, progress: Option<ProgressBar>, message: String) {
|
let insertion_index = {
|
||||||
if let Some(progress) = progress {
|
let current_section = state
|
||||||
multi_progress.remove(&progress)
|
.sections
|
||||||
|
.get(section_id)
|
||||||
|
.expect("Section ID not found");
|
||||||
|
current_section.header_index + current_section.task_count + 1 // +1 to insert after header
|
||||||
|
};
|
||||||
|
|
||||||
|
let pb = self.mp.insert(insertion_index, ProgressBar::new_spinner());
|
||||||
|
pb.set_style(crate::theme::SPINNER_STYLE.clone());
|
||||||
|
pb.set_prefix(" ");
|
||||||
|
pb.set_message(message.to_string());
|
||||||
|
pb.enable_steady_tick(Duration::from_millis(80));
|
||||||
|
|
||||||
|
state.pb_count += 1;
|
||||||
|
|
||||||
|
let section = state
|
||||||
|
.sections
|
||||||
|
.get_mut(section_id)
|
||||||
|
.expect("Section ID not found");
|
||||||
|
section.task_count += 1;
|
||||||
|
|
||||||
|
// We inserted a new progress bar, so we must update the header_index
|
||||||
|
// for all subsequent sections.
|
||||||
|
for (id, s) in state.sections.iter_mut() {
|
||||||
|
if id != section_id && s.header_index >= insertion_index {
|
||||||
|
s.header_index += 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
state.tasks.insert(task_id.to_string(), pb);
|
||||||
}
|
}
|
||||||
|
|
||||||
let progress = multi_progress.add(ProgressBar::new_spinner());
|
fn finish_task(&self, id: &str, message: &str) {
|
||||||
progress.set_style(crate::theme::ERROR_SPINNER_STYLE.clone());
|
let state = self.state.lock().unwrap();
|
||||||
progress.finish_with_message(message);
|
if let Some(pb) = state.tasks.get(id) {
|
||||||
}
|
pb.set_style(crate::theme::SUCCESS_SPINNER_STYLE.clone());
|
||||||
|
pb.finish_with_message(message.to_string());
|
||||||
pub fn skip(multi_progress: &MultiProgress, progress: Option<ProgressBar>, message: String) {
|
}
|
||||||
if let Some(progress) = progress {
|
|
||||||
multi_progress.remove(&progress)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
let progress = multi_progress.add(ProgressBar::new_spinner());
|
fn fail_task(&self, id: &str, message: &str) {
|
||||||
progress.set_style(crate::theme::SKIP_SPINNER_STYLE.clone());
|
let state = self.state.lock().unwrap();
|
||||||
progress.finish_with_message(message);
|
if let Some(pb) = state.tasks.get(id) {
|
||||||
|
pb.set_style(crate::theme::ERROR_SPINNER_STYLE.clone());
|
||||||
|
pb.finish_with_message(message.to_string());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn skip_task(&self, id: &str, message: &str) {
|
||||||
|
let state = self.state.lock().unwrap();
|
||||||
|
if let Some(pb) = state.tasks.get(id) {
|
||||||
|
pb.set_style(crate::theme::SKIP_SPINNER_STYLE.clone());
|
||||||
|
pb.finish_with_message(message.to_string());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn contains_section(&self, id: &str) -> bool {
|
||||||
|
let state = self.state.lock().unwrap();
|
||||||
|
state.sections.contains_key(id)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn clear(&self) {
|
||||||
|
let mut state = self.state.lock().unwrap();
|
||||||
|
|
||||||
|
state.tasks.values().for_each(|p| self.mp.remove(p));
|
||||||
|
state.tasks.clear();
|
||||||
|
state.sections.values().for_each(|s| self.mp.remove(&s.pb));
|
||||||
|
state.sections.clear();
|
||||||
|
state.pb_count = 0;
|
||||||
|
|
||||||
|
let _ = self.mp.clear();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@ -8,10 +8,14 @@ pub static EMOJI_SKIP: Emoji<'_, '_> = Emoji("⏭️", "");
|
|||||||
pub static EMOJI_ERROR: Emoji<'_, '_> = Emoji("⚠️", "");
|
pub static EMOJI_ERROR: Emoji<'_, '_> = Emoji("⚠️", "");
|
||||||
pub static EMOJI_DEPLOY: Emoji<'_, '_> = Emoji("🚀", "");
|
pub static EMOJI_DEPLOY: Emoji<'_, '_> = Emoji("🚀", "");
|
||||||
pub static EMOJI_TOPOLOGY: Emoji<'_, '_> = Emoji("📦", "");
|
pub static EMOJI_TOPOLOGY: Emoji<'_, '_> = Emoji("📦", "");
|
||||||
|
pub static EMOJI_SCORE: Emoji<'_, '_> = Emoji("🎶", "");
|
||||||
|
|
||||||
lazy_static! {
|
lazy_static! {
|
||||||
|
pub static ref SECTION_STYLE: ProgressStyle = ProgressStyle::default_spinner()
|
||||||
|
.template("{wide_msg:.bold}")
|
||||||
|
.unwrap();
|
||||||
pub static ref SPINNER_STYLE: ProgressStyle = ProgressStyle::default_spinner()
|
pub static ref SPINNER_STYLE: ProgressStyle = ProgressStyle::default_spinner()
|
||||||
.template(" {spinner:.green} {msg}")
|
.template(" {spinner:.green} {wide_msg}")
|
||||||
.unwrap()
|
.unwrap()
|
||||||
.tick_strings(&["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"]);
|
.tick_strings(&["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"]);
|
||||||
pub static ref SUCCESS_SPINNER_STYLE: ProgressStyle = SPINNER_STYLE
|
pub static ref SUCCESS_SPINNER_STYLE: ProgressStyle = SPINNER_STYLE
|
||||||
|
|||||||
@ -1,10 +1,7 @@
|
|||||||
use indicatif::{MultiProgress, ProgressBar};
|
use harmony_cli::progress::{IndicatifProgressTracker, ProgressTracker};
|
||||||
use indicatif_log_bridge::LogWrapper;
|
use indicatif::MultiProgress;
|
||||||
use log::error;
|
use log::error;
|
||||||
use std::{
|
use std::sync::Arc;
|
||||||
collections::HashMap,
|
|
||||||
sync::{Arc, Mutex},
|
|
||||||
};
|
|
||||||
|
|
||||||
use crate::instrumentation::{self, HarmonyComposerEvent};
|
use crate::instrumentation::{self, HarmonyComposerEvent};
|
||||||
|
|
||||||
@ -22,85 +19,57 @@ pub fn init() -> tokio::task::JoinHandle<()> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn configure_logger() {
|
fn configure_logger() {
|
||||||
let logger =
|
env_logger::Builder::from_env(env_logger::Env::default().default_filter_or("info")).build();
|
||||||
env_logger::Builder::from_env(env_logger::Env::default().default_filter_or("info")).build();
|
|
||||||
let level = logger.filter();
|
|
||||||
let multi = MultiProgress::new();
|
|
||||||
LogWrapper::new(multi.clone(), logger).try_init().unwrap();
|
|
||||||
log::set_max_level(level);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn handle_events() {
|
pub async fn handle_events() {
|
||||||
const PROGRESS_SETUP: &str = "project-initialization";
|
let progress_tracker = Arc::new(IndicatifProgressTracker::new(MultiProgress::new()));
|
||||||
|
|
||||||
|
const SETUP_SECTION: &str = "project-initialization";
|
||||||
|
const COMPILTATION_TASK: &str = "compilation";
|
||||||
const PROGRESS_DEPLOYMENT: &str = "deployment";
|
const PROGRESS_DEPLOYMENT: &str = "deployment";
|
||||||
|
|
||||||
instrumentation::subscribe("Harmony Composer Logger", {
|
instrumentation::subscribe("Harmony Composer Logger", {
|
||||||
let progresses: Arc<Mutex<HashMap<String, MultiProgress>>> =
|
|
||||||
Arc::new(Mutex::new(HashMap::new()));
|
|
||||||
let compilation_progress = Arc::new(Mutex::new(None::<ProgressBar>));
|
|
||||||
|
|
||||||
move |event| {
|
move |event| {
|
||||||
let progresses_clone = Arc::clone(&progresses);
|
let progress_tracker = Arc::clone(&progress_tracker);
|
||||||
let compilation_progress_clone = Arc::clone(&compilation_progress);
|
|
||||||
|
|
||||||
async move {
|
async move {
|
||||||
let mut progresses_guard = progresses_clone.lock().unwrap();
|
|
||||||
let mut compilation_progress_guard = compilation_progress_clone.lock().unwrap();
|
|
||||||
|
|
||||||
match event {
|
match event {
|
||||||
HarmonyComposerEvent::HarmonyComposerStarted => {}
|
HarmonyComposerEvent::HarmonyComposerStarted => {}
|
||||||
HarmonyComposerEvent::ProjectInitializationStarted => {
|
HarmonyComposerEvent::ProjectInitializationStarted => {
|
||||||
let multi_progress = harmony_cli::progress::new_section(format!(
|
progress_tracker.add_section(
|
||||||
"{} Initializing Harmony project...",
|
SETUP_SECTION,
|
||||||
harmony_cli::theme::EMOJI_HARMONY,
|
&format!(
|
||||||
));
|
"{} Initializing Harmony project...",
|
||||||
(*progresses_guard).insert(PROGRESS_SETUP.to_string(), multi_progress);
|
harmony_cli::theme::EMOJI_HARMONY,
|
||||||
|
),
|
||||||
|
);
|
||||||
}
|
}
|
||||||
HarmonyComposerEvent::ProjectInitialized => println!("\n"),
|
HarmonyComposerEvent::ProjectInitialized => {}
|
||||||
HarmonyComposerEvent::ProjectCompilationStarted { details } => {
|
HarmonyComposerEvent::ProjectCompilationStarted { details } => {
|
||||||
let initialization_progress =
|
progress_tracker.add_task(SETUP_SECTION, COMPILTATION_TASK, &details);
|
||||||
(*progresses_guard).get(PROGRESS_SETUP).unwrap();
|
|
||||||
let _ = initialization_progress.clear();
|
|
||||||
|
|
||||||
let progress =
|
|
||||||
harmony_cli::progress::add_spinner(initialization_progress, details);
|
|
||||||
*compilation_progress_guard = Some(progress);
|
|
||||||
}
|
}
|
||||||
HarmonyComposerEvent::ProjectCompiled => {
|
HarmonyComposerEvent::ProjectCompiled => {
|
||||||
let initialization_progress =
|
progress_tracker.finish_task(COMPILTATION_TASK, "project compiled");
|
||||||
(*progresses_guard).get(PROGRESS_SETUP).unwrap();
|
|
||||||
|
|
||||||
harmony_cli::progress::success(
|
|
||||||
initialization_progress,
|
|
||||||
(*compilation_progress_guard).take(),
|
|
||||||
"project compiled".to_string(),
|
|
||||||
);
|
|
||||||
}
|
}
|
||||||
HarmonyComposerEvent::ProjectCompilationFailed { details } => {
|
HarmonyComposerEvent::ProjectCompilationFailed { details } => {
|
||||||
let initialization_progress =
|
progress_tracker.fail_task(COMPILTATION_TASK, "failed to compile project");
|
||||||
(*progresses_guard).get(PROGRESS_SETUP).unwrap();
|
|
||||||
|
|
||||||
harmony_cli::progress::error(
|
|
||||||
initialization_progress,
|
|
||||||
(*compilation_progress_guard).take(),
|
|
||||||
"failed to compile project".to_string(),
|
|
||||||
);
|
|
||||||
|
|
||||||
error!("{details}");
|
error!("{details}");
|
||||||
}
|
}
|
||||||
HarmonyComposerEvent::DeploymentStarted { target } => {
|
HarmonyComposerEvent::DeploymentStarted { target } => {
|
||||||
let multi_progress = harmony_cli::progress::new_section(format!(
|
progress_tracker.add_section(
|
||||||
"{} Starting deployment to {target}...\n\n",
|
PROGRESS_DEPLOYMENT,
|
||||||
harmony_cli::theme::EMOJI_DEPLOY
|
&format!(
|
||||||
));
|
"\n{} Deploying project to {target}...\n",
|
||||||
(*progresses_guard).insert(PROGRESS_DEPLOYMENT.to_string(), multi_progress);
|
harmony_cli::theme::EMOJI_DEPLOY,
|
||||||
|
),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
HarmonyComposerEvent::DeploymentCompleted => {
|
||||||
|
progress_tracker.clear();
|
||||||
}
|
}
|
||||||
HarmonyComposerEvent::DeploymentCompleted => println!("\n"),
|
|
||||||
HarmonyComposerEvent::Shutdown => {
|
HarmonyComposerEvent::Shutdown => {
|
||||||
for (_, progresses) in (*progresses_guard).iter() {
|
|
||||||
progresses.clear().unwrap();
|
|
||||||
}
|
|
||||||
|
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@ -23,9 +23,18 @@ static HARMONY_COMPOSER_EVENT_BUS: Lazy<broadcast::Sender<HarmonyComposerEvent>>
|
|||||||
});
|
});
|
||||||
|
|
||||||
pub fn instrument(event: HarmonyComposerEvent) -> Result<(), &'static str> {
|
pub fn instrument(event: HarmonyComposerEvent) -> Result<(), &'static str> {
|
||||||
match HARMONY_COMPOSER_EVENT_BUS.send(event) {
|
#[cfg(not(test))]
|
||||||
Ok(_) => Ok(()),
|
{
|
||||||
Err(_) => Err("send error: no subscribers"),
|
match HARMONY_COMPOSER_EVENT_BUS.send(event) {
|
||||||
|
Ok(_) => Ok(()),
|
||||||
|
Err(_) => Err("send error: no subscribers"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
{
|
||||||
|
let _ = event; // Suppress the "unused variable" warning for `event`
|
||||||
|
Ok(())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@ -20,7 +20,7 @@ mod instrumentation;
|
|||||||
#[derive(Parser)]
|
#[derive(Parser)]
|
||||||
#[command(version, about, long_about = None, flatten_help = true, propagate_version = true)]
|
#[command(version, about, long_about = None, flatten_help = true, propagate_version = true)]
|
||||||
struct GlobalArgs {
|
struct GlobalArgs {
|
||||||
#[arg(long, default_value = "harmony")]
|
#[arg(long, default_value = ".")]
|
||||||
harmony_path: String,
|
harmony_path: String,
|
||||||
|
|
||||||
#[arg(long)]
|
#[arg(long)]
|
||||||
|
|||||||
@ -1,5 +1,5 @@
|
|||||||
use futures_util::StreamExt;
|
use futures_util::StreamExt;
|
||||||
use log::{debug, info, warn};
|
use log::{debug, warn};
|
||||||
use sha2::{Digest, Sha256};
|
use sha2::{Digest, Sha256};
|
||||||
use std::io::Read;
|
use std::io::Read;
|
||||||
use std::path::PathBuf;
|
use std::path::PathBuf;
|
||||||
@ -45,7 +45,7 @@ pub(crate) struct DownloadableAsset {
|
|||||||
impl DownloadableAsset {
|
impl DownloadableAsset {
|
||||||
fn verify_checksum(&self, file: PathBuf) -> bool {
|
fn verify_checksum(&self, file: PathBuf) -> bool {
|
||||||
if !file.exists() {
|
if !file.exists() {
|
||||||
warn!("File does not exist: {:?}", file);
|
debug!("File does not exist: {:?}", file);
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -155,7 +155,7 @@ impl DownloadableAsset {
|
|||||||
return Err(CHECKSUM_FAILED_MSG.to_string());
|
return Err(CHECKSUM_FAILED_MSG.to_string());
|
||||||
}
|
}
|
||||||
|
|
||||||
info!(
|
debug!(
|
||||||
"File downloaded and verified successfully: {}",
|
"File downloaded and verified successfully: {}",
|
||||||
target_file_path.to_string_lossy()
|
target_file_path.to_string_lossy()
|
||||||
);
|
);
|
||||||
|
|||||||
@ -64,7 +64,6 @@ impl K3d {
|
|||||||
.text()
|
.text()
|
||||||
.await
|
.await
|
||||||
.unwrap();
|
.unwrap();
|
||||||
println!("body: {body}");
|
|
||||||
|
|
||||||
let checksum = body
|
let checksum = body
|
||||||
.lines()
|
.lines()
|
||||||
@ -104,8 +103,7 @@ impl K3d {
|
|||||||
.get_latest()
|
.get_latest()
|
||||||
.await
|
.await
|
||||||
.map_err(|e| e.to_string())?;
|
.map_err(|e| e.to_string())?;
|
||||||
// debug!("Got k3d releases {releases:#?}");
|
debug!("Got k3d releases {latest_release:#?}");
|
||||||
println!("Got k3d first releases {latest_release:#?}");
|
|
||||||
|
|
||||||
Ok(latest_release)
|
Ok(latest_release)
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user