package main

import (
	"bytes"
	"io"
	"io/ioutil"
	"os"

	"gitea.inhome.blapointe.com/gogs/pttodo/pttodo"

	"gopkg.in/yaml.v2"
)

const (
	DUMP_ALL       = "all"
	DUMP_TODO      = "todo"
	DUMP_SCHEDULED = "scheduled"
	DUMP_DONE      = "done"
)

func main() {
	if err := _main(); err != nil {
		panic(err)
	}
}

func _main() error {
	config := getConfig()
	if err := add(&config); err != nil {
		return err
	}
	if err := edit(&config); err != nil {
		return err
	}
	return dump(config)
}

func merge(filepath string, mergeTargetFilePath string) error {
	baseReader, err := filePathReader(filepath)
	if err != nil {
		return err
	}
	baseB, err := ioutil.ReadAll(baseReader)
	if err != nil {
		return err
	}

	mergingReader, err := filePathReader(mergeTargetFilePath)
	if err != nil {
		return err
	}
	mergingB, err := ioutil.ReadAll(mergingReader)
	if err != nil {
		return err
	}

	var base, merging pttodo.Root
	if err := yaml.Unmarshal(baseB, &base); err != nil {
		return err
	}
	if err := yaml.Unmarshal(mergingB, &merging); err != nil {
		return err
	}

	base.MergeIn(merging)

	tmppath, err := marshalRootToTempFile(base)
	if err != nil {
		return err
	}
	return os.Rename(tmppath, filepath)
}

func marshalRootToTempFile(root pttodo.Root) (string, error) {
	f, err := ioutil.TempFile(os.TempDir(), "tmp")
	if err != nil {
		return "", err
	}
	f.Close()
	os.Remove(f.Name())
	b, err := yaml.Marshal(root)
	if err != nil {
		return "", err
	}
	filepath := f.Name() + ".yaml"
	err = ioutil.WriteFile(filepath, b, os.ModePerm)
	return filepath, err
}

func filePathReader(path string) (io.Reader, error) {
	if path == "-" {
		return os.Stdin, nil
	}
	b, err := ioutil.ReadFile(path)
	if os.IsNotExist(err) {
		return bytes.NewReader([]byte("{}")), nil
	}
	if err != nil {
		return nil, err
	}
	return bytes.NewReader(b), nil
}