initial release
This commit is contained in:
commit
8d86bdd33d
45 changed files with 2558 additions and 0 deletions
cmd/ssh
94
cmd/ssh/main.go
Normal file
94
cmd/ssh/main.go
Normal file
|
@ -0,0 +1,94 @@
|
|||
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)
|
||||
}
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue