1package prose
 2
 3import (
 4	"context"
 5	"os"
 6	"os/signal"
 7	"syscall"
 8	"time"
 9
10	"github.com/picosh/pico/pkg/db/postgres"
11	"github.com/picosh/pico/pkg/pssh"
12	"github.com/picosh/pico/pkg/send/auth"
13	"github.com/picosh/pico/pkg/send/list"
14	"github.com/picosh/pico/pkg/send/pipe"
15	"github.com/picosh/pico/pkg/send/protocols/rsync"
16	"github.com/picosh/pico/pkg/send/protocols/scp"
17	"github.com/picosh/pico/pkg/send/protocols/sftp"
18	"github.com/picosh/pico/pkg/shared"
19	"github.com/picosh/pico/pkg/storage"
20)
21
22func StartSshServer() {
23	appName := "prose-ssh"
24
25	host := shared.GetEnv("PROSE_HOST", "0.0.0.0")
26	port := shared.GetEnv("PROSE_SSH_PORT", "2222")
27	promPort := shared.GetEnv("PROSE_PROM_PORT", "9222")
28	cfg := NewConfig(appName)
29	logger := cfg.Logger
30
31	ctx, cancel := context.WithCancel(context.Background())
32	defer cancel()
33
34	dbh := NewDB(postgres.NewDB(cfg.DbURL, cfg.Logger))
35	defer func() {
36		_ = dbh.Close()
37	}()
38
39	adapter := storage.GetStorageTypeFromEnv()
40	st, err := storage.NewStorage(cfg.Logger, adapter)
41	if err != nil {
42		logger.Error("loading storage", "err", err)
43		return
44	}
45
46	handler := NewFileHandler(cfg, dbh, st)
47
48	sshAuth := shared.NewSshAuthHandler(dbh, logger, "prose")
49
50	// Create a new SSH server
51	server, err := pssh.NewSSHServerWithConfig(
52		ctx,
53		logger,
54		appName,
55		host,
56		port,
57		promPort,
58		"ssh_data/term_info_ed25519",
59		sshAuth.PubkeyAuthHandler,
60		[]pssh.SSHServerMiddleware{
61			pipe.Middleware(handler, ".md"),
62			list.Middleware(handler),
63			scp.Middleware(handler),
64			rsync.Middleware(handler),
65			auth.Middleware(handler),
66			pssh.PtyMdw(pssh.DeprecatedNotice(), 200*time.Millisecond),
67			pssh.LogMiddleware(handler, dbh),
68		},
69		[]pssh.SSHServerMiddleware{
70			sftp.Middleware(handler),
71			pssh.LogMiddleware(handler, dbh),
72		},
73		nil,
74	)
75
76	if err != nil {
77		logger.Error("failed to create ssh server", "err", err.Error())
78		os.Exit(1)
79	}
80
81	done := make(chan os.Signal, 1)
82	signal.Notify(done, os.Interrupt, syscall.SIGINT, syscall.SIGTERM)
83	logger.Info("Starting SSH server", "addr", server.Config.ListenAddr)
84	go func() {
85		if err = server.ListenAndServe(); err != nil {
86			logger.Error("serve", "err", err.Error())
87			os.Exit(1)
88		}
89	}()
90
91	exit := func() {
92		logger.Info("stopping ssh server")
93		cancel()
94	}
95
96	<-done
97	exit()
98}