gitea-tea/internal/app/editor.go

// Copyright 2026 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT

package app

import (
	"context"
	"errors"
	"fmt"
	"io"
	"os"
	"strings"

	"gitea.dev/tea/internal/clierr"
	"gitea.dev/tea/internal/terminal"
	"github.com/spf13/cobra"
)

func finishDraft(draft *terminal.Draft, result *error) {
	if draft.Path == "" {
		return
	}
	if *result == nil {
		if err := os.Remove(draft.Path); err != nil {
			*result = errors.New("operation completed, but its draft could not be removed")
		}
	}
	if *result != nil {
		*result = fmt.Errorf("%w; draft saved at %s", *result, draft.Path)
	}
}

func (a *application) editor() (terminal.Editor, string, error) {
	preferences, err := a.preferences()
	if err != nil {
		return nil, "", err
	}
	if preferences.Editor == "" {
		return nil, "", clierr.Wrap(clierr.Usage, errors.New("configure an editor using tea config set editor or EDITOR"))
	}
	return a.deps.Editor, preferences.Editor, nil
}

func (a *application) editDraft(cmd *cobra.Command, body string) (terminal.Draft, error) {
	editor, command, err := a.editor()
	if err != nil {
		return terminal.Draft{}, err
	}
	return editor.Edit(cmd.Context(), command, body)
}

func (a *application) bodyEditor(cmd *cobra.Command) terminal.BodyEditor {
	return func(_ context.Context, initial string) (terminal.Draft, error) { return a.editDraft(cmd, initial) }
}

func (a *application) editPath(cmd *cobra.Command, path string) error {
	editor, command, err := a.editor()
	if err != nil {
		return err
	}
	return editor.EditPath(cmd.Context(), command, path)
}

// readBody checks source conflicts before reading stdin or invoking any service.
func (a *application) readBody(cmd *cobra.Command, body, file string) (string, error) {
	return a.readTextSource(cmd, body, file, "body", "body-file")
}

func (a *application) readTextSource(cmd *cobra.Command, text, file, textFlag, fileFlag string) (string, error) {
	if err := validateTextSources(cmd, textFlag, fileFlag); err != nil {
		return "", err
	}
	if !cmd.Flags().Changed(fileFlag) {
		return text, nil
	}
	data, err := readFileOrStdin(a.deps.Streams.In, file)
	if err != nil {
		return "", fmt.Errorf("read %s: %w", textFlag, err)
	}
	return string(data), nil
}

// readSecret is readTextSource for credentials: one trailing line ending from a file is dropped.
func (a *application) readSecret(cmd *cobra.Command, text, file, textFlag, fileFlag string) (string, error) {
	value, err := a.readTextSource(cmd, text, file, textFlag, fileFlag)
	if cmd.Flags().Changed(fileFlag) {
		value = strings.TrimSuffix(strings.TrimSuffix(value, "\n"), "\r")
	}
	return value, err
}

// readFileOrStdin reads a file, or stdin when path is "-".
func readFileOrStdin(stdin io.Reader, path string) ([]byte, error) {
	if path == "-" {
		return io.ReadAll(stdin)
	}
	return os.ReadFile(path)
}

func validateBodySources(cmd *cobra.Command) error {
	return validateTextSources(cmd, "body", "body-file")
}

func validateTextSources(cmd *cobra.Command, textFlag, fileFlag string) error {
	text, file := cmd.Flags().Changed(textFlag), cmd.Flags().Changed(fileFlag)
	if flag := cmd.Flags().Lookup("editor"); flag != nil {
		editor, err := cmd.Flags().GetBool("editor")
		if err != nil {
			return err
		}
		if editor && (text || file) {
			return clierr.Wrap(clierr.Usage, fmt.Errorf("--editor, --%s, and --%s are mutually exclusive", textFlag, fileFlag))
		}
	}
	if text && file {
		return clierr.Wrap(clierr.Usage, fmt.Errorf("--%s and --%s are mutually exclusive", textFlag, fileFlag))
	}
	if file {
		path, err := cmd.Flags().GetString(fileFlag)
		if err != nil {
			return err
		}
		if path == "" {
			return clierr.Wrap(clierr.Usage, fmt.Errorf("--%s must name a file or -", fileFlag))
		}
	}
	return nil
}