feat: auth configuration

still pending using certificate for SSH
shoriwe · Jun 2, 2023 · 44da474 · 44da474
1 parent 84ebbcf
commit 44da474
Show file tree

Hide file tree

Showing 2 changed files with 102 additions and 0 deletions.
diff --git a/compose/auth.go b/compose/auth.go
@@ -0,0 +1,48 @@
+package compose
+
+import (
+	"fmt"
+
+	"golang.org/x/crypto/ssh"
+	"golang.org/x/net/proxy"
+)
+
+type Auth struct {
+	Username   *string `yaml:"username,omitempty" json:"username,omitempty"`
+	Password   *string `yaml:"password,omitempty" json:"password,omitempty"`
+	PrivateKey *string `yaml:"privateKey,omitempty" json:"privateKey,omitempty"`
+}
+
+func (a *Auth) SSHClientConfig() (config *ssh.ClientConfig, err error) {
+	if a.Username == nil {
+		err = fmt.Errorf("no username provided")
+	}
+	if err == nil {
+		config = new(ssh.ClientConfig)
+		config.User = *a.Username
+		if a.Password != nil {
+			config.Auth = append(config.Auth, ssh.Password(*a.Password))
+		}
+		if a.PrivateKey != nil {
+			// TODO: FIXME: This code doesn't work
+			config.Auth = append(config.Auth, ssh.PublicKeys())
+		}
+	}
+	return config, err
+}
+
+func (a *Auth) Socks5() (auth *proxy.Auth, err error) {
+	if a.Username == nil {
+		err = fmt.Errorf("no username provided")
+	}
+	if a.Password == nil {
+		err = fmt.Errorf("no password provided")
+	}
+	if err == nil {
+		auth = &proxy.Auth{
+			User:     *a.Username,
+			Password: *a.Password,
+		}
+	}
+	return auth, err
+}
diff --git a/compose/auth_test.go b/compose/auth_test.go
@@ -0,0 +1,54 @@
+package compose
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestAuth_SSHClientConfig(t *testing.T) {
+	t.Run("Valid", func(tt *testing.T) {
+		auth := Auth{
+			Username: new(string),
+			Password: new(string),
+		}
+		*auth.Username = "sulcud"
+		*auth.Password = "password"
+		_, err := auth.SSHClientConfig()
+		assert.Nil(tt, err)
+	})
+	t.Run("Private Key", func(tt *testing.T) {
+		// TODO: Implement me!
+	})
+	t.Run("No Username", func(tt *testing.T) {
+		auth := Auth{}
+		_, err := auth.SSHClientConfig()
+		assert.NotNil(tt, err)
+	})
+}
+
+func TestAuth_Socks5(t *testing.T) {
+	t.Run("Valid", func(tt *testing.T) {
+		auth := Auth{
+			Username: new(string),
+			Password: new(string),
+		}
+		*auth.Username = "sulcud"
+		*auth.Password = "password"
+		_, err := auth.Socks5()
+		assert.Nil(tt, err)
+	})
+	t.Run("No Username", func(tt *testing.T) {
+		auth := Auth{}
+		_, err := auth.Socks5()
+		assert.NotNil(tt, err)
+	})
+	t.Run("No Password", func(tt *testing.T) {
+		auth := Auth{
+			Username: new(string),
+		}
+		*auth.Username = "sulcud"
+		_, err := auth.Socks5()
+		assert.NotNil(tt, err)
+	})
+}