Initial code commit

internal/wireguard/client.go

@@ -0,0 +1,90 @@
+package wireguard
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+)
+
+type ClientProfile struct {
+	OS           string
+	Architecture  string
+	ConfigContent string
+	ConfigPath    string
+}
+
+func GenerateClientProfiles(serverConfig *ServerConfig, serverIP string, count int) ([]ClientProfile, error) {
+	var profiles []ClientProfile
+
+	platforms := []struct {
+		OS          string
+		Architecture string
+	}{
+		{"linux", "amd64"},
+		{"linux", "arm64"},
+		{"darwin", "amd64"},
+		{"darwin", "arm64"},
+		{"windows", "amd64"},
+	}
+
+	numProfiles := count
+	if numProfiles <= 0 {
+		numProfiles = len(platforms)
+	}
+
+	for i := 0; i < numProfiles; i++ {
+		platform := platforms[i%len(platforms)]
+		
+		clientPrivateKey, clientPublicKey, err := GenerateClientKeyPair()
+		if err != nil {
+			return nil, fmt.Errorf("failed to generate client key pair: %w", err)
+		}
+
+		clientAddress := fmt.Sprintf("10.0.0.%d/24", i+2)
+		clientConfig := GenerateClientConfig(
+			serverIP,
+			serverConfig.Port,
+			serverConfig.PublicKey,
+			clientPrivateKey,
+			clientPublicKey,
+			clientAddress,
+			"0.0.0.0/0",
+		)
+
+		profile := ClientProfile{
+			OS:           platform.OS,
+			Architecture: platform.Architecture,
+			ConfigContent: clientConfig.ToConfigFile(),
+		}
+
+		profiles = append(profiles, profile)
+	}
+
+	return profiles, nil
+}
+
+func SaveClientProfile(profile ClientProfile, outputDir string) (string, error) {
+	if err := os.MkdirAll(outputDir, 0755); err != nil {
+		return "", fmt.Errorf("failed to create output directory: %w", err)
+	}
+
+	var filename string
+	switch profile.OS {
+	case "darwin":
+		filename = fmt.Sprintf("wg-%s-%s.conf", profile.OS, profile.Architecture)
+	case "linux":
+		filename = fmt.Sprintf("wg-%s-%s.conf", profile.OS, profile.Architecture)
+	case "windows":
+		filename = fmt.Sprintf("wg-%s-%s.conf", profile.OS, profile.Architecture)
+	default:
+		filename = fmt.Sprintf("wg-%s-%s.conf", profile.OS, profile.Architecture)
+	}
+
+	configPath := filepath.Join(outputDir, filename)
+
+	if err := os.WriteFile(configPath, []byte(profile.ConfigContent), 0600); err != nil {
+		return "", fmt.Errorf("failed to write client config: %w", err)
+	}
+
+	return configPath, nil
+}