Initial commit

internal/web/ssh.go

@@ -0,0 +1,79 @@
+package web
+
+import (
+	"crypto/sha256"
+	"encoding/base64"
+	"github.com/labstack/echo/v4"
+	"golang.org/x/crypto/ssh"
+	"opengist/internal/models"
+	"strconv"
+)
+
+func sshKeys(ctx echo.Context) error {
+	user := getUserLogged(ctx)
+
+	keys, err := models.GetSSHKeysByUserID(user.ID)
+	if err != nil {
+		return errorRes(500, "Cannot get SSH keys", err)
+	}
+
+	setData(ctx, "sshKeys", keys)
+	setData(ctx, "htmlTitle", "Manage SSH keys")
+	return html(ctx, "ssh_keys.html")
+}
+
+func sshKeysProcess(ctx echo.Context) error {
+	setData(ctx, "htmlTitle", "Manage SSH keys")
+
+	user := getUserLogged(ctx)
+
+	var key = new(models.SSHKey)
+	if err := ctx.Bind(key); err != nil {
+		return errorRes(400, "Cannot bind data", err)
+	}
+
+	if err := ctx.Validate(key); err != nil {
+		addFlash(ctx, validationMessages(&err), "error")
+		return redirect(ctx, "/ssh-keys")
+	}
+
+	key.UserID = user.ID
+
+	pubKey, _, _, _, err := ssh.ParseAuthorizedKey([]byte(key.Content))
+	if err != nil {
+		addFlash(ctx, "Invalid SSH key", "error")
+		return redirect(ctx, "/ssh-keys")
+	}
+
+	sha := sha256.Sum256(pubKey.Marshal())
+	key.SHA = base64.StdEncoding.EncodeToString(sha[:])
+
+	if err := models.AddSSHKey(key); err != nil {
+		return errorRes(500, "Cannot add SSH key", err)
+	}
+
+	addFlash(ctx, "SSH key added", "success")
+	return redirect(ctx, "/ssh-keys")
+}
+
+func sshKeysDelete(ctx echo.Context) error {
+	user := getUserLogged(ctx)
+	keyId, err := strconv.Atoi(ctx.Param("id"))
+
+	if err != nil {
+		return redirect(ctx, "/ssh-keys")
+	}
+
+	key, err := models.GetSSHKeyByID(uint(keyId))
+
+	if err != nil || key.UserID != user.ID {
+		return redirect(ctx, "/ssh-keys")
+	}
+
+	if err := models.RemoveSSHKey(key); err != nil {
+		return errorRes(500, "Cannot delete SSH key", err)
+	}
+
+	addFlash(ctx, "SSH key deleted", "success")
+	return redirect(ctx, "/ssh-keys")
+}