feat: Add plugin system

.devcontainer/devcontainer.json

@@ -0,0 +1,10 @@
+{
+	"name": "JetKVM",
+	"image": "mcr.microsoft.com/devcontainers/go:1-1.23-bookworm",
+	"features": {
+		"ghcr.io/devcontainers/features/node:1": {
+			// Should match what is defined in ui/package.json
+			"version": "21.1.0"
+		}
+	}
+}

Makefile

@@ -9,7 +9,7 @@ build_dev: hash_resource
 	GOOS=linux GOARCH=arm GOARM=7 go build -ldflags="-s -w -X kvm.builtAppVersion=$(VERSION_DEV)" -o bin/jetkvm_app cmd/main.go
 
 frontend:
-	cd ui && npm run build:device
+	cd ui && npm ci && npm run build:device
 
 dev_release: build_dev
 	@echo "Uploading release..."

dev_deploy.sh

@@ -3,35 +3,28 @@ set -e
 
 # Function to display help message
 show_help() {
-    echo "Usage: $0 [options] -h <host_ip> -r <remote_ip>"
+    echo "Usage: $0 [options] -r <remote_ip>"
     echo
     echo "Required:"
-    echo "  -h, --host <host_ip>      Local host IP address"
     echo "  -r, --remote <remote_ip>   Remote host IP address"
     echo
     echo "Optional:"
     echo "  -u, --user <remote_user>   Remote username (default: root)"
-    echo "  -p, --port <port>          Python server port (default: 8000)"
     echo "      --help                 Display this help message"
     echo
     echo "Example:"
-    echo "  $0 -h 192.168.0.13 -r 192.168.0.17"
-    echo "  $0 -h 192.168.0.13 -r 192.168.0.17 -u admin -p 8080"
+    echo "  $0 -r 192.168.0.17"
+    echo "  $0 -r 192.168.0.17 -u admin"
     exit 0
 }
 
 # Default values
-PYTHON_PORT=8000
 REMOTE_USER="root"
 REMOTE_PATH="/userdata/jetkvm/bin"
 
 # Parse command line arguments
 while [[ $# -gt 0 ]]; do
     case $1 in
-        -h|--host)
-            HOST_IP="$2"
-            shift 2
-            ;;
         -r|--remote)
             REMOTE_HOST="$2"
             shift 2
@@ -40,10 +33,6 @@ while [[ $# -gt 0 ]]; do
             REMOTE_USER="$2"
             shift 2
             ;;
-        -p|--port)
-            PYTHON_PORT="$2"
-            shift 2
-            ;;
         --help)
             show_help
             exit 0
