mugit: internal/cli/ssh_shell.go (be1cdfdd76f4248e14a065d5b2c6710a5f79b2fe)

mugit/internal/cli/ssh_shell.go (view raw)

Oleksandr Smirnov ssh: write only ssh logs to file (othewise they are shown to user..., 22 days ago

package cli

import (
	"context"
	"errors"
	"fmt"
	"log/slog"
	"os"

	"github.com/urfave/cli/v3"
)

var errSSHDisabled = errors.New("ssh is disabled")

func (c *Cli) sshShellAction(ctx context.Context, cmd *cli.Command) error {
	if !c.cfg.SSH.Enable {
		return errSSHDisabled
	}
	if err := c.setupLogger(); err != nil {
		return err
	}

	sshCommand := os.Getenv("SSH_ORIGINAL_COMMAND")
	if err := c.ssh.HandleCommand(ctx, sshCommand, os.Stdin, os.Stdout, os.Stderr); err != nil {
		slog.Error("ssh command failed", "command", sshCommand, "err", err)
		os.Exit(1)
		return nil
	}
	return nil
}

func (c *Cli) sshAuthorizedKeysAction(ctx context.Context, cmd *cli.Command) error {
	if !c.cfg.SSH.Enable {
		return errSSHDisabled
	}
	if err := c.setupLogger(); err != nil {
		return err
	}

	fingerprint := cmd.Args().First()
	if fingerprint == "" {
		return fmt.Errorf("fingerprint is required")
	}

	executablePath, err := os.Executable()
	if err != nil {
		return err
	}

	out := c.ssh.AuthorizedKeys(executablePath)
	_, _ = fmt.Fprint(os.Stdout, out)

	return nil
}

1	package cli
2
3	import (
4	"context"
5	"errors"
6	"fmt"
7	"log/slog"
8	"os"
9
10	"github.com/urfave/cli/v3"
11	)
12
13	var errSSHDisabled = errors.New("ssh is disabled")
14
15	func (c Cli) sshShellAction(ctx context.Context, cmd cli.Command) error {
16	if !c.cfg.SSH.Enable {
17	return errSSHDisabled
18	}
19	if err := c.setupLogger(); err != nil {
20	return err
21	}
22
23	sshCommand := os.Getenv("SSH_ORIGINAL_COMMAND")
24	if err := c.ssh.HandleCommand(ctx, sshCommand, os.Stdin, os.Stdout, os.Stderr); err != nil {
25	slog.Error("ssh command failed", "command", sshCommand, "err", err)
26	os.Exit(1)
27	return nil
28	}
29	return nil
30	}
31
32	func (c Cli) sshAuthorizedKeysAction(ctx context.Context, cmd cli.Command) error {
33	if !c.cfg.SSH.Enable {
34	return errSSHDisabled
35	}
36	if err := c.setupLogger(); err != nil {
37	return err
38	}
39
40	fingerprint := cmd.Args().First()
41	if fingerprint == "" {
42	return fmt.Errorf("fingerprint is required")
43	}
44
45	executablePath, err := os.Executable()
46	if err != nil {
47	return err
48	}
49
50	out := c.ssh.AuthorizedKeys(executablePath)
51	_, _ = fmt.Fprint(os.Stdout, out)
52
53	return nil
54	}