forked from kageru/tracc
131 lines
3.4 KiB
Rust
131 lines
3.4 KiB
Rust
use super::Mode;
|
|
use serde::{Deserialize, Serialize};
|
|
use serde_json::from_reader;
|
|
use std::fs::File;
|
|
use std::io::{self, BufReader, Write};
|
|
use tui::backend::Backend;
|
|
use tui::Terminal;
|
|
|
|
pub struct Tracc {
|
|
// We use owned strings here because they’re easier to manipulate when editing.
|
|
pub todos: Vec<Todo>,
|
|
pub selected: usize,
|
|
pub mode: Mode,
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize, Default, Clone)]
|
|
pub struct Todo {
|
|
text: String,
|
|
done: bool,
|
|
}
|
|
|
|
impl Todo {
|
|
pub fn new(text: &str) -> Self {
|
|
Todo {
|
|
text: text.to_owned(),
|
|
done: false,
|
|
}
|
|
}
|
|
}
|
|
|
|
const JSON_PATH: &str = "tracc.json";
|
|
|
|
fn read_todos() -> Option<Vec<Todo>> {
|
|
File::open(JSON_PATH)
|
|
.ok()
|
|
.map(|f| BufReader::new(f))
|
|
.and_then(|r| from_reader(r).ok())
|
|
}
|
|
|
|
impl Tracc {
|
|
pub fn open_or_create() -> Self {
|
|
Self {
|
|
todos: read_todos().unwrap_or(vec![Todo::new("This is a list entry")]),
|
|
selected: 0,
|
|
mode: Mode::Normal,
|
|
}
|
|
}
|
|
|
|
pub fn printable_todos(&self) -> Vec<String> {
|
|
self.todos
|
|
.iter()
|
|
.map(|todo| format!("[{}] {}", if todo.done { 'x' } else { ' ' }, todo.text))
|
|
.collect()
|
|
}
|
|
|
|
pub fn selection_down(&mut self) {
|
|
self.selected = (self.selected + 1).min(self.todos.len().saturating_sub(1));
|
|
}
|
|
|
|
pub fn selection_up(&mut self) {
|
|
self.selected = self.selected.saturating_sub(1);
|
|
}
|
|
|
|
pub fn insert(&mut self, todo: Todo) {
|
|
if self.selected == self.todos.len().saturating_sub(1) {
|
|
self.todos.push(todo);
|
|
self.selected = self.todos.len() - 1;
|
|
} else {
|
|
self.todos.insert(self.selected + 1, todo);
|
|
self.selected += 1;
|
|
}
|
|
self.mode = Mode::Normal;
|
|
}
|
|
|
|
pub fn remove_current(&mut self) -> Option<Todo> {
|
|
if self.todos.is_empty() {
|
|
return None;
|
|
}
|
|
let index = self.selected;
|
|
self.selected = index.min(self.todos.len() - 1);
|
|
return Some(self.todos.remove(index));
|
|
}
|
|
|
|
pub fn toggle_current(&mut self) {
|
|
self.todos[self.selected].done = !self.todos[self.selected].done;
|
|
}
|
|
|
|
fn current(&self) -> &Todo {
|
|
&self.todos[self.selected]
|
|
}
|
|
|
|
pub fn set_mode(
|
|
&mut self,
|
|
mode: Mode,
|
|
term: &mut Terminal<impl Backend>,
|
|
) -> Result<(), io::Error> {
|
|
match mode {
|
|
Mode::Insert => term.show_cursor()?,
|
|
Mode::Normal => {
|
|
if self.current().text.is_empty() {
|
|
self.remove_current();
|
|
self.selected = self.selected.saturating_sub(1);
|
|
}
|
|
term.hide_cursor()?
|
|
}
|
|
};
|
|
self.mode = mode;
|
|
Ok(())
|
|
}
|
|
|
|
pub fn append_to_current(&mut self, chr: char) {
|
|
self.todos[self.selected].text.push(chr);
|
|
}
|
|
|
|
pub fn current_pop(&mut self) {
|
|
self.todos[self.selected].text.pop();
|
|
}
|
|
|
|
pub fn persist(self) {
|
|
let string = serde_json::to_string(&self.todos).unwrap();
|
|
std::fs::OpenOptions::new()
|
|
.create(true)
|
|
.write(true)
|
|
.truncate(true)
|
|
.open(JSON_PATH)
|
|
.ok()
|
|
.or_else(|| panic!("Can’t save todos to JSON. Dumping raw data:\n{}", string))
|
|
.map(|mut f| f.write(string.as_bytes()));
|
|
}
|
|
}
|