Builtin: Implement unset array support

2 files changed, 60 insertions, 1 deletions

diff --git a/src/builtins/tests/unset_tests.cpp b/src/builtins/tests/unset_tests.cpp
new file mode 100644
index 0000000..8350507
--- /dev/null
+++ b/src/builtins/tests/unset_tests.cpp
@@ -0,0 +1,50 @@
+/*
+   Please use git log for copyright holder and year information
+
+   This file is part of libbash.
+
+   libbash is free software: you can redistribute it and/or modify
+   it under the terms of the GNU General Public License as published by
+   the Free Software Foundation, either version 2 of the License, or
+   (at your option) any later version.
+
+   libbash is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with libbash.  If not, see <http://www.gnu.org/licenses/>.
+*/
+///
+/// \file unset_tests.cpp
+/// \brief series of unit tests for unset builtin
+///
+#include <boost/lexical_cast.hpp>
+#include <gtest/gtest.h>
+
+#include "core/interpreter.h"
+#include "cppbash_builtin.h"
+#include "exceptions.h"
+
+TEST(unset_builtin_test, normal)
+{
+  interpreter walker;
+
+  walker.set_value("var", "foo");
+  EXPECT_EQ(0, cppbash_builtin::exec("unset", {"var"}, std::cout, std::cerr, std::cin, walker));
+  EXPECT_STREQ("", walker.resolve<std::string>("var").c_str());
+}
+
+TEST(unset_builtin_test, array)
+{
+  interpreter walker;
+
+  walker.set_value("array", "foo bar", 2);
+  EXPECT_EQ(0, cppbash_builtin::exec("unset", {"array"}, std::cout, std::cerr, std::cin, walker));
+  EXPECT_STREQ("", walker.resolve<std::string>("array", 2).c_str());
+
+  walker.set_value("array", "foo bar", 2);
+  EXPECT_EQ(0, cppbash_builtin::exec("unset", {"array[2]"}, std::cout, std::cerr, std::cin, walker));
+  EXPECT_STREQ("", walker.resolve<std::string>("array", 2).c_str());
+}
diff --git a/src/builtins/unset_builtin.cpp b/src/builtins/unset_builtin.cpp
index d8178f9..2eaafc2 100644
--- a/src/builtins/unset_builtin.cpp
+++ b/src/builtins/unset_builtin.cpp
@@ -26,6 +26,8 @@
 
 #include "core/interpreter.h"
 
+using namespace boost::xpressive;
+
 int unset_builtin::exec(const std::vector<std::string>& bash_args)
 {
   if(bash_args.empty())
@@ -51,7 +53,14 @@ int unset_builtin::exec(const std::vector<std::string>& bash_args)
    * */
     for_each(bash_args.front() == "-v" ? bash_args.begin() + 1 : bash_args.begin(),
              bash_args.end(),
-             [&](const std::string& name) { _walker.unset(name); });
+             [&](const std::string& name) {
+                static const sregex index_pattern = sregex::compile("^(.*)\\[(\\d*)\\]$");
+                smatch match;
+                if(regex_match(name, match, index_pattern))
+                  _walker.unset(match[1], boost::lexical_cast<unsigned int>(match[2]));
+                else
+                  _walker.unset(name);
+             });
 
   return 0;
 }