116 lines
2.0 KiB
Go
116 lines
2.0 KiB
Go
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 rename(tmppath, filepath)
|
|
}
|
|
|
|
func rename(oldpath, newpath string) error {
|
|
b, err := os.ReadFile(oldpath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := os.WriteFile(b, newpath, os.ModePerm); err != nil {
|
|
return err
|
|
}
|
|
|
|
return os.Remove(oldpath)
|
|
}
|
|
|
|
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
|
|
}
|