1package server
 2
 3import (
 4	"context"
 5	"fmt"
 6	"net"
 7	"path/filepath"
 8	"strings"
 9	"testing"
10
11	"github.com/charmbracelet/keygen"
12	"github.com/charmbracelet/soft-serve/server/config"
13	"github.com/charmbracelet/ssh"
14	"github.com/matryer/is"
15	gossh "golang.org/x/crypto/ssh"
16)
17
18func randomPort() int {
19	addr, _ := net.Listen("tcp", ":0") //nolint:gosec
20	_ = addr.Close()
21	return addr.Addr().(*net.TCPAddr).Port
22}
23
24func setupServer(tb testing.TB) (*Server, *config.Config, string) {
25	tb.Helper()
26	tb.Log("creating keypair")
27	pub, pkPath := createKeyPair(tb)
28	dp := tb.TempDir()
29	sshPort := fmt.Sprintf(":%d", randomPort())
30	tb.Setenv("SOFT_SERVE_DATA_PATH", dp)
31	tb.Setenv("SOFT_SERVE_INITIAL_ADMIN_KEY", authorizedKey(pub))
32	tb.Setenv("SOFT_SERVE_SSH_LISTEN_ADDR", sshPort)
33	tb.Setenv("SOFT_SERVE_GIT_LISTEN_ADDR", fmt.Sprintf(":%d", randomPort()))
34	cfg := config.DefaultConfig()
35	tb.Log("configuring server")
36	s, err := NewServer(cfg)
37	if err != nil {
38		tb.Fatal(err)
39	}
40	go func() {
41		tb.Log("starting server")
42		s.Start(context.TODO())
43	}()
44	tb.Cleanup(func() {
45		s.Close()
46	})
47	return s, cfg, pkPath
48}
49
50func createKeyPair(tb testing.TB) (ssh.PublicKey, string) {
51	tb.Helper()
52	is := is.New(tb)
53	keyDir := tb.TempDir()
54	kp, err := keygen.NewWithWrite(filepath.Join(keyDir, "id"), nil, keygen.Ed25519)
55	is.NoErr(err)
56	pubkey, _, _, _, err := ssh.ParseAuthorizedKey(kp.PublicKey())
57	is.NoErr(err)
58	return pubkey, filepath.Join(keyDir, "id_ed25519")
59}
60
61func authorizedKey(pk ssh.PublicKey) string {
62	return strings.TrimSpace(string(gossh.MarshalAuthorizedKey(pk)))
63}