add extensions and lib

1 files changed, 2 insertions, 10 deletions

diff --git a/src/arg.lua b/src/arg.lua
index a0acd9c..1254b71 100644
--- a/src/arg.lua
+++ b/src/arg.lua
@@ -1,7 +1,7 @@
 
 local function help(opts)
 
-	io.write(arg[0])
+	io.write(Path:new(arg[0]):name())
 	io.write(" ")
 
 	for _, v in ipairs(opts) do
@@ -48,14 +48,6 @@ local function help(opts)
 	os.exit()
 end
 
-function string:flag()
-	if self:sub(1, 2) ~= "--" then
-		return nil
-	end
-
-	return self:sub(3)
-end
-
 return function (opts)
 	local args = {}
 
@@ -76,7 +68,7 @@ return function (opts)
 	for _, value in ipairs(arg) do
 		local key = nil
 		if current_flag == nil then
-			local flag = value:flag()
+			local flag = value:flag_name()
 			if flag == nil then
 				local option = params[current_index]
 				current_index = current_index + 1