1package main
2
3import (
4 "fmt"
5 "log"
6 "path/filepath"
7 "strings"
8
9 tea "github.com/charmbracelet/bubbletea"
10 "github.com/charmbracelet/charm/keygen"
11 "github.com/gliderlabs/ssh"
12 gossh "golang.org/x/crypto/ssh"
13)
14
15type Middleware func(ssh.Handler) ssh.Handler
16
17func LoggingMiddleware() Middleware {
18 return func(sh ssh.Handler) ssh.Handler {
19 return func(s ssh.Session) {
20 hpk := s.PublicKey() != nil
21 log.Printf("%s connect %v %v\n", s.RemoteAddr().String(), hpk, s.Command())
22 sh(s)
23 log.Printf("%s disconnect %v %v\n", s.RemoteAddr().String(), hpk, s.Command())
24 }
25 }
26}
27
28func BubbleTeaMiddleware(bth func(ssh.Session) tea.Model) Middleware {
29 return func(sh ssh.Handler) ssh.Handler {
30 return func(s ssh.Session) {
31 m := bth(s)
32 if m != nil {
33 p := tea.NewProgram(m, tea.WithAltScreen(), tea.WithInput(s), tea.WithOutput(s))
34 err := p.Start()
35 if err != nil {
36 log.Printf("%s error %v: %s\n", s.RemoteAddr().String(), s.Command(), err)
37 }
38 }
39 sh(s)
40 }
41 }
42}
43
44type Server struct {
45 server *ssh.Server
46 key gossh.PublicKey
47}
48
49func NewServer(port int, keyPath string, mw ...Middleware) (*Server, error) {
50 s := &Server{server: &ssh.Server{}}
51 s.server.Version = "OpenSSH_7.6p1"
52 s.server.Addr = fmt.Sprintf(":%d", port)
53 s.server.PasswordHandler = s.passHandler
54 s.server.PublicKeyHandler = s.authHandler
55 kps := strings.Split(keyPath, string(filepath.Separator))
56 kp := strings.Join(kps[:len(kps)-1], string(filepath.Separator))
57 n := strings.TrimRight(kps[len(kps)-1], "_ed25519")
58 _, err := keygen.NewSSHKeyPair(kp, n, nil, "ed25519")
59 if err != nil {
60 return nil, err
61 }
62 k := ssh.HostKeyFile(keyPath)
63 err = s.server.SetOption(k)
64 if err != nil {
65 return nil, err
66 }
67 h := func(s ssh.Session) {}
68 for _, m := range mw {
69 h = m(h)
70 }
71 s.server.Handler = h
72 return s, nil
73}
74
75func (srv *Server) sessionHandler(s ssh.Session) {
76}
77
78func (srv *Server) authHandler(ctx ssh.Context, key ssh.PublicKey) bool {
79 return true
80}
81
82func (srv *Server) passHandler(ctx ssh.Context, pass string) bool {
83 return true
84}
85
86func (srv *Server) Start() error {
87 return srv.server.ListenAndServe()
88}