mirror of
https://github.com/dnote/dnote
synced 2026-03-15 15:05:51 +01:00
146 lines
3.4 KiB
Go
146 lines
3.4 KiB
Go
/* Copyright (C) 2019 Monomax Software Pty Ltd
|
|
*
|
|
* This file is part of Dnote CLI.
|
|
*
|
|
* Dnote CLI is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* Dnote CLI 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 General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with Dnote CLI. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
// Package ui provides the user interface for the program
|
|
package ui
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"os/exec"
|
|
"strings"
|
|
|
|
"github.com/dnote/dnote/pkg/cli/consts"
|
|
"github.com/dnote/dnote/pkg/cli/context"
|
|
"github.com/dnote/dnote/pkg/cli/infra"
|
|
"github.com/dnote/dnote/pkg/cli/utils"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// GetTmpContentPath returns the path to the temporary file containing
|
|
// content being added or edited
|
|
func GetTmpContentPath(ctx context.DnoteCtx) string {
|
|
return fmt.Sprintf("%s/%s", ctx.DnoteDir, consts.TmpContentFilename)
|
|
}
|
|
|
|
// getEditorCommand returns the system's editor command with appropriate flags,
|
|
// if necessary, to make the command wait until editor is close to exit.
|
|
func getEditorCommand() string {
|
|
editor := os.Getenv("EDITOR")
|
|
|
|
var ret string
|
|
|
|
switch editor {
|
|
case "atom":
|
|
ret = "atom -w"
|
|
case "subl":
|
|
ret = "subl -n -w"
|
|
case "mate":
|
|
ret = "mate -w"
|
|
case "vim":
|
|
ret = "vim"
|
|
case "nano":
|
|
ret = "nano"
|
|
case "emacs":
|
|
ret = "emacs"
|
|
case "nvim":
|
|
ret = "nvim"
|
|
default:
|
|
ret = "vi"
|
|
}
|
|
|
|
return ret
|
|
}
|
|
|
|
// SanitizeContent sanitizes note content
|
|
func SanitizeContent(s string) string {
|
|
var ret string
|
|
|
|
ret = strings.Trim(s, " ")
|
|
|
|
// Remove newline at the end of the file because POSIX defines a line as
|
|
// characters followed by a newline
|
|
ret = strings.TrimSuffix(ret, "\n")
|
|
ret = strings.TrimSuffix(ret, "\r\n")
|
|
|
|
return ret
|
|
}
|
|
|
|
func newEditorCmd(ctx context.DnoteCtx, fpath string) (*exec.Cmd, error) {
|
|
config, err := infra.ReadConfig(ctx)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "reading config")
|
|
}
|
|
|
|
args := strings.Fields(config.Editor)
|
|
args = append(args, fpath)
|
|
|
|
return exec.Command(args[0], args[1:]...), nil
|
|
}
|
|
|
|
// GetEditorInput gets the user input by launching a text editor and waiting for
|
|
// it to exit
|
|
func GetEditorInput(ctx context.DnoteCtx, fpath string, content *string) error {
|
|
if !utils.FileExists(fpath) {
|
|
f, err := os.Create(fpath)
|
|
if err != nil {
|
|
return errors.Wrap(err, "creating a temporary content file")
|
|
}
|
|
err = f.Close()
|
|
if err != nil {
|
|
return errors.Wrap(err, "closing the temporary content file")
|
|
}
|
|
}
|
|
|
|
cmd, err := newEditorCmd(ctx, fpath)
|
|
if err != nil {
|
|
return errors.Wrap(err, "creating an editor command")
|
|
}
|
|
|
|
cmd.Stdin = os.Stdin
|
|
cmd.Stdout = os.Stdout
|
|
cmd.Stderr = os.Stderr
|
|
|
|
err = cmd.Start()
|
|
if err != nil {
|
|
return errors.Wrapf(err, "launching an editor")
|
|
}
|
|
|
|
err = cmd.Wait()
|
|
if err != nil {
|
|
return errors.Wrap(err, "waiting for the editor")
|
|
}
|
|
|
|
b, err := ioutil.ReadFile(fpath)
|
|
if err != nil {
|
|
return errors.Wrap(err, "reading the temporary content file")
|
|
}
|
|
|
|
err = os.Remove(fpath)
|
|
if err != nil {
|
|
return errors.Wrap(err, "removing the temporary content file")
|
|
}
|
|
|
|
raw := string(b)
|
|
c := SanitizeContent(raw)
|
|
|
|
*content = c
|
|
|
|
return nil
|
|
}
|