Files
movies-diary/crates/tui/src/main.rs

313 lines
12 KiB
Rust

use std::sync::Arc;
use std::time::Duration;
use tokio::sync::mpsc;
use ratatui::crossterm::event::{self, Event, KeyCode, KeyModifiers};
use tui::app::{
self, Action, App, BulkImportStage, Command, Screen, SettingsField, Tab,
};
use tui::client::ApiClient;
use tui::config::Config;
fn main() -> anyhow::Result<()> {
Config::init_keyring()?;
tokio::runtime::Builder::new_multi_thread()
.enable_all()
.build()?
.block_on(run())
}
async fn run() -> anyhow::Result<()> {
let mut config = Config::load();
// env var override
if let Ok(url) = std::env::var("MOVIES_API_URL") {
match &mut config {
Some(c) => c.api_url = url,
None => config = Some(Config { api_url: url }),
}
}
let initial_url = config.as_ref().map(|c| c.api_url.as_str()).unwrap_or("http://localhost:3000");
let client = Arc::new(ApiClient::new(initial_url));
let saved_token = tokio::task::spawn_blocking(Config::load_token).await.unwrap_or(None);
let mut app = App::new(config, saved_token.clone());
let (tx, mut rx) = mpsc::channel::<Action>(64);
let mut terminal = ratatui::init();
// If we start directly in Main (saved token), trigger an initial diary load
if matches!(app.screen, Screen::Main(_)) {
if let Some(token) = &saved_token {
let c = client.clone();
let t = token.clone();
let tx2 = tx.clone();
tokio::spawn(async move {
let action = match c.get_diary(&t, 0, 20).await {
Ok(r) => Action::DiaryLoaded { entries: r.items, total: r.total_count },
Err(e) => Action::DiaryLoadFailed(e.to_string()),
};
let _ = tx2.send(action).await;
});
}
}
let result = async {
loop {
terminal.draw(|f| tui::ui::render(f, &app))?;
// Poll keyboard — non-blocking with short timeout
if event::poll(Duration::from_millis(50))? {
if let Event::Key(key) = event::read()? {
if key.kind != ratatui::crossterm::event::KeyEventKind::Press {
continue;
}
if let Some(action) = key_to_action(&app, key) {
if matches!(action, Action::Quit) {
break;
}
let cmds = app::update(&mut app, action);
for cmd in cmds {
handle_command(cmd, &app, &client, &tx);
}
}
}
}
// Drain async results
while let Ok(action) = rx.try_recv() {
let cmds = app::update(&mut app, action);
for cmd in cmds {
handle_command(cmd, &app, &client, &tx);
}
}
}
Ok::<(), anyhow::Error>(())
}.await;
ratatui::restore();
result
}
// ── Command executor ──────────────────────────────────────────────────────────
fn handle_command(cmd: Command, app: &App, client: &Arc<ApiClient>, tx: &mpsc::Sender<Action>) {
match cmd {
Command::SaveConfig(url) => {
let config = Config { api_url: url.clone() };
if let Err(e) = config.save() {
let tx2 = tx.clone();
let msg = format!("Failed to save config: {e}");
tokio::spawn(async move { let _ = tx2.send(Action::DiaryLoadFailed(msg)).await; });
}
client.update_url(&url);
}
Command::SaveToken(token) => {
let tx2 = tx.clone();
tokio::spawn(async move {
if let Err(e) = tokio::task::spawn_blocking(move || Config::save_token(&token))
.await
.unwrap_or_else(|e| Err(anyhow::anyhow!(e)))
{
let msg = format!("Token not saved to keychain: {e}");
let _ = tx2.send(Action::DiaryLoadFailed(msg)).await;
}
});
}
Command::ClearToken => {
tokio::spawn(async {
let _ = tokio::task::spawn_blocking(Config::clear_token).await;
});
}
Command::Login { email, password } => {
let c = client.clone();
let tx = tx.clone();
tokio::spawn(async move {
let action = match c.login(&email, &password).await {
Ok(r) => Action::AuthOk(r.token),
Err(e) => Action::AuthFail(e.to_string()),
};
let _ = tx.send(action).await;
});
}
Command::LoadDiary { offset } => {
let Some(token) = app.token.clone() else { return };
let c = client.clone();
let tx = tx.clone();
tokio::spawn(async move {
let action = match c.get_diary(&token, offset, 20).await {
Ok(r) => Action::DiaryLoaded { entries: r.items, total: r.total_count },
Err(e) => Action::DiaryLoadFailed(e.to_string()),
};
let _ = tx.send(action).await;
});
}
Command::LoadHistory { movie_id } => {
let Some(token) = app.token.clone() else { return };
let c = client.clone();
let tx = tx.clone();
tokio::spawn(async move {
let action = match c.get_movie_history(&token, movie_id).await {
Ok(r) => Action::HistoryLoaded(r),
Err(e) => Action::HistoryLoadFailed(e.to_string()),
};
let _ = tx.send(action).await;
});
}
Command::CreateReview(req) => {
let Some(token) = app.token.clone() else { return };
let c = client.clone();
let tx = tx.clone();
tokio::spawn(async move {
let action = match c.create_review(&token, &req).await {
Ok(()) => Action::ReviewCreated,
Err(e) => Action::ReviewCreateFailed(e.to_string()),
};
let _ = tx.send(action).await;
});
}
Command::DeleteReview(id) => {
let Some(token) = app.token.clone() else { return };
let c = client.clone();
let tx = tx.clone();
tokio::spawn(async move {
let action = match c.delete_review(&token, id).await {
Ok(()) => Action::ReviewDeleted(id),
Err(e) => Action::ReviewDeleteFailed(e.to_string()),
};
let _ = tx.send(action).await;
});
}
Command::ImportNext(index) => {
let Some(token) = app.token.clone() else { return };
let req = match &app.screen {
Screen::Main(m) => match m.bulk_import.valid_requests.get(index) {
Some(r) => r.clone(),
None => return,
},
_ => return,
};
let c = client.clone();
let tx = tx.clone();
tokio::spawn(async move {
let error = c.create_review(&token, &req).await.err().map(|e| e.to_string());
let _ = tx.send(Action::BulkItemDone { index, error }).await;
});
}
}
}
// ── Key → Action ──────────────────────────────────────────────────────────────
fn tab_shortcut(code: KeyCode) -> Option<Action> {
match code {
KeyCode::Char('1') => Some(Action::TabSelect(Tab::Diary)),
KeyCode::Char('2') => Some(Action::TabSelect(Tab::AddReview)),
KeyCode::Char('3') => Some(Action::TabSelect(Tab::BulkImport)),
KeyCode::Char('4') => Some(Action::TabSelect(Tab::Settings)),
_ => None,
}
}
fn key_to_action(app: &App, key: ratatui::crossterm::event::KeyEvent) -> Option<Action> {
// Ctrl+C always quits
if key.modifiers.contains(KeyModifiers::CONTROL) && key.code == KeyCode::Char('c') {
return Some(Action::Quit);
}
match &app.screen {
Screen::Setup(_) => match key.code {
KeyCode::Char(c) => Some(Action::InputChar(c)),
KeyCode::Backspace => Some(Action::Backspace),
KeyCode::Enter => Some(Action::SetupSubmit),
KeyCode::Esc => Some(Action::Escape),
_ => None,
},
Screen::Login(_) => match key.code {
KeyCode::Char(c) => Some(Action::InputChar(c)),
KeyCode::Backspace => Some(Action::Backspace),
KeyCode::Tab => Some(Action::FocusNext),
KeyCode::BackTab => Some(Action::FocusPrev),
KeyCode::Enter => Some(Action::LoginSubmit),
_ => None,
},
Screen::Main(m) => match m.tab {
Tab::Diary => match key.code {
KeyCode::Up | KeyCode::Char('k') => Some(Action::ScrollUp),
KeyCode::Down | KeyCode::Char('j') => Some(Action::ScrollDown),
KeyCode::Enter => Some(Action::OpenHistory),
KeyCode::Char('d') => Some(Action::DeleteInit),
KeyCode::Char('y') if m.diary.delete_pending.is_some() => Some(Action::DeleteConfirm),
KeyCode::Char('n') if m.diary.delete_pending.is_some() => Some(Action::DeleteCancel),
KeyCode::Esc => Some(Action::Escape),
KeyCode::Char('q') => Some(Action::Quit),
KeyCode::Tab => Some(Action::TabNext),
KeyCode::BackTab => Some(Action::TabPrev),
KeyCode::Char('>') | KeyCode::Char('m') => Some(Action::LoadMore),
KeyCode::Char('<') | KeyCode::Char('b') => Some(Action::LoadPrev),
_ => tab_shortcut(key.code),
},
Tab::AddReview => match key.code {
KeyCode::Char(c) => Some(Action::InputChar(c)),
KeyCode::Backspace => Some(Action::Backspace),
KeyCode::Tab => Some(Action::FocusNext),
KeyCode::BackTab => Some(Action::FocusPrev),
KeyCode::Left => Some(Action::RatingDown),
KeyCode::Right => Some(Action::RatingUp),
KeyCode::Enter => Some(Action::ReviewSubmit),
KeyCode::Esc => Some(Action::Escape),
_ => None,
},
Tab::BulkImport => {
let in_path = m.bulk_import.stage == BulkImportStage::EnterPath;
match key.code {
KeyCode::Char(c) if in_path => Some(Action::InputChar(c)),
KeyCode::Backspace if in_path => Some(Action::Backspace),
KeyCode::Enter => match m.bulk_import.stage {
BulkImportStage::EnterPath => Some(Action::BulkParseFile),
BulkImportStage::Preview => Some(Action::BulkImportAll),
_ => None,
},
KeyCode::Esc => Some(Action::BulkCancel),
KeyCode::Tab if !in_path => Some(Action::TabNext),
KeyCode::BackTab if !in_path => Some(Action::TabPrev),
KeyCode::Char('q') if !in_path => Some(Action::Quit),
_ if !in_path => tab_shortcut(key.code),
_ => None,
}
}
Tab::Settings => {
let on_url = m.settings.focused == SettingsField::ApiUrl;
match key.code {
KeyCode::Char(c) if on_url => Some(Action::InputChar(c)),
KeyCode::Backspace if on_url => Some(Action::Backspace),
KeyCode::Tab => Some(Action::FocusNext),
KeyCode::BackTab => Some(Action::FocusPrev),
KeyCode::Enter => match m.settings.focused {
SettingsField::Save | SettingsField::ApiUrl => Some(Action::SettingsSave),
SettingsField::Logout => Some(Action::SettingsLogout),
},
KeyCode::Esc => Some(Action::Escape),
KeyCode::Char('q') => Some(Action::Quit),
_ if !on_url => tab_shortcut(key.code),
_ => None,
}
}
},
}
}