diary/cmd/edit.go

/*
Copyright © 2021 NAME HERE <EMAIL ADDRESS>

*/
package cmd

import (
	"io/ioutil"
	"log"
	"os"
	"strconv"

	"github.com/spf13/cobra"
)

// editCmd represents the edit command
var editCmd = &cobra.Command{
	Use:   "edit",
	Short: "Edit the title of a diary entry",
	Args:  cobra.ExactArgs(1),
	Long: `
	Opens a file with the title of the entry. Edit this file to edit the title.
	Note that trailing spaces will be trimmed.
	`,
	PreRunE: func(cmd *cobra.Command, args []string) error {
		id, err := strconv.Atoi(args[0])
		if err != nil {
			return err
		}
		_, err = App.GetDiaryEntry(int64(id))
		if err != nil {
			return err
		}
		return nil
	},
	Run: func(cmd *cobra.Command, args []string) {
		id, _ := strconv.Atoi(args[0])
		entry, _ := App.GetDiaryEntry(int64(id))
		temp_file, err := ioutil.TempFile(os.TempDir(), "diary_note_title.*.md")
		if err != nil {
			log.Fatal(err)
		}

		// Preload the data in the file
		_, err = temp_file.Write([]byte(entry.Title))
		if err != nil {
			log.Fatal(err)
		}

		err = OpenEditor(temp_file.Name())
		if err != nil {
			log.Fatal(err)
		}
		new_content, err := ioutil.ReadFile(temp_file.Name())
		if err != nil {
			log.Fatal(err)
		}
		App.EditDiaryEntry(entry.Id, string(new_content))

	},
}

func init() {
	rootCmd.AddCommand(editCmd)

	// Here you will define your flags and configuration settings.

	// Cobra supports Persistent Flags which will work for this command
	// and all subcommands, e.g.:
	// editCmd.PersistentFlags().String("foo", "", "A help for foo")

	// Cobra supports local flags which will only run when this command
	// is called directly, e.g.:
	// editCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
}