Merge pull request #2099 from danielflira/mount-helper

fix parameter multiple values

weed/command/fuse.go

@@ -12,120 +12,174 @@ func init() {
 	cmdFuse.Run = runFuse // break init cycle
 }
 
+type parameter struct {
+	name string
+	value string
+}
+
 func runFuse(cmd *Command, args []string) bool {
-	argsLen := len(args)
+	rawArgs := strings.Join(args, " ")
-	options := []string{}
+	rawArgsLen := len(rawArgs)
+	option := strings.Builder{}
+	options := []parameter{}
 
-	// at least target mount path should be passed
+	// first parameter
-	if argsLen < 1 {
+	i := 0
-		return false
+	for i = 0; i < rawArgsLen && rawArgs[i] != ' '; i++ {
+		option.WriteByte(rawArgs[i])
+	}
+	options = append(options, parameter{"arg0", option.String()})
+	option.Reset()
+
+	for i++; i < rawArgsLen; i++ {
+
+		// space separator check for filled option
+		if rawArgs[i] == ' ' {
+			if option.Len() > 0 {
+				options = append(options, parameter{option.String(), "true"})
+				option.Reset()
 			}
 
-	// first option is always target mount path
+		// dash separator read option until next space
-	mountOptions.dir = &args[0]
+		} else if rawArgs[i] == '-' {
+			for i++; i < rawArgsLen && rawArgs[i] != ' '; i++ {
+				option.WriteByte(rawArgs[i])
+			}
+			options = append(options, parameter{option.String(), "true"})
+			option.Reset()
 
-	// scan parameters looking for one or more -o options
+		// equal separator start option with pending value
-	// -o options receive parameters on format key=value[,key=value]...
+		} else if rawArgs[i] == '=' {
-	for i := 0; i < argsLen; i++ {
+			name := option.String()
-		if args[i] == "-o" && i+1 <= argsLen {
+			option.Reset()
-			options = strings.Split(args[i+1], ",")
+
-			i++
+			for i++; i < rawArgsLen && rawArgs[i] != ','; i++ {
+				// double quote separator read option until next double quote
+				if rawArgs[i] == '"' {
+					for i++; i < rawArgsLen && rawArgs[i] != '"'; i++ {
+						option.WriteByte(rawArgs[i])
+					}
+
+				// single quote separator read option until next single quote
+				} else if rawArgs[i] == '\'' {
+					for i++; i < rawArgsLen && rawArgs[i] != '\''; i++ {
+						option.WriteByte(rawArgs[i])
+					}
+
+				// add chars before comma
+				} else if rawArgs[i] != ' ' {
+					option.WriteByte(rawArgs[i])
 				}
 			}
 
-	// for each option passed with -o
+			options = append(options, parameter{name, option.String()})
-	for _, option := range options {
+			option.Reset()
-		// split just first = character
-		parts := strings.SplitN(option, "=", 2)
 
-		// if doesn't key and value skip
+		// comma separator just read current option
-		if len(parts) != 2 {
+		} else if rawArgs[i] == ',' {
-			continue
+			options = append(options, parameter{option.String(), "true"})
+			option.Reset()
+
+		// what is not a separator fill option buffer
+		} else {
+			option.WriteByte(rawArgs[i])
+		}
 	}
 
-		key, value := parts[0], parts[1]
+	// get residual option data
+	if option.Len() > 0 {
+		// add value to pending option
+		options = append(options, parameter{option.String(), "true"})
+		option.Reset()
+	}
 
-		// switch key keeping "weed mount" parameters
+	// scan each parameter
-		switch key {
+	for i := 0; i < len(options); i++ {
+		parameter := options[i]
+
+		switch parameter.name  {
+		case "arg0":
+			mountOptions.dir = &parameter.value
 		case "filer":
-				mountOptions.filer = &value
+			mountOptions.filer = &parameter.value
 		case "filer.path":
-				mountOptions.filerMountRootPath = &value
+			mountOptions.filerMountRootPath = &parameter.value
 		case "dirAutoCreate":
-				if parsed, err := strconv.ParseBool(value); err != nil {
+			if parsed, err := strconv.ParseBool(parameter.value); err != nil {
 				mountOptions.dirAutoCreate = &parsed
 			} else {
 				panic(fmt.Errorf("dirAutoCreate: %s", err))
 			}
 		case "collection":
-				mountOptions.collection = &value
+			mountOptions.collection = &parameter.value
 		case "replication":
-				mountOptions.replication = &value
+			mountOptions.replication = &parameter.value
 		case "disk":
-				mountOptions.diskType = &value
+			mountOptions.diskType = &parameter.value
 		case "ttl":
-				if parsed, err := strconv.ParseInt(value, 0, 32); err != nil {
+			if parsed, err := strconv.ParseInt(parameter.value, 0, 32); err != nil {
 				intValue := int(parsed)
 				mountOptions.ttlSec = &intValue
 			} else {
 				panic(fmt.Errorf("ttl: %s", err))
 			}
 		case "chunkSizeLimitMB":
-				if parsed, err := strconv.ParseInt(value, 0, 32); err != nil {
+			if parsed, err := strconv.ParseInt(parameter.value, 0, 32); err != nil {
 				intValue := int(parsed)
 				mountOptions.chunkSizeLimitMB = &intValue
 			} else {
 				panic(fmt.Errorf("chunkSizeLimitMB: %s", err))
 			}
 		case "concurrentWriters":
-				if parsed, err := strconv.ParseInt(value, 0, 32); err != nil {
+			i++
+			if parsed, err := strconv.ParseInt(parameter.value, 0, 32); err != nil {
 				intValue := int(parsed)
 				mountOptions.concurrentWriters = &intValue
 			} else {
 				panic(fmt.Errorf("concurrentWriters: %s", err))
 			}
 		case "cacheDir":
-				mountOptions.cacheDir = &value
+			mountOptions.cacheDir = &parameter.value
 		case "cacheCapacityMB":
-				if parsed, err := strconv.ParseInt(value, 0, 64); err != nil {
+			if parsed, err := strconv.ParseInt(parameter.value, 0, 64); err != nil {
 				mountOptions.cacheSizeMB = &parsed
 			} else {
 				panic(fmt.Errorf("cacheCapacityMB: %s", err))
 			}
 		case "dataCenter":
-				mountOptions.dataCenter = &value
+			mountOptions.dataCenter = &parameter.value
 		case "allowOthers":
-				if parsed, err := strconv.ParseBool(value); err != nil {
+			if parsed, err := strconv.ParseBool(parameter.value); err != nil {
 				mountOptions.allowOthers = &parsed
 			} else {
 				panic(fmt.Errorf("allowOthers: %s", err))
 			}
 		case "umask":
-				mountOptions.umaskString = &value
+			mountOptions.umaskString = &parameter.value
 		case "nonempty":
-				if parsed, err := strconv.ParseBool(value); err != nil {
+			if parsed, err := strconv.ParseBool(parameter.value); err != nil {
 				mountOptions.nonempty = &parsed
 			} else {
 				panic(fmt.Errorf("nonempty: %s", err))
 			}
 		case "volumeServerAccess":
-				mountOptions.volumeServerAccess = &value
+			mountOptions.volumeServerAccess = &parameter.value
 		case "map.uid":
-				mountOptions.uidMap = &value
+			mountOptions.uidMap = &parameter.value
 		case "map.gid":
-				mountOptions.gidMap = &value
+			mountOptions.gidMap = &parameter.value
 		case "readOnly":
-				if parsed, err := strconv.ParseBool(value); err != nil {
+			if parsed, err := strconv.ParseBool(parameter.value); err != nil {
 				mountOptions.readOnly = &parsed
 			} else {
 				panic(fmt.Errorf("readOnly: %s", err))
 			}
 		case "cpuprofile":
-				mountCpuProfile = &value
+			mountCpuProfile = &parameter.value
 		case "memprofile":
-				mountMemProfile = &value
+			mountMemProfile = &parameter.value
 		case "readRetryTime":
-				if parsed, err := time.ParseDuration(value); err != nil {
+			if parsed, err := time.ParseDuration(parameter.value); err != nil {
 				mountReadRetryTime = &parsed
 			} else {
 				panic(fmt.Errorf("readRetryTime: %s", err))
@@ -160,6 +214,9 @@ var cmdFuse = &Command{
   mount -t fuse./home/user/bin/weed fuse /mnt -o "filer=localhost:8888,filer.path=/"
   mount -t fuse "/home/user/bin/weed#fuse" /mnt -o "filer=localhost:8888,filer.path=/"
 
+  To pass more than one parameter use quotes, example:
+  mount -t weed fuse /mnt -o "filer='192.168.0.1:8888,192.168.0.2:8888',filer.path=/"
+
   To check valid options look "weed mount --help"
   `,
 }