Add "literal" option to addCommand, to treat all args after argCount as a literal string

content/commands.js

@@ -80,6 +80,7 @@ liberator.Command = function (specs, description, action, extraInfo) //{{{
     this.options     = extraInfo.options || [];
     this.bang        = extraInfo.bang || false;
     this.count       = extraInfo.count || false;
+    this.literal     = extraInfo.literal || false;
 
     this.isUserCommand   = extraInfo.isUserCommand || false;
     this.replacementText = extraInfo.replacementText || null;
@@ -99,7 +100,7 @@ liberator.Command.prototype = {
         // we use our args parser instead of passing a string to the callback
         if (this.options.length > 0 || this.argCount)
         {
-            args = liberator.commands.parseArgs(args, this.options, this.argCount);
+            args = liberator.commands.parseArgs(args, this.options, this.argCount, false, this.literal);
             if (args == null)
                 return false;
         }
@@ -158,21 +159,27 @@ liberator.Commands = function () //{{{
 
     var exCommands = [];
 
-    function getMatchingUserCommands(name)
+    function getMatchingUserCommands(name, filter)
     {
         var matches = [];
-        for (let i = 0; i < exCommands.length; i++)
+        outer:
+        for (let [,cmd] in Iterator(exCommands))
         {
-            if (exCommands[i].isUserCommand)
+            if (cmd.isUserCommand)
             {
+                for (let [k, v] in Iterator(filter))
+                {
+                    if (v != null && cmd[k] != v)
+                        continue outer;
+                }
                 if (name)
                 {
-                    if (exCommands[i].name.match("^" + name))
+                    if (cmd.name.match("^" + name))
-                        matches.push(exCommands[i]);
+                        matches.push(cmd);
                 }
                 else
                 {
-                    matches.push(exCommands[i]);
+                    matches.push(cmd);
                 }
             }
         }
@@ -318,7 +325,7 @@ liberator.Commands = function () //{{{
         // @param allowUnknownOptions: -foo won't result in an error, if -foo isn't
         //                             specified in "options"
         // TODO: should it handle comments?
-        parseArgs: function (str, options, argCount, allowUnknownOptions, complete) //{{{
+        parseArgs: function (str, options, argCount, allowUnknownOptions, literal, complete) //{{{
         {
             function quoteArg(quote)
             {
@@ -428,12 +435,13 @@ liberator.Commands = function () //{{{
 
             if (!argCount)
                 argCount = "*";
+            if (literal)
+                argCount = parseInt(argCount);
 
             var args = {};       // parsed options
             args.arguments = []; // remaining arguments
             args.string = str;   // for access to the unparsed string
-            // FIXME: quick hack! Best way to do this? -- djk
+            args.literalArg = "";
-            args.argumentsString = "";
 
             var invalid = false;
             var onlyArgumentsRemaining = allowUnknownOptions || false; // after a -- has been found
@@ -593,9 +601,11 @@ liberator.Commands = function () //{{{
                     }
                 } //}}}
 
-                // FIXME: quick hack! -- djk
+                if (literal && args.arguments.length == argCount)
-                if (!args.argumentsString)
+                {
-                    args.argumentsString = str.substr(i);
+                    args.literalArg = sub;
+                    break;
+                }
 
                 // if not an option, treat this token as an argument
                 var [count, arg] = getNextArg(sub);
@@ -616,6 +626,9 @@ liberator.Commands = function () //{{{
                 i += count; // hopefully count is always > 0, otherwise we get an endless loop
             }
 
+            if (literal)
+                return args;
+
             // check for correct number of arguments
             if (args.arguments.length == 0 && (argCount == "1" || argCount == "+"))
             {
@@ -691,35 +704,7 @@ liberator.Commands = function () //{{{
     ////////////////////// COMMANDS ////////////////////////////////////////////////
     /////////////////////////////////////////////////////////////////////////////{{{
 
-    // TODO: Vim allows commands to be defined without {rep} if there are {attr}s
+    function userCommand(args, special, count, modifiers)
-    // specified - useful?
-    commandManager.add(["com[mand]"],
-        "List and define commands",
-        function (args, special)
-        {
-            if (args.argumentsString)
-            {
-                let matches = args.argumentsString.match(/^(\w+)(?:\s+(.+))?$/);
-
-                if (!matches)
-                {
-                    liberator.echoerr("E182: Invalid command name");
-                    return;
-                }
-
-                var [cmd, rep] = [matches[1], matches[2]];
-            }
-
-            if (rep)
-            {
-                let nargsOpt = args["-nargs"] || "0";
-                let bangOpt  = "-bang"  in args;
-                let countOpt = "-count" in args;
-
-                if (!liberator.commands.addUserCommand(
-                        [cmd],
-                        "User defined command",
-                        function (args, special, count, modifiers)
     {
         function replaceTokens(token)
         {
@@ -753,13 +738,37 @@ liberator.Commands = function () //{{{
             return quote ? '"' + ret.replace('"', '\\"', "g") + '"' : ret;
         }
 
-                            liberator.execute(rep.replace(/<(?:q-)?[a-z]+>/g, replaceTokens));
+        liberator.execute(this.replacementText.replace(/<(?:q-)?[a-z]+>/g, replaceTokens));
-                        },
+    }
+
+    // TODO: Vim allows commands to be defined without {rep} if there are {attr}s
+    // specified - useful?
+    commandManager.add(["com[mand]"],
+        "List and define commands",
+        function (args, special)
+        {
+            let cmd = args.arguments[0];
+            if (cmd != null && /\W/.test(cmd))
+            {
+                liberator.echoerr("E182: Invalid command name");
+                return;
+            }
+
+            if (args.literalArg)
+            {
+                let nargsOpt = args["-nargs"] || "0";
+                let bangOpt  = "-bang"  in args;
+                let countOpt = "-count" in args;
+
+                if (!liberator.commands.addUserCommand(
+                        [cmd],
+                        "User defined command",
+                        userCommand,
                         {
                             argCount: nargsOpt,
                             bang: bangOpt,
                             count: countOpt,
-                            replacementText: rep
+                            replacementText: args.literalArg
                         },
                         special)
                     )
@@ -769,7 +778,12 @@ liberator.Commands = function () //{{{
             }
             else
             {
-                let cmds = getMatchingUserCommands(cmd);
+                let filter = {
+                    argCount: args["-nargs"],
+                    bang:     "-bang" in args && true,
+                    count:    args["-count"],
+                };
+                let cmds = getMatchingUserCommands(cmd, filter);
 
                 if (cmds.length > 0)
                 {
@@ -789,6 +803,8 @@ liberator.Commands = function () //{{{
             }
         },
         {
+            literal: true,
+            argCount: 1,
             bang: true,
             completer: function (filter) liberator.completion.userCommand(filter),
             options: [