From 0a7731cf0d49d36a09f9b8ad804d517e872b3411 Mon Sep 17 00:00:00 2001
From: ishuah <kariuki@ishuah.com>
Date: Mon, 8 Jan 2018 11:20:01 +0300
Subject: [PATCH] cryptdecode: added option to return encrypted file names.
 Fixes #1923

---
 cmd/cryptdecode/cryptdecode.go | 60 ++++++++++++++++++++++++++++++----
 crypt/crypt.go                 |  5 +++
 2 files changed, 58 insertions(+), 7 deletions(-)

diff --git a/cmd/cryptdecode/cryptdecode.go b/cmd/cryptdecode/cryptdecode.go
index c1c130857..fe46dd10f 100644
--- a/cmd/cryptdecode/cryptdecode.go
+++ b/cmd/cryptdecode/cryptdecode.go
@@ -10,8 +10,15 @@ import (
 	"github.com/spf13/cobra"
 )
 
+// Options set by command line flags
+var (
+	Reverse = false
+)
+
 func init() {
 	cmd.Root.AddCommand(commandDefinition)
+	flags := commandDefinition.Flags()
+	fs.BoolVarP(flags, &Reverse, "reverse", "", Reverse, "Reverse cryptdecode, encrypts filenames")
 }
 
 var commandDefinition = &cobra.Command{
@@ -21,25 +28,44 @@ var commandDefinition = &cobra.Command{
 rclone cryptdecode returns unencrypted file names when provided with
 a list of encrypted file names. List limit is 10 items.
 
+If you supply the --reverse flag, it will return encrypted file names.
+
 use it like this
 
 	rclone cryptdecode encryptedremote: encryptedfilename1 encryptedfilename2
+
+	rclone cryptdecode --reverse encryptedremote: filename1 filename2
 `,
 	Run: func(command *cobra.Command, args []string) {
 		cmd.CheckArgs(2, 11, command, args)
 		fsrc := cmd.NewFsSrc(args)
-		cmd.Run(false, false, command, func() error {
-			return cryptDecode(fsrc, args[1:])
-		})
+		if Reverse {
+			cmd.Run(false, false, command, func() error {
+				return cryptEncode(fsrc, args[1:])
+			})
+		} else {
+			cmd.Run(false, false, command, func() error {
+				return cryptDecode(fsrc, args[1:])
+			})
+		}
 	},
 }
 
+// Check if fsrc is a crypt
+func assertCryptFs(fsrc fs.Fs) (*crypt.Fs, error) {
+	fcrypt, ok := fsrc.(*crypt.Fs)
+	if !ok {
+		return nil, errors.Errorf("%s:%s is not a crypt remote", fsrc.Name(), fsrc.Root())
+	}
+	return fcrypt, nil
+}
+
 // cryptDecode returns the unencrypted file name
 func cryptDecode(fsrc fs.Fs, args []string) error {
-	// Check if fsrc is a crypt
-	fcrypt, ok := fsrc.(*crypt.Fs)
-	if !ok {
-		return errors.Errorf("%s:%s is not a crypt remote", fsrc.Name(), fsrc.Root())
+	fcrypt, err := assertCryptFs(fsrc)
+
+	if err != nil {
+		return err
 	}
 
 	output := ""
@@ -57,3 +83,23 @@ func cryptDecode(fsrc fs.Fs, args []string) error {
 
 	return nil
 }
+
+// cryptEncode returns the encrypted file name
+func cryptEncode(fsrc fs.Fs, args []string) error {
+	fcrypt, err := assertCryptFs(fsrc)
+
+	if err != nil {
+		return err
+	}
+
+	output := ""
+
+	for _, fileName := range args {
+		encryptedFileName := fcrypt.EncryptFileName(fileName)
+		output += fmt.Sprintln(fileName, "\t", encryptedFileName)
+	}
+
+	fmt.Printf(output)
+
+	return nil
+}
diff --git a/crypt/crypt.go b/crypt/crypt.go
index dcd69798c..911454524 100644
--- a/crypt/crypt.go
+++ b/crypt/crypt.go
@@ -445,6 +445,11 @@ func (f *Fs) UnWrap() fs.Fs {
 	return f.Fs
 }
 
+// EncryptFileName returns an encrypted file name
+func (f *Fs) EncryptFileName(fileName string) string {
+	return f.cipher.EncryptFileName(fileName)
+}
+
 // DecryptFileName returns a decrypted file name
 func (f *Fs) DecryptFileName(encryptedFileName string) (string, error) {
 	return f.cipher.DecryptFileName(encryptedFileName)