# HG changeset patch
# User Tero Koskinen <tero.koskinen@iki.fi>
# Date 1499203681 -10800
#      Wed Jul 05 00:28:01 2017 +0300
# Node ID fa352f506cefc72bc1ed3107f9cb90de1eafa602
# Parent  d79e9ae599c0cbf8278c817b04bd35a624ddb957
Fix the command completion.

diff --git a/hgcomplete.lua b/hgcomplete.lua
--- a/hgcomplete.lua
+++ b/hgcomplete.lua
@@ -35,18 +35,34 @@
   return files
 end
 
+-- print("")
 -- print("SCRIPT PARAMETERS:")
 -- print("amount of parameters: " .. #arg)
 -- for a,b in pairs(arg)
 -- do
---   print(a .. ":" .. b)
+--    print(a .. ":" .. b)
 -- end
 
-possible_params = { "add", "clone", "commit", "push", "pull", "revert", "rm", "init" }
+possible_params = {
+  "add", "branch",
+  "clone",
+  "commit",
+  "export",
+  "fold",
+  "heads",
+  "merge",
+  "status",
+  "push",
+  "pull",
+  "revert",
+  "rm",
+  "init" }
 
--- print("arg: " .. arg[#arg])
-
-argument_table = string.split(arg[#arg], " ")
+arg_copy = arg[#arg]
+if string.sub(arg_copy, 1, 1) == "\"" and string.sub(arg_copy, string.len(arg_copy)) == "\"" then
+  arg_copy = string.sub(arg_copy, 2, #arg_copy - 1)
+end
+argument_table = string.split(arg_copy, " ")
 
 if #argument_table < 2 then
   -- print "all arguments:"
@@ -76,7 +92,7 @@
 --   print(a .. ":" .. b .."|")
 -- end
 
-if #arg == 4 and #argument_table == 2 then
+if #arg == 4 and #argument_table == 2 and string.sub(arg_copy, string.len(arg_copy)) ~= " " then
   for i,p in pairs(possible_params)
   do
     if string.sub(p, 1, string.len(argument_table[2])) == argument_table[2] then