From 75b01c8b38d56f426d5b405ce5a3d8da4d12f459 Mon Sep 17 00:00:00 2001
From: mpeterv <petjamelnik@yandex.ru>
Date: Mon, 20 Jan 2014 20:34:44 +0400
Subject: [PATCH] added actions

---
 src/argparse.lua | 37 ++++++++++++++++++++++++++++---------
 1 file changed, 28 insertions(+), 9 deletions(-)

diff --git a/src/argparse.lua b/src/argparse.lua
index 0906d5a..674dbac 100644
--- a/src/argparse.lua
+++ b/src/argparse.lua
@@ -41,7 +41,8 @@ local Parser = class {
    commands = {},
    require_command = false,
    fields = {
-      "name", "description", "target", "require_command"
+      "name", "description", "target", "require_command",
+      "action"
    }
 }:include(Declarative)
 
@@ -56,7 +57,8 @@ local Argument = class {
    count = 1,
    fields = {
       "name", "description", "target", "args",
-      "minargs", "maxargs", "default", "convert"
+      "minargs", "maxargs", "default", "convert",
+      "action"
    }
 }:include(Declarative)
 
@@ -69,7 +71,7 @@ local Option = Argument:extends {
       "name", "aliases", "description", "target", 
       "args", "minargs", "maxargs", "count",
       "mincount", "maxcount", "default", "convert",
-      "overwrite"
+      "overwrite", "action"
    }
 }
 
@@ -255,6 +257,7 @@ function Parser:parse(args)
    local result = {}
    local invocations = {}
    local passed = {}
+   local com_callbacks = {}
    local cur_option
    local cur_arg_i = 1
    local cur_arg
@@ -337,13 +340,21 @@ function Parser:parse(args)
          else
             parser:error("too few arguments")
          end
-      end
+      else
+         if element == cur_option then
+            cur_option = nil
+         elseif element == cur_arg then
+            cur_arg_i = cur_arg_i+1
+            cur_arg = arguments[cur_arg_i]
+         end
 
-      if element == cur_option then
-         cur_option = nil
-      elseif element == cur_arg then
-         cur_arg_i = cur_arg_i+1
-         cur_arg = arguments[cur_arg_i]
+         if element.action then
+            if element.type == "multi-count" or element.type == "multi-count multi-arg" then
+               element.action(result[element.target][#result[element.target]])
+            else
+               element.action(result[element.target])
+            end
+         end
       end
    end
 
@@ -351,6 +362,10 @@ function Parser:parse(args)
       parser = p:prepare()
       charset = p.charset
 
+      if p.action then
+         table.insert(com_callbacks, p.action)
+      end
+
       for _, option in ipairs(p.options) do
          table.insert(options, option)
 
@@ -490,6 +505,10 @@ function Parser:parse(args)
       )
    end
 
+   for _, callback in ipairs(com_callbacks) do
+      callback(result)
+   end
+
    return result
 end