main.go

package main

import (
	"fmt"
	"log"
	"os"
	"os/user"
	"strconv"
	"strings"
	"time"

	cmd "main/command"
	db "main/database"
)

func main() {
	user, err := user.Current()
	if err != nil {
		log.Fatal(err)
	}

	configdir := user.HomeDir + "/.RTM"

	if len(os.Args) <= 1 {
		log.Fatal("Wrong input. Run rtm --help")
	}

	local_db := db.Check_database()

	args := os.Args[1:]
	command := args[0]

	switch command {

	//Status
	case "status":
		var temp_task []db.Task

		status := local_db.Find(&temp_task)

		if status.Error != nil {
			log.Fatal(status.Error)
		} else {
			fmt.Println("ID \t Status \t Description")
			for _, item := range temp_task {
				if item.Status == "DONE" {
					fmt.Println(item.ID, "\t\033[32m", item.Status, "\033[0m\t\t", item.Name)
				} else {
					fmt.Println(item.ID, "\t\033[34m", item.Status, "\033[0m\t\t", item.Name)
				}
			}
		}

	//ADD
	case "add":
		task_description := strings.Join(os.Args[2:], " ")
		task := db.Task{Name: task_description, Status: "TODO"}
		result := local_db.Create(&task)
		if result.Error != nil {
			log.Fatal(result.Error)
		} else {
			fmt.Println("Task:", task.Name, "added")
		}

	//Delete
	case "del", "rm":
		task_id := os.Args[2]
		//Delete based on ID
		delresult := local_db.Delete(&db.Task{}, task_id)
		if delresult.Error != nil {
			log.Fatal(delresult.Error)
		} else {
			fmt.Println("Task ID", task_id, "Deleted")
		}

	//Done
	case "done":
		task_id := os.Args[2]

		result := local_db.Model(&db.Task{}).Where("ID = ?", task_id).Update("Status", "DONE")
		if result.Error != nil {
			log.Fatal(result.Error)
		} else {
			fmt.Println("Task", task_id, "is now Done! Good Job.")
		}

	//Export
	case "export":

		var temp_task []db.Task
		data := local_db.Find(&temp_task)
		if data.Error != nil {
			log.Fatal(data.Error)
		}

		now := time.Now()
		timenow := now.Format(time.RFC3339)
		export_file := configdir + "/dump-" + timenow

		//File
		file, err := os.Create(export_file)
		if err != nil {
			log.Fatal(err)
		} else {

			for _, item := range temp_task {
				recordID := strconv.FormatUint(uint64(item.ID), 10)
				record := fmt.Sprintf("%s \t %s \t %s \n", recordID, item.Status, item.Name)
				_, err := file.WriteString(record)
				if err != nil {
					log.Fatal(err)
				}
			}
		}

	//Help
	case "--help", "help", "-h":
		cmd.Help()
	default:
		fmt.Println("Please provide a proper input.")
		fmt.Println("Type rtm --help for help")
	}
}