@@ -57,8 +46,8 @@ while [[ $# -gt 0 ]]; do
 done
 
 # Verify required parameters
-if [ -z "$HOST_IP" ] || [ -z "$REMOTE_HOST" ]; then
-    echo "Error: Host IP and Remote IP are required parameters"
+if [ -z "$REMOTE_HOST" ]; then
+    echo "Error: Remote IP is a required parameter"
     show_help
 fi
 
@@ -69,12 +58,8 @@ make build_dev
 # Change directory to the binary output directory
 cd bin
 
-# Start a Python HTTP server in the background to serve files
-python3 -m http.server "$PYTHON_PORT" &
-PYTHON_SERVER_PID=$!
-
-# Ensure that the Python server is terminated if the script exits unexpectedly
-trap "echo 'Terminating Python server...'; kill $PYTHON_SERVER_PID" EXIT
+# Copy the binary to the remote host
+cat jetkvm_app | ssh "${REMOTE_USER}@${REMOTE_HOST}" "cat > $REMOTE_PATH/jetkvm_app_debug"
 
 # Deploy and run the application on the remote host
 ssh "${REMOTE_USER}@${REMOTE_HOST}" ash <<EOF
@@ -90,23 +75,12 @@ killall jetkvm_app_debug || true
 # Navigate to the directory where the binary will be stored
 cd "$REMOTE_PATH"
 
-# Remove any old binary
-rm -f jetkvm_app
-
-# Download the new binary from the host
-wget ${HOST_IP}:${PYTHON_PORT}/jetkvm_app
-
 # Make the new binary executable
-chmod +x jetkvm_app
-
-# Rename the binary to jetkvm_app_debug
-mv jetkvm_app jetkvm_app_debug
+chmod +x jetkvm_app_debug
 
 # Run the application in the background
 ./jetkvm_app_debug
 
 EOF
 
-# Once the SSH session finishes, shut down the Python server
-kill "$PYTHON_SERVER_PID"
 echo "Deployment complete."

internal/plugin/database.go

@@ -0,0 +1,60 @@
+package plugin
+
+import (
+	"encoding/json"
+	"fmt"
+	"os"
+	"sync"
+)
+
+const databaseFile = pluginsFolder + "/plugins.json"
+
+type PluginDatabase struct {
+	// Map with the plugin name as the key
+	Plugins map[string]*PluginInstall `json:"plugins"`
+
+	saveMutex sync.Mutex
+}
+
+var pluginDatabase = PluginDatabase{}
+
+func (d *PluginDatabase) Load() error {
+	file, err := os.Open(databaseFile)
+	if os.IsNotExist(err) {
+		d.Plugins = make(map[string]*PluginInstall)
+		return nil
+	}
+	if err != nil {
+		return fmt.Errorf("failed to open plugin database: %v", err)
+	}
+	defer file.Close()
+
+	if err := json.NewDecoder(file).Decode(d); err != nil {
+		return fmt.Errorf("failed to decode plugin database: %v", err)
+	}
+
+	return nil
+}
+
+func (d *PluginDatabase) Save() error {
+	d.saveMutex.Lock()
+	defer d.saveMutex.Unlock()
+
+	file, err := os.Create(databaseFile + ".tmp")
+	if err != nil {
+		return fmt.Errorf("failed to create plugin database tmp: %v", err)
+	}
+	defer file.Close()
+
+	encoder := json.NewEncoder(file)
+	encoder.SetIndent("", "  ")
+	if err := encoder.Encode(d); err != nil {
+		return fmt.Errorf("failed to encode plugin database: %v", err)
+	}
+
+	if err := os.Rename(databaseFile+".tmp", databaseFile); err != nil {
+		return fmt.Errorf("failed to move plugin database to active file: %v", err)
+	}
+
+	return nil
+}

internal/plugin/extract.go

@@ -0,0 +1,83 @@
+package plugin
+
+import (
+	"archive/tar"
+	"compress/gzip"
+	"fmt"
+	"io"
+	"os"
+	"path"
+	"path/filepath"
+	"strings"
+
+	"github.com/google/uuid"
+)
+
+const pluginsExtractsFolder = pluginsFolder + "/extracts"
+
+func init() {
+	_ = os.MkdirAll(pluginsExtractsFolder, 0755)
+}
+
+func extractPlugin(filePath string) (*string, error) {
+	file, err := os.Open(filePath)
+	if err != nil {
+		return nil, fmt.Errorf("failed to open file for extraction: %v", err)
+	}
+	defer file.Close()
+
+	var reader io.Reader = file
+	// TODO: there's probably a better way of doing this without relying on the file extension
+	if strings.HasSuffix(filePath, ".gz") {
+		gzipReader, err := gzip.NewReader(file)
+		if err != nil {
+			return nil, fmt.Errorf("failed to create gzip reader: %v", err)
+		}
+		defer gzipReader.Close()
+		reader = gzipReader
+	}
+
+	destinationFolder := path.Join(pluginsExtractsFolder, uuid.New().String())
+	if err := os.MkdirAll(destinationFolder, 0755); err != nil {
+		return nil, fmt.Errorf("failed to create extracts folder: %v", err)
+	}
+
+	tarReader := tar.NewReader(reader)
+
+	for {
+		header, err := tarReader.Next()
+		if err == io.EOF {
+			break
+		}
+		if err != nil {
+			return nil, fmt.Errorf("failed to read tar header: %v", err)
+		}
+
+		// Prevent path traversal attacks
+		targetPath := filepath.Join(destinationFolder, header.Name)
+		if !strings.HasPrefix(targetPath, filepath.Clean(destinationFolder)+string(os.PathSeparator)) {
+			return nil, fmt.Errorf("tar file contains illegal path: %s", header.Name)
+		}
+
+		switch header.Typeflag {
+		case tar.TypeDir:
+			if err := os.MkdirAll(targetPath, os.FileMode(header.Mode)); err != nil {
+				return nil, fmt.Errorf("failed to create directory: %v", err)
+			}
+		case tar.TypeReg:
+			file, err := os.OpenFile(targetPath, os.O_CREATE|os.O_WRONLY, os.FileMode(header.Mode))
+			if err != nil {
+				return nil, fmt.Errorf("failed to create file: %v", err)
+			}
+			defer file.Close()
+
+			if _, err := io.Copy(file, tarReader); err != nil {
+				return nil, fmt.Errorf("failed to extract file: %v", err)
+			}
+		default:
+			return nil, fmt.Errorf("unsupported tar entry type: %v", header.Typeflag)
+		}
+	}
+
+	return &destinationFolder, nil
+}