use crate::{
config,
model::{Project, Task, TaskStatus},
services,
theme::Theme,
ui,
};
use crossterm::event::{self, Event, KeyCode, KeyEventKind};
use std::io;
use tui_input::backend::crossterm::EventHandler;
use tui_input::Input;
#[derive(Debug)]
pub enum CurrentScreen {
Project,
Task,
}
#[derive(Debug)]
pub struct App {
pub current_screen: CurrentScreen,
pub projects: Vec<Project>,
pub tasks: Vec<Task>,
pub selected_project: Option<usize>, pub selected_task: Option<usize>, pub selected_task_status: Option<TaskStatus>,
pub edit_project: bool,
pub add_project: bool,
pub edit_task: bool,
pub add_task: bool,
pub input: Input,
pub tip_message: Option<String>,
pub confirm_message: Option<String>,
pub theme: Theme,
exit: bool,
}
impl App {
pub fn new() -> Self {
Self {
current_screen: CurrentScreen::Project,
projects: vec![],
tasks: vec![],
selected_project: None,
selected_task: None,
selected_task_status: Some(TaskStatus::Todo),
edit_project: false,
add_project: false,
edit_task: false,
add_task: false,
input: Input::default(),
tip_message: None,
confirm_message: None,
theme: Theme::from(config::get_config().theme.as_str()),
exit: false,
}
}
pub fn run(&mut self) -> io::Result<()> {
let mut terminal = ratatui::init();
loop {
terminal.draw(|f| ui::draw(f, self))?;
if self.exit {
break;
}
self.handle_event()?;
}
ratatui::restore();
Ok(())
}
fn handle_event(&mut self) -> io::Result<()> {
let event = event::read()?;
if let Event::Key(key) = event {
if key.kind != KeyEventKind::Press {
return Ok(());
}
match self.current_screen {
CurrentScreen::Project => {
if self.tip_message.is_some() {
match key.code {
KeyCode::Enter => self.tip_message = None,
_ => {}
}
return Ok(());
}
if self.confirm_message.is_some() {
match key.code {
KeyCode::Enter => {
self.confirm_message = None;
self.delete_project();
}
KeyCode::Esc => self.confirm_message = None,
_ => {}
}
return Ok(());
}
if self.add_project || self.edit_project {
match key.code {
KeyCode::Enter => self.submit_project(),
KeyCode::Esc => self.close_project_popup(),
_ => {
self.input.handle_event(&event);
}
}
return Ok(());
}
match key.code {
KeyCode::Char('q') => {
self.exit = true;
}
KeyCode::Enter => {
if self.selected_project.is_some() {
self.selected_task_status = Some(TaskStatus::Todo);
self.current_screen = CurrentScreen::Task;
}
}
KeyCode::Down => self.next_project(),
KeyCode::Up => self.prev_project(),
KeyCode::Char('a') => self.add_project = true,
KeyCode::Char('e') => {
if let Some(project) = self.get_project() {
self.edit_project = true;
self.input = Input::default().with_value(project.name);
} else {
self.tip_message = Some(String::from("请先选择项目"));
}
}
KeyCode::Char('d') => {
if let Some(project) = self.get_project() {
self.confirm_message =
Some(format!("确认删除项目 {} 吗？", project.name));
} else {
self.tip_message = Some(String::from("请先选择项目"));
}
}
KeyCode::Char('t') => self.change_theme(),
_ => {}
}
}
CurrentScreen::Task => {
if self.tip_message.is_some() {
match key.code {
KeyCode::Enter => self.tip_message = None,
_ => {}
}
return Ok(());
}
if self.confirm_message.is_some() {
match key.code {
KeyCode::Enter => {
self.confirm_message = None;
self.delete_task();
}
KeyCode::Esc => self.confirm_message = None,
_ => {}
}
return Ok(());
}
if self.add_task || self.edit_task {
match key.code {
KeyCode::Enter => self.submit_task(),
KeyCode::Esc => self.close_task_popup(),
_ => {
self.input.handle_event(&event);
}
}
return Ok(());
}
match key.code {
KeyCode::Char('q') => {
self.exit = true;
}
KeyCode::Down => self.next_task(),
KeyCode::Up => self.prev_task(),
KeyCode::Left => self.prev_task_status(),
KeyCode::Right => self.next_task_status(),
KeyCode::Char('a') => self.add_task = true,
KeyCode::Char('e') => {
if let Some(task) = self.get_task() {
self.edit_task = true;
self.input = Input::default().with_value(task.title);
} else {
self.tip_message = Some(String::from("请先选择任务"));
}
}
KeyCode::Char('s') => self.start_task(),
KeyCode::Char('c') => self.complete_task(),
KeyCode::Char('d') => {
if let Some(task) = self.get_task() {
self.confirm_message =
Some(format!("确认删除任务 {} 吗？", task.title));
} else {
self.tip_message = Some(String::from("请先选择任务"));
}
}
KeyCode::Esc => {
self.current_screen = CurrentScreen::Project;
self.selected_task = None
}
KeyCode::Char('t') => self.change_theme(),
_ => {}
}
}
}
}
Ok(())
}
fn close_project_popup(&mut self) {
self.add_project = false;
self.edit_project = false;
self.input.reset();
}
fn close_task_popup(&mut self) {
self.add_task = false;
self.edit_task = false;
self.input.reset();
}
fn next_project(&mut self) {
let len = self.projects.len();
if len == 0 {
return;
}
if let Some(selected) = self.selected_project {
if selected < len - 1 {
self.selected_project = Some(selected + 1);
} else {
self.selected_project = Some(0);
}
} else {
self.selected_project = Some(0);
}
}
fn prev_project(&mut self) {
let len = self.projects.len();
if len == 0 {
return;
}
if let Some(selected) = self.selected_project {
if selected > 0 {
self.selected_project = Some(selected - 1);
} else {
self.selected_project = Some(len - 1);
}
} else {
self.selected_project = Some(len - 1);
}
}
fn next_task(&mut self) {
let status = self.selected_task_status.unwrap_or(TaskStatus::Todo);
let tasks = self.get_tasks_with_status(status);
let len = tasks.len();
if len == 0 {
return;
}
if let Some(selected) = self.selected_task {
if selected < len - 1 {
self.selected_task = Some(selected + 1);
} else {
self.selected_task = Some(0);
}
} else {
self.selected_task = Some(0);
}
}
fn prev_task(&mut self) {
let status = self.selected_task_status.unwrap_or(TaskStatus::Todo);
let tasks = self.get_tasks_with_status(status);
let len = tasks.len();
if len == 0 {
return;
}
if let Some(selected) = self.selected_task {
if selected > 0 {
self.selected_task = Some(selected - 1);
} else {
self.selected_task = Some(len - 1);
}
} else {
self.selected_task = Some(len - 1);
}
}
fn next_task_status(&mut self) {
let next = self.selected_task_status.unwrap_or(TaskStatus::Todo).next();
self.selected_task_status = Some(next);
let tasks = self.get_tasks_with_status(next);
self.selected_task = if tasks.is_empty() { None } else { Some(0) };
}
fn prev_task_status(&mut self) {
let prev = self.selected_task_status.unwrap_or(TaskStatus::Todo).prev();
self.selected_task_status = Some(prev);
let tasks = self.get_tasks_with_status(prev);
self.selected_task = if tasks.is_empty() { None } else { Some(0) };
}
fn submit_project(&mut self) {
if self.add_project {
let name = self.input.value_and_reset();
if let Err(err) = services::project::add(String::from(name)) {
self.tip_message = Some(format!("新增项目失败：{}", err));
}
self.add_project = false;
}
if self.edit_project {
let new_name = self.input.value_and_reset().trim().to_string();
if let Some(project) = self.get_project() {
let old_name = project.name;
match services::project::edit(old_name.clone(), new_name.clone()) {
Ok(_) => {
let mut config = config::get_config();
if config.current_project.as_deref() == Some(old_name.as_str()) {
config.current_project = Some(new_name.clone());
if let Err(err) = config.save() {
self.tip_message = Some(format!(
"项目已改名，但保存当前项目失败：{}；请运行 `tam project use {}`",
err, new_name
));
}
}
}
Err(err) => self.tip_message = Some(format!("重命名项目失败：{}", err)),
}
}
self.edit_project = false;
}
}
fn submit_task(&mut self) {
if self.add_task {
let title = String::from(self.input.value_and_reset());
if let Some(project) = self.get_project() {
log::info!("current project in submit_task: {:?}", project);
let project_id = project.id;
if let Err(err) = services::task::add(title, project_id) {
self.tip_message = Some(format!("新增任务失败：{}", err));
}
} else {
self.tip_message = Some(String::from("请先选择项目"));
}
self.add_task = false;
}
if self.edit_task {
let title = String::from(self.input.value_and_reset());
if let Some(task) = self.get_task() {
if let Some(id) = task.id {
if let Err(err) = services::task::edit(id, title) {
self.tip_message = Some(format!("修改任务失败：{}", err));
}
}
}
self.edit_task = false;
}
}
fn delete_project(&mut self) {
let Some(project) = self.get_project() else {
return;
};
match services::project::delete(project.name.clone()) {
Ok(_) => {
let mut config = config::get_config();
if config.current_project.as_deref() == Some(project.name.as_str()) {
config.current_project = None;
if let Err(err) = config.save() {
self.tip_message = Some(format!(
"项目已删除，但清除当前项目失败：{}；请运行 `tam project use <existing project>`",
err
));
}
}
self.selected_project = None;
self.selected_task = None;
self.tasks.clear();
}
Err(err) => self.tip_message = Some(format!("删除项目失败：{}", err)),
}
}
fn start_task(&mut self) {
if let Some(task) = self.get_task() {
if let Some(id) = task.id {
if let Err(err) = services::task::start(id) {
self.tip_message = Some(format!("任务开始失败：{}", err));
}
}
} else {
self.tip_message = Some(String::from("请先选择任务"));
}
}
fn complete_task(&mut self) {
if let Some(task) = self.get_task() {
if let Some(id) = task.id {
if let Err(err) = services::task::done(id) {
self.tip_message = Some(format!("任务完成失败：{}", err));
}
}
} else {
self.tip_message = Some(String::from("请先选择任务"));
}
}
fn delete_task(&mut self) {
if let Some(task) = self.get_task() {
if let Some(id) = task.id {
match services::task::delete(id) {
Ok(_) => {
self.selected_task = None;
self.query_tasks();
}
Err(err) => self.tip_message = Some(format!("删除任务失败：{}", err)),
}
}
} else {
self.tip_message = Some(String::from("请先选择任务"));
}
}
fn change_theme(&mut self) {
match self.theme.name.as_str() {
"dark" => {
self.theme = Theme::light();
}
"light" => {
self.theme = Theme::dark();
}
_ => {
self.theme = Theme::dark();
}
}
let mut c = config::get_config();
c.set_theme(self.theme.name.clone());
}
pub fn query_projects(&mut self) {
match services::project::list() {
Ok(projects) => self.projects = projects,
Err(err) => self.tip_message = Some(format!("加载项目失败：{}", err)),
}
}
pub fn query_tasks(&mut self) {
if let Some(project) = self.get_project() {
if let Some(id) = project.id {
match services::task::list(id, None) {
Ok(tasks) => self.tasks = tasks,
Err(err) => self.tip_message = Some(format!("加载任务失败：{}", err)),
}
}
}
}
pub fn get_project(&mut self) -> Option<Project> {
if let Some(selected) = self.selected_project {
if let Some(project) = self.projects.get(selected).cloned() {
return Some(project);
}
self.selected_project = None;
}
self.tip_message = Some(String::from("请先选择项目"));
None
}
pub fn get_task(&self) -> Option<Task> {
if let Some(status) = self.selected_task_status {
if let Some(selected) = self.selected_task {
let tasks = self.get_tasks_with_status(status);
return tasks.get(selected).cloned();
}
}
None
}
pub fn get_tasks_with_status(&self, status: TaskStatus) -> Vec<Task> {
self.tasks
.iter()
.filter(|t| t.status == status)
.cloned()
.collect()
}
}