htracker/src/habitmgr.rs

/*
 * 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,
}

impl HabitMgr
{
	pub fn new(data_dir:&PathBuf) -> Self
	{
		Self
		{
			habits: Vec::new(),
			data_dir: data_dir.clone(),
		}
	}

	fn import_habits(&mut self)
	{
		// TODO: move habits_path to new function to avoid redundant code
		let habits_path:PathBuf = self.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)
				});
		}
		let habits_file = File::open(&habits_path)
			.unwrap_or_else(|e| {
				panic!("Error opening file {}:\n{}", 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_path:PathBuf = self.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)
				});
		}
		let habits_file = OpenOptions::new()
			.write(true)
			.append(true)
			.open(&habits_path)
			.unwrap_or_else(|e| {
				panic!("Error opening file {}:\n{}", habits_path.display(), e);
			});

		let habit = Habit::new(name, bad, weight);
	}

	pub fn list(&mut self, all:bool, verbose:bool)
	{
		self.import_habits();
		for i in &self.habits
		{
			println!("{}", i.get_name());
		}
	}
}