95 lines
2.2 KiB
Go
95 lines
2.2 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
"os/signal"
|
||
|
|
"syscall"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"git.sr.ht/~erock/pastes.sh/internal"
|
||
|
|
"git.sr.ht/~erock/wish/cms"
|
||
|
|
"git.sr.ht/~erock/wish/cms/db/postgres"
|
||
|
|
"git.sr.ht/~erock/wish/proxy"
|
||
|
|
"git.sr.ht/~erock/wish/send/scp"
|
||
|
|
"git.sr.ht/~erock/wish/send/sftp"
|
||
|
|
"github.com/charmbracelet/wish"
|
||
|
|
bm "github.com/charmbracelet/wish/bubbletea"
|
||
|
|
lm "github.com/charmbracelet/wish/logging"
|
||
|
|
"github.com/gliderlabs/ssh"
|
||
|
|
)
|
||
|
|
|
||
|
|
type SSHServer struct{}
|
||
|
|
|
||
|
|
func (me *SSHServer) authHandler(ctx ssh.Context, key ssh.PublicKey) bool {
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
|
||
|
|
func createRouter(handler *internal.DbHandler) proxy.Router {
|
||
|
|
return func(sh ssh.Handler, s ssh.Session) []wish.Middleware {
|
||
|
|
cmd := s.Command()
|
||
|
|
mdw := []wish.Middleware{}
|
||
|
|
|
||
|
|
if len(cmd) == 0 {
|
||
|
|
mdw = append(mdw,
|
||
|
|
bm.Middleware(cms.Middleware(&handler.Cfg.ConfigCms, handler.Cfg)),
|
||
|
|
lm.Middleware(),
|
||
|
|
)
|
||
|
|
} else if cmd[0] == "scp" {
|
||
|
|
mdw = append(mdw, scp.Middleware(handler))
|
||
|
|
}
|
||
|
|
|
||
|
|
return mdw
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func withProxy(handler *internal.DbHandler) ssh.Option {
|
||
|
|
return func(server *ssh.Server) error {
|
||
|
|
err := sftp.SSHOption(handler)(server)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
return proxy.WithProxy(createRouter(handler))(server)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
host := internal.GetEnv("PASTES_HOST", "0.0.0.0")
|
||
|
|
port := internal.GetEnv("PASTES_SSH_PORT", "2222")
|
||
|
|
cfg := internal.NewConfigSite()
|
||
|
|
logger := cfg.Logger
|
||
|
|
dbh := postgres.NewDB(&cfg.ConfigCms)
|
||
|
|
defer dbh.Close()
|
||
|
|
handler := internal.NewDbHandler(dbh, cfg)
|
||
|
|
|
||
|
|
sshServer := &SSHServer{}
|
||
|
|
s, err := wish.NewServer(
|
||
|
|
wish.WithAddress(fmt.Sprintf("%s:%s", host, port)),
|
||
|
|
wish.WithHostKeyPath("ssh_data/term_info_ed25519"),
|
||
|
|
wish.WithPublicKeyAuth(sshServer.authHandler),
|
||
|
|
withProxy(handler),
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
logger.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
done := make(chan os.Signal, 1)
|
||
|
|
signal.Notify(done, os.Interrupt, syscall.SIGINT, syscall.SIGTERM)
|
||
|
|
logger.Infof("Starting SSH server on %s:%s", host, port)
|
||
|
|
go func() {
|
||
|
|
if err = s.ListenAndServe(); err != nil {
|
||
|
|
logger.Fatal(err)
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
|
||
|
|
<-done
|
||
|
|
logger.Info("Stopping SSH server")
|
||
|
|
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
||
|
|
defer func() { cancel() }()
|
||
|
|
if err := s.Shutdown(ctx); err != nil {
|
||
|
|
logger.Fatal(err)
|
||
|
|
}
|
||
|
|
}
|