added proton drive integration

cmd/root.go

@@ -8,6 +8,7 @@ import (
 	"github.com/spf13/cobra"
 
 	"dirmd/internal/config"
+	"dirmd/internal/drive"
 	"dirmd/internal/renderer"
 	"dirmd/internal/walker"
 )
@@ -30,7 +31,13 @@ Examples:
   dirmd -a -i README.md -o existing.md
 
   # Append another directory
-  dirmd -a -o existing.md ~/repos/anotherproject`,
+  dirmd -a -o existing.md ~/repos/anotherproject
+
+  # Generate and upload to Proton Drive (default remote path /my-files/md/<filename>)
+  dirmd -o output.md ~/repos/myproject --proton-drive
+
+  # Generate and upload to a specific remote path
+  dirmd -o output.md ~/repos/myproject --proton-drive --drive-path /my-files/docs/api.md`,
 		RunE: run,
 	}
 
@@ -41,6 +48,8 @@ Examples:
 	maxDepth       int
 	ignorePatterns []string
 	extensions     []string
+	protonDrive    bool
+	drivePath      string
 )
 
 func Execute() {
@@ -57,18 +66,26 @@ func init() {
 	rootCmd.Flags().IntVar(&maxDepth, "max-depth", config.DefaultMaxDepth, "max directory recursion depth")
 	rootCmd.Flags().StringArrayVar(&ignorePatterns, "ignore", nil, "additional ignore patterns (glob)")
 	rootCmd.Flags().StringArrayVar(&extensions, "extensions", nil, "additional file extensions to skip (repeatable)")
+	rootCmd.Flags().BoolVar(&protonDrive, "proton-drive", false, "upload output to Proton Drive after writing locally")
+	rootCmd.Flags().StringVar(&drivePath, "drive-path", "", "full remote path on Proton Drive (e.g. /my-files/md/report.md). Requires --proton-drive")
 
 	rootCmd.MarkPersistentFlagRequired("output")
 }
 
 func run(cmd *cobra.Command, args []string) error {
+	if drivePath != "" && !protonDrive {
+		return fmt.Errorf("--drive-path requires --proton-drive")
+	}
+
 	cfg := &config.Config{
-		OutputPath: outputPath,
-		AppendMode: appendMode,
-		MaxSize:    maxSize,
-		MaxDepth:   maxDepth,
-		Ignores:    append(config.DefaultIgnores, ignorePatterns...),
-		Extensions: append(config.DefaultExts, extensions...),
+		OutputPath:  outputPath,
+		AppendMode:  appendMode,
+		MaxSize:     maxSize,
+		MaxDepth:    maxDepth,
+		Ignores:     append(config.DefaultIgnores, ignorePatterns...),
+		Extensions:  append(config.DefaultExts, extensions...),
+		ProtonDrive: protonDrive,
+		DrivePath:   drivePath,
 	}
 
 	if singleFile != "" {
@@ -111,6 +128,27 @@ func run(cmd *cobra.Command, args []string) error {
 	}
 
 	fmt.Fprintf(os.Stderr, "wrote %d entries to %s\n", len(result.Entries), outputPath)
+
+	if protonDrive {
+		remotePath := drivePath
+		if remotePath == "" {
+			remotePath = "/my-files/md/" + filepath.Base(outputPath)
+		}
+
+		if err := drive.CheckAuth(); err != nil {
+			fmt.Fprintf(os.Stderr, "error: %v\n", err)
+			fmt.Fprintln(os.Stderr, "local file was written successfully, but upload was skipped")
+			return err
+		}
+
+		if err := drive.Upload(outputPath, remotePath); err != nil {
+			fmt.Fprintf(os.Stderr, "error: %v\n", err)
+			fmt.Fprintln(os.Stderr, "local file was written successfully, but upload failed")
+			return err
+		}
+
+		fmt.Fprintf(os.Stderr, "uploaded %s to %s on Proton Drive\n", outputPath, remotePath)
+	}
+
 	return nil
 }
-