feat: Automatically upload non-empty result files to R2 after writing and log their public URLs.

Author: Mosaad Sallam

internal/modules/r2storage/r2storage.go

@@ -123,6 +123,36 @@ func IsEnabled() bool {
 	return isEnabled && r2Config != nil && r2Config.Enabled
 }
 
+// UploadResultFileAndLog uploads a result file to R2 only if it's non-empty.
+// The r2Key is the object key (e.g. "results/example.com/subs/subdomains.txt").
+// It prints a prominent [R2-RESULT-URL] log line so AI agents can extract the URL.
+// Returns the public URL or an empty string if the file was empty or R2 is not enabled.
+func UploadResultFileAndLog(localPath, r2Key string) string {
+	if !IsEnabled() {
+		return ""
+	}
+
+	info, err := os.Stat(localPath)
+	if err != nil {
+		return "" // File doesn't exist
+	}
+	if info.Size() == 0 {
+		log.Printf("[R2] ⏭️  Skipping empty file: %s", localPath)
+		return ""
+	}
+
+	publicURL, err := UploadFile(localPath, r2Key, true) // skipTimestamp=true, key already has domain/path
+	if err != nil {
+		log.Printf("[R2] ⚠️  Failed to upload result file %s: %v", localPath, err)
+		return ""
+	}
+
+	// Print a very visible marker line — AI agents and log parsers use this
+	log.Printf("[R2-RESULT-URL] %s -> %s", localPath, publicURL)
+	fmt.Printf("\n🔗 R2 Result: %s\n   URL: %s\n\n", localPath, publicURL)
+	return publicURL
+}
+
 // UploadFile uploads a file to R2 and returns the public URL
 // If skipTimestamp is true, the file is uploaded without a timestamp prefix (for cache/backups)
 func UploadFile(filePath, objectKey string, skipTimestamp bool) (string, error) {

internal/modules/utils/file.go

@@ -3,14 +3,19 @@ package utils
 import (
 	"context"
 	"fmt"
+	"log"
 	"os"
 	"path/filepath"
 	"strings"
 	"time"
+
+	"github.com/h0tak88r/AutoAR/internal/modules/r2storage"
 )
 
 // WriteLines writes lines to a file (one per line)
-// Creates parent directories if they don't exist
+// Creates parent directories if they don't exist.
+// After writing, if R2 storage is enabled and the file is non-empty,
+// it automatically uploads the file and prints the public URL.
 func WriteLines(path string, lines []string) error {
 	if err := os.MkdirAll(filepath.Dir(path), 0755); err != nil {
 		return err
@@ -19,9 +24,36 @@ func WriteLines(path string, lines []string) error {
 	if len(lines) > 0 {
 		data += "\n"
 	}
-	return os.WriteFile(path, []byte(data), 0644)
+	if err := os.WriteFile(path, []byte(data), 0644); err != nil {
+		return err
+	}
+
+	// Auto-upload to R2 if enabled and file is non-empty
+	if len(lines) > 0 && r2storage.IsEnabled() {
+		go uploadResultAsync(path)
+	}
+
+	return nil
 }
 
+// uploadResultAsync uploads a result file to R2 in the background.
+// The R2 key strips the leading "/" and uses the full local path as key.
+func uploadResultAsync(path string) {
+	// Derive R2 key from the path:
+	// Strip absolute prefix up to "new-results" so the key is e.g.
+	//   new-results/example.com/subs/subdomains.txt
+	r2Key := path
+	if idx := strings.Index(path, "new-results/"); idx >= 0 {
+		r2Key = path[idx:]
+	} else if strings.HasPrefix(r2Key, "/") {
+		r2Key = r2Key[1:]
+	}
+	r2storage.UploadResultFileAndLog(path, r2Key)
+	log.Printf("[R2] ✅ Auto-uploaded result file: %s", path)
+}
+
+
+
 // ReadLines reads lines from a file
 func ReadLines(path string) ([]string, error) {
 	data, err := os.ReadFile(path)