97 lines
2.6 KiB
Rust
97 lines
2.6 KiB
Rust
/*
|
|
* Copyright (C) 2022 Ortega Froysa, Nicolás <nicolas@ortegas.org>
|
|
* Author: Ortega Froysa, Nicolás <nicolas@ortegas.org>
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as
|
|
* published by the Free Software Foundation, either version 3 of the
|
|
* License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Affero General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
use std::path::PathBuf;
|
|
use std::fs::{File, OpenOptions};
|
|
use std::io::{BufReader, BufRead};
|
|
|
|
use crate::habit::Habit;
|
|
|
|
pub struct HabitMgr
|
|
{
|
|
habits:Vec<Habit>,
|
|
//data_dir:PathBuf,
|
|
habits_path:PathBuf,
|
|
}
|
|
|
|
impl HabitMgr
|
|
{
|
|
pub fn new(data_dir:&PathBuf) -> Self
|
|
{
|
|
let habits_path:PathBuf = data_dir.clone().join("active_habits.json");
|
|
if !habits_path.is_file()
|
|
{
|
|
File::create(&habits_path)
|
|
.unwrap_or_else(|e| {
|
|
panic!("Error creating file {}:\n{}",
|
|
habits_path.display(), e);
|
|
});
|
|
}
|
|
Self
|
|
{
|
|
habits: Vec::new(),
|
|
//data_dir: data_dir.clone(),
|
|
habits_path,
|
|
}
|
|
}
|
|
|
|
fn import_habits(&mut self)
|
|
{
|
|
let habits_file = File::open(&self.habits_path)
|
|
.unwrap_or_else(|e| {
|
|
panic!("Error opening file {}:\n{}", self.habits_path.display(), e);
|
|
});
|
|
let reader = BufReader::new(&habits_file);
|
|
// TODO: figure out a way to do this with from_reader() instead to simplify
|
|
//habits = serde_json::from_reader(reader).unwrap();
|
|
|
|
for (_i, line) in reader.lines().enumerate()
|
|
{
|
|
let habit:Habit = serde_json::from_str(line.unwrap().as_str()).unwrap();
|
|
self.habits.push(habit);
|
|
}
|
|
}
|
|
|
|
pub fn add(&mut self, name:String, bad:bool, weight:u8, _days:String)
|
|
{
|
|
let habits_file = OpenOptions::new()
|
|
.write(true)
|
|
.append(true)
|
|
.open(&self.habits_path)
|
|
.unwrap_or_else(|e| {
|
|
panic!("Error opening file {}:\n{}", self.habits_path.display(), e);
|
|
});
|
|
|
|
let habit = Habit::new(name, bad, weight);
|
|
serde_json::to_writer(habits_file, &habit)
|
|
.unwrap_or_else(|e| {
|
|
panic!("Error writing to file {}:\n{}",
|
|
self.habits_path.display(), e);
|
|
})
|
|
}
|
|
|
|
pub fn list(&mut self, all:bool, verbose:bool)
|
|
{
|
|
self.import_habits();
|
|
for (i, habit) in self.habits.iter().enumerate()
|
|
{
|
|
println!("{}\t{}", i, habit.get_name());
|
|
}
|
|
}
|
|
}
|