Skip to content

Commit

Permalink
fix(iroh-cli): store console files in subdirectory of iroh data dir
Browse files Browse the repository at this point in the history
## Description

We used to store the console files (history and current author) in a
subdirectory `console` under the iroh data dir. This got lost in some
refacorings, and the console files currently are created in the iroh
data directory directly.

This is bad because the iroh data directory is a single namespace, and
we should use it cautiously.

This PR goes back to using a `console` subdirectory. Existing files are
migrated on first start of the console.

Based on #2299 because this already made the environment init functions
async.

## Breaking Changes

<!-- Optional, if there are any breaking changes document them,
including how to migrate older code. -->

## Notes & open questions

<!-- Any notes, remarks or open questions you have to make about the PR.
-->

## Change checklist

- [ ] Self-review.
- [ ] Documentation updates if relevant.
- [ ] Tests if relevant.
- [ ] All breaking changes documented.
  • Loading branch information
Frando authored May 23, 2024
1 parent 62423aa commit 0b8188c
Show file tree
Hide file tree
Showing 2 changed files with 55 additions and 41 deletions.
2 changes: 1 addition & 1 deletion iroh-cli/src/commands/console.rs
Original file line number Diff line number Diff line change
Expand Up @@ -53,7 +53,7 @@ impl Repl {
pub fn run(self) -> anyhow::Result<()> {
let mut rl =
DefaultEditor::with_config(Config::builder().check_cursor_position(true).build())?;
let history_path = ConsolePaths::History.with_root(self.env.iroh_data_dir());
let history_path = ConsolePaths::History.with_iroh_data_dir(self.env.iroh_data_dir());
rl.load_history(&history_path).ok();
loop {
// prepare a channel to receive a signal from the main thread when a command completed
Expand Down
94 changes: 54 additions & 40 deletions iroh-cli/src/config.rs
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
//! Configuration for the iroh CLI.
use std::{
env, fmt,
env,
net::SocketAddr,
path::{Path, PathBuf},
str::FromStr,
Expand All @@ -21,6 +21,7 @@ use iroh::{
use parking_lot::RwLock;
use quic_rpc::ServiceConnection;
use serde::{Deserialize, Serialize};
use tracing::warn;

const ENV_AUTHOR: &str = "IROH_AUTHOR";
const ENV_DOC: &str = "IROH_DOC";
Expand All @@ -30,47 +31,20 @@ const ENV_FILE_RUST_LOG: &str = "IROH_FILE_RUST_LOG";
/// CONFIG_FILE_NAME is the name of the optional config file located in the iroh home directory
pub(crate) const CONFIG_FILE_NAME: &str = "iroh.config.toml";

#[derive(Debug, Clone, Copy)]
#[derive(Debug, Clone, Copy, Eq, PartialEq, strum::AsRefStr, strum::EnumString, strum::Display)]
pub(crate) enum ConsolePaths {
DefaultAuthor,
#[strum(serialize = "current-author")]
CurrentAuthor,
#[strum(serialize = "history")]
History,
}

impl From<&ConsolePaths> for &'static str {
fn from(value: &ConsolePaths) -> Self {
match value {
ConsolePaths::DefaultAuthor => "default_author.pubkey",
ConsolePaths::History => "history",
}
}
}
impl FromStr for ConsolePaths {
type Err = anyhow::Error;
fn from_str(s: &str) -> Result<Self> {
Ok(match s {
"default_author.pubkey" => Self::DefaultAuthor,
"history" => Self::History,
_ => bail!("unknown file or directory"),
})
}
}

impl fmt::Display for ConsolePaths {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
let s: &str = self.into();
write!(f, "{s}")
}
}
impl AsRef<Path> for ConsolePaths {
fn as_ref(&self) -> &Path {
let s: &str = self.into();
Path::new(s)
}
}

impl ConsolePaths {
pub fn with_root(self, root: impl AsRef<Path>) -> PathBuf {
PathBuf::from(root.as_ref()).join(self)
fn root(iroh_data_dir: impl AsRef<Path>) -> PathBuf {
PathBuf::from(iroh_data_dir.as_ref()).join("console")
}
pub fn with_iroh_data_dir(self, iroh_data_dir: impl AsRef<Path>) -> PathBuf {
Self::root(iroh_data_dir).join(self.as_ref())
}
}

Expand Down Expand Up @@ -163,6 +137,18 @@ impl ConsoleEnv {
iroh_data_dir: PathBuf,
iroh: &Iroh<C>,
) -> Result<Self> {
let console_data_dir = ConsolePaths::root(&iroh_data_dir);
tokio::fs::create_dir_all(&console_data_dir)
.await
.with_context(|| {
format!(
"failed to create console data directory at `{}`",
console_data_dir.to_string_lossy()
)
})?;

Self::migrate_console_files_016_017(&iroh_data_dir).await?;

let configured_author = Self::get_console_default_author(&iroh_data_dir)?;
let author = env_author(configured_author, iroh).await?;
let env = ConsoleEnvInner {
Expand All @@ -189,8 +175,8 @@ impl ConsoleEnv {
Ok(Self(Arc::new(RwLock::new(env))))
}

fn get_console_default_author(root: &Path) -> anyhow::Result<Option<AuthorId>> {
let author_path = ConsolePaths::DefaultAuthor.with_root(root);
fn get_console_default_author(iroh_data_root: &Path) -> anyhow::Result<Option<AuthorId>> {
let author_path = ConsolePaths::CurrentAuthor.with_iroh_data_dir(iroh_data_root);
if let Ok(s) = std::fs::read_to_string(&author_path) {
let author = AuthorId::from_str(&s).with_context(|| {
format!(
Expand Down Expand Up @@ -219,7 +205,7 @@ impl ConsoleEnv {
/// Will error if not running in the Iroh console.
/// Will persist to a file in the Iroh data dir otherwise.
pub(crate) fn set_author(&self, author: AuthorId) -> anyhow::Result<()> {
let author_path = ConsolePaths::DefaultAuthor.with_root(self.iroh_data_dir());
let author_path = ConsolePaths::CurrentAuthor.with_iroh_data_dir(self.iroh_data_dir());
let mut inner = self.0.write();
if !inner.is_console {
bail!("Switching the author is only supported within the Iroh console, not on the command line");
Expand Down Expand Up @@ -262,6 +248,34 @@ impl ConsoleEnv {
let inner = self.0.read();
inner.author
}

pub(crate) async fn migrate_console_files_016_017(iroh_data_dir: &Path) -> Result<()> {
// In iroh up to 0.16, we stored console settings directly in the data directory. Starting
// from 0.17, they live in a subdirectory and have new paths.
let old_current_author = iroh_data_dir.join("default_author.pubkey");
if old_current_author.is_file() {
if let Err(err) = tokio::fs::rename(
&old_current_author,
ConsolePaths::CurrentAuthor.with_iroh_data_dir(iroh_data_dir),
)
.await
{
warn!(path=%old_current_author.to_string_lossy(), "failed to migrate the console's current author file: {err}");
}
}
let old_history = iroh_data_dir.join("history");
if old_history.is_file() {
if let Err(err) = tokio::fs::rename(
&old_history,
ConsolePaths::History.with_iroh_data_dir(iroh_data_dir),
)
.await
{
warn!(path=%old_history.to_string_lossy(), "failed to migrate the console's history file: {err}");
}
}
Ok(())
}
}

async fn env_author<C: ServiceConnection<RpcService>>(
Expand Down

0 comments on commit 0b8188c

Please sign in to comment.