feat(encrypt): search for ssh and gpg keys

Author: officiallyutso

system/backup/search.go

@@ -0,0 +1,132 @@
+package backup
+
+import (
+    "os"
+    "path/filepath"
+    "strings"
+)
+
+// common location of keys - more can be added here
+var StandardKeyLocations = []string{
+    "~/.ssh/",
+    "~/.gnupg/",
+}
+
+// any saved keylcoation
+type KeyLocation struct {
+        Path        string
+
+        Type        string // "ssh", "gpg", "custom" strongs can be stored
+        
+        Files       []string
+        
+        IsDirectory bool
+}
+
+// searches for keys in standard locations
+func searchStandardLocations() ([]KeyLocation, error) {
+    var locations []KeyLocation
+    homeDir, err := os.UserHomeDir()
+    if err != nil {
+        return nil, err
+    }
+
+    for _, location := range StandardKeyLocations {
+        // reeplace ~operator with actual home directory
+        fullPath := strings.Replace(location, "~", homeDir, 1)
+        
+        if _, err := os.Stat(fullPath); os.IsNotExist(err) {
+            continue // skip on dir invalid
+        }
+
+        
+        keyType := determineKeyType(fullPath)
+        files, err := discoverKeyFiles(fullPath)
+        if err != nil {
+            continue
+
+        }
+
+        if len(files) > 0 {
+            locations = append(locations, KeyLocation{
+                
+                Path:        fullPath,
+                Type:        keyType,
+                Files:       files,
+                IsDirectory: true,
+            })
+        }
+    }
+
+    return locations, nil
+}
+
+// determineKeyType identifies the type of keys based on directory path
+func determineKeyType(path string) string {
+    if strings.Contains(path, ".ssh") {
+        return "ssh"
+    }
+    if strings.Contains(path, ".gnupg") {
+        return "gpg"
+    }
+    return "custom"
+}
+
+// discoverKeyFiles finds all key files in a directory
+func discoverKeyFiles(dirPath string) ([]string, error) {
+    var keyFiles []string
+    
+    err := filepath.Walk(dirPath, func(path string, info os.FileInfo, err error) error {
+        if err != nil {
+            return err
+        }
+        
+        if !info.IsDir() && isKeyFile(path, info) {
+            keyFiles = append(keyFiles, path)
+        }
+        
+        return nil
+    })
+    
+    return keyFiles, err
+}
+
+// isKeyFile determines if a file is likely a key file
+func isKeyFile(path string, info os.FileInfo) bool {
+    name := info.Name()
+    
+    // SSH key patterns
+    sshPatterns := []string{
+        "id_rsa", "id_dsa", "id_ecdsa", "id_ed25519",
+        "authorized_keys", "known_hosts", "config",
+    }
+    
+    // GPG key patterns
+    gpgPatterns := []string{
+        "pubring.gpg", "secring.gpg", "trustdb.gpg",
+        "gpg.conf", "gpg-agent.conf",
+    }
+    
+    // check SSH 
+    for _, pattern := range sshPatterns {
+        if strings.Contains(name, pattern) {
+            return true
+        }
+    }
+    
+    // cgeck GPG
+    for _, pattern := range gpgPatterns {
+        if strings.Contains(name, pattern) {
+            return true
+        }
+    }
+    
+    // check for private key extensions - more can be added
+    if strings.HasSuffix(name, ".pub") || 
+       strings.HasSuffix(name, ".pem") || 
+       strings.HasSuffix(name, ".key") {
+        return true
+    }
+    
+    return false
+}