atelier_sdk/

config.rs

use std::env;
use std::fmt;
use std::fs;
use std::path::{Path, PathBuf};

use serde::{Deserialize, Serialize};

use crate::error::{Error, config_err};

/// The actor a workspace attributes its snapshots and journal entries to.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct Actor {
    /// The actor's display name, as it appears in the journal.
    pub name: String,
    /// What kind of actor this is.
    pub kind: ActorKind,
}

/// What kind of actor acted: a person, an AI agent, or an automation.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
#[serde(rename_all = "lowercase")]
pub enum ActorKind {
    /// A person.
    Human,
    /// An AI agent.
    Agent,
    /// An unattended process: a watcher, a bot, a pipeline.
    Automation,
}

impl ActorKind {
    /// The kind's canonical lowercase name, as stored and rendered.
    #[must_use]
    pub fn as_str(self) -> &'static str {
        match self {
            Self::Human => "human",
            Self::Agent => "agent",
            Self::Automation => "automation",
        }
    }
}

impl fmt::Display for ActorKind {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        f.write_str(self.as_str())
    }
}

impl std::str::FromStr for ActorKind {
    type Err = Error;

    fn from_str(text: &str) -> Result<Self, Self::Err> {
        match text {
            "human" => Ok(Self::Human),
            "agent" => Ok(Self::Agent),
            "automation" => Ok(Self::Automation),
            other => Err(Error::Config(format!("unknown actor kind: {other}"))),
        }
    }
}

#[derive(Debug, Deserialize)]
struct ActorFile {
    actor: Option<ActorSection>,
}

#[derive(Debug, Deserialize)]
struct ActorSection {
    name: String,
    kind: ActorKind,
}

/// Resolve the actor from the first config home that applies.
///
/// Order: `$ATELIER_CONFIG_HOME/config.toml`, else
/// `$XDG_CONFIG_HOME/atelier/config.toml`, else
/// `~/.config/atelier/config.toml`. A missing file or a file without an
/// `[actor]` section yields [`Error::NoActorConfigured`].
pub fn resolve_actor() -> Result<Actor, Error> {
    let Some(path) = actor_config_path() else {
        return Err(Error::NoActorConfigured);
    };
    if !path.is_file() {
        return Err(Error::NoActorConfigured);
    }
    let text = fs::read_to_string(&path)?;
    let parsed: ActorFile = toml::from_str(&text).map_err(config_err)?;
    match parsed.actor {
        Some(section) => Ok(Actor {
            name: section.name,
            kind: section.kind,
        }),
        None => Err(Error::NoActorConfigured),
    }
}

fn actor_config_path() -> Option<PathBuf> {
    if let Ok(home) = env::var("ATELIER_CONFIG_HOME") {
        return Some(PathBuf::from(home).join("config.toml"));
    }
    if let Ok(xdg) = env::var("XDG_CONFIG_HOME") {
        return Some(PathBuf::from(xdg).join("atelier").join("config.toml"));
    }
    if let Ok(home) = env::var("HOME") {
        return Some(
            PathBuf::from(home)
                .join(".config")
                .join("atelier")
                .join("config.toml"),
        );
    }
    None
}

/// An external origin the workspace is attached to, as held in memory and as
/// persisted under `[[source]]` in `.atelier/config.toml`. The mount names
/// the subdirectory whose engine carries the source's history; `/` is the
/// v1 root import — content folded into source zero, no engine of its own.
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
pub struct Source {
    /// The kind of origin the source is.
    pub kind: SourceKind,
    /// The origin's location: a folder path, a git repository path, or a
    /// bucket URL for remote sources.
    pub path: PathBuf,
    /// How content moves between the workspace and the source.
    pub sync: SyncPolicy,
    /// The subdirectory whose engine carries the source's history; `/`
    /// marks a v1 root import.
    pub mount: String,
    /// The git branch adoption found checked out; every landing on this
    /// source moves it, so plain `git push` carries the shared line.
    /// Absent for folder sources and detached-HEAD adoptions.
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub branch: Option<String>,
}

/// The mount value of a v1 root import.
pub(crate) const ROOT_MOUNT: &str = "/";

/// The kind of origin a source is.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
#[serde(rename_all = "kebab-case")]
pub enum SourceKind {
    /// A plain folder; content mirrors both ways on landings.
    LocalFolder,
    /// A git repository; adoption preserves its history and landings move
    /// its branch.
    LocalGit,
    /// A bucket prefix behind the remote adapter (ADR-0012): s3://, gs://,
    /// az://, or file:// for tests. The source path holds the URL.
    Remote,
}

impl fmt::Display for SourceKind {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            Self::LocalFolder => f.write_str("local-folder"),
            Self::LocalGit => f.write_str("local-git"),
            Self::Remote => f.write_str("remote"),
        }
    }
}

/// How content moves between a workspace and its source.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
#[serde(rename_all = "kebab-case")]
pub enum SyncPolicy {
    /// Changes land back into the source and source changes fold in.
    TwoWay,
}

impl fmt::Display for SyncPolicy {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            Self::TwoWay => f.write_str("two-way"),
        }
    }
}

/// The on-disk `.atelier/config.toml` describing one workspace.
#[derive(Debug, Serialize, Deserialize)]
pub struct WorkspaceConfig {
    pub schema: u32,
    pub workspace: WorkspaceSection,
    #[serde(default)]
    pub landing: LandingPolicy,
    #[serde(default)]
    pub journal: JournalPolicy,
    #[serde(default, rename = "source")]
    pub sources: Vec<Source>,
}

#[derive(Debug, Serialize, Deserialize)]
pub struct WorkspaceSection {
    pub name: String,
}

/// The `[landing]` policy: what a landing request needs before its change
/// lands (ADR-0007). The default profile takes one approval, allows
/// self-approval, and dismisses approvals when the change gains a snapshot.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
#[serde(default)]
pub struct LandingPolicy {
    /// How many approvals a landing request needs.
    pub approvals: u32,
    /// Whether the requester may approve their own request.
    pub allow_self_approve: bool,
    /// Whether a new snapshot on the change dismisses standing approvals.
    pub dismiss_approvals_on_new_snapshots: bool,
}

impl Default for LandingPolicy {
    fn default() -> Self {
        Self {
            approvals: 1,
            allow_self_approve: true,
            dismiss_approvals_on_new_snapshots: true,
        }
    }
}

/// The `[journal]` policy: how much of a session's instruction the journal
/// keeps (ADR-0004).
#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
#[serde(default)]
pub struct JournalPolicy {
    /// How much of a session's instruction the journal keeps.
    pub instruction_fidelity: InstructionFidelity,
}

impl Default for JournalPolicy {
    fn default() -> Self {
        Self {
            instruction_fidelity: InstructionFidelity::Summary,
        }
    }
}

/// What the journal records of an instruction: the summary plus run
/// reference, or additionally the verbatim body (audit profiles).
#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
#[serde(rename_all = "lowercase")]
pub enum InstructionFidelity {
    /// The summary and run reference only.
    Summary,
    /// Additionally the instruction's verbatim body.
    Verbatim,
}

impl WorkspaceConfig {
    pub fn new(name: String) -> Self {
        Self {
            schema: 1,
            workspace: WorkspaceSection { name },
            landing: LandingPolicy::default(),
            journal: JournalPolicy::default(),
            sources: Vec::new(),
        }
    }
}

/// Read `.atelier/config.toml` from a workspace's control directory.
pub fn read_workspace_config(atelier_dir: &Path) -> Result<WorkspaceConfig, Error> {
    let text = fs::read_to_string(atelier_dir.join("config.toml"))?;
    toml::from_str(&text).map_err(config_err)
}

/// Write `.atelier/config.toml` into a workspace's control directory.
pub fn write_workspace_config(atelier_dir: &Path, config: &WorkspaceConfig) -> Result<(), Error> {
    let text = toml::to_string(config).map_err(config_err)?;
    fs::write(atelier_dir.join("config.toml"), text)?;
    Ok(())
}