Initial commit

include/cmd/arg.hpp

@@ -0,0 +1,224 @@
+#pragma once
+
+#include <string>
+#include <sstream>
+#include <fstream>
+#include <vector>
+#include <unordered_map>
+#include <iostream>
+
+#include "ast/ast_component.hpp"
+#include "util/input.hpp"
+
+namespace fsh{
+
+class _Argument {
+
+public:
+    virtual void svalue(std::shared_ptr<CommandArgumentNode> can) {
+        svalue(can->gvalue(), can->gtoken_type());
+    }
+    virtual void svalue(const std::string& can, const Lexer::TokenType& type = Lexer::TokenType::FLAG) {}
+
+    template<typename T>
+    static std::shared_ptr<_Argument> create() {return std::make_shared<T>();}
+protected:
+    _Argument(){} 
+};
+
+template<typename T>
+class Argument : public _Argument {
+    public:
+    Argument() {} 
+
+
+    virtual void svalue(const std::string& val, const Lexer::TokenType& type) override{
+        if constexpr (std::is_same_v<T, std::string>) {
+            value = val;
+        } else {
+            std::stringstream ss_val(val);
+            if(!(ss_val >> value)) {
+                throw std::invalid_argument("Incorrect type");
+            }
+        }
+        is_string_literal = type == Lexer::TokenType::STRING_LITERAL; 
+    }
+
+    virtual T& gvalue() {return value;}
+    static T& get(std::shared_ptr<_Argument> a) {
+        return std::dynamic_pointer_cast<Argument<T> >(a)->gvalue();
+    }
+    private:
+    bool is_string_literal;
+    T value;
+};
+
+class ArgInput : public _Argument {
+    public:
+    ArgInput() {} 
+
+    virtual void svalue(const std::string& val, const Lexer::TokenType& type) override{
+        const std::string& txt = val;
+        if(type == Lexer::TokenType::STRING_LITERAL) {
+            str = std::stringstream(txt);
+        } else {
+            file = std::ifstream(txt, std::ios::in);
+            if(!*file) {
+                throw std::runtime_error("Failed to open file");
+            }
+        }
+    }
+
+    virtual std::istream& gvalue(){
+        if(str)
+            return *str;
+        return *file;
+    }
+
+    static std::istream& get(std::shared_ptr<_Argument> a) {
+        return std::dynamic_pointer_cast<ArgInput>(a)->gvalue();
+    }
+
+    private:
+    std::optional<std::stringstream> str;
+    std::optional<std::ifstream> file;
+    
+};
+
+class ArgManager {
+    friend class ArgFactory;
+
+    using PosArgs = std::vector<std::shared_ptr<_Argument> >;
+    using FlagOpts = std::unordered_map< std::string ,std::shared_ptr<_Argument> >;
+    public:
+    ArgManager() {}
+
+
+    template<typename T>
+    std::optional<T> get(const int id) {
+        if(id < pos_argument.size()) return Argument<T>::get(pos_argument[id]);
+        return std::make_optional<T>();
+    }
+
+    std::istream& get_input(const int id) {
+        if(id < pos_argument.size()) return ArgInput::get(pos_argument[id]);
+        return util::cin;
+    }
+
+    template<typename T>
+    std::optional<T> get(const std::string& id) {
+        if(flags.find(id) != flags.end()) return Argument<T>::get(flags[id]);
+        return std::make_optional<T>();
+    }
+
+    bool get(const std::string& id) {
+        return flags.find(id) != flags.end();
+    }
+
+    void push_arg(std::shared_ptr<_Argument> arg) {
+        pos_argument.push_back(arg);
+    }
+
+    void push_flag(const std::string &id,std::shared_ptr<_Argument> arg = nullptr) {
+        flags[id] = arg;
+    }
+    
+    private:
+    PosArgs pos_argument;
+    FlagOpts flags;
+};
+
+
+class ArgFactory {
+    
+    using FlagNode = std::optional<std::string>&;
+    using ArgNodes = std::vector<std::shared_ptr<CommandArgumentNode > >;
+
+
+    struct ArgRule 
+    {
+        std::shared_ptr<_Argument> (*build) (void);
+        bool mandatory;
+        bool extends;
+    };
+
+    struct FlagRule
+    {
+        std::shared_ptr<_Argument> (*build) (void);
+        bool specialinput; //Not implemented
+        bool capturing;
+        bool extends; //Not implemented
+    };
+    
+
+    public:
+    template<typename T>
+    void add_rule(bool mandatory, bool extends = false) {
+        pos_arg_rules.push_back({&_Argument::create<Argument<T> >, mandatory, extends});
+    }
+
+    void add_input_rule() {
+        has_input = true;
+        pos_arg_rules.push_back({&_Argument::create<ArgInput>, false, false});
+    }
+
+    template<typename T = bool >
+    void add_rule(const std::string name, bool capturing = false) {
+        flag_rules[name] = {_Argument::create<Argument<T> >, false, capturing, false};
+    }
+
+    void parse(ArgManager& manager, ArgNodes& vec, FlagNode flag) {
+        if(flag) {
+            parse_flag(manager, flag);
+        }
+        int i = 0;
+        for(const auto& arg : vec) {
+            std::shared_ptr<_Argument> a;
+            if(i >= pos_arg_rules.size()) throw std::invalid_argument("More arguments then excpected");
+            manager.push_arg(build_arg(pos_arg_rules[i].build, arg));
+            if(!pos_arg_rules[i].extends) i++;
+        }
+    }
+
+    bool ghas_input() {return has_input;}
+
+    private:
+    std::vector<ArgRule> pos_arg_rules;
+    std::unordered_map<std::string, FlagRule> flag_rules;
+    bool has_input = false;
+
+    void parse_flag(ArgManager& manager, FlagNode flag) {
+        const std::string f = *flag;
+        int f_sz = f.size();
+        for(const auto& [key, rule] : flag_rules) {
+            const std::string_view k(key);
+            if(k == f) {
+                manager.push_flag(f);
+                return;
+            }
+
+            if(rule.capturing && f == k.substr(0, f_sz)) {
+                auto arg = build_arg(rule.build, (std::string) k.substr(f_sz));
+                manager.push_flag(f, arg);
+                return;
+            }
+        }
+        throw std::invalid_argument("Invalid flag");
+    }
+
+    std::shared_ptr<_Argument> build_arg(std::shared_ptr<_Argument> (*build_func) (void),const std::string& str) {
+        auto arg = build_func();
+        arg->svalue(str);
+        return arg;
+    }
+
+    std::shared_ptr<_Argument> build_arg(std::shared_ptr<_Argument> (*build_func) (void),
+        std::shared_ptr<CommandArgumentNode> cmd_arg) {
+        auto arg = build_func();
+        arg->svalue(cmd_arg);
+        return arg;
+    }
+
+};
+
+}

include/cmd/cmd.hpp

@@ -0,0 +1,53 @@
+#pragma once
+
+#include <unordered_map>
+
+#include "cmd/cmd_base.hpp"
+#include "cmd/cmd_wc.hpp"
+#include "cmd/cmd_time.hpp"
+#include "cmd/cmd_date.hpp"
+#include "cmd/cmd_echo.hpp"
+#include "cmd/cmd_touch.hpp"
+#include "cmd/cmd_misc.hpp"
+
+namespace fsh
+{
+
+class CommandRegistry {
+
+    public:
+
+    static CommandRegistry& instance() {
+        static CommandRegistry cmd_registry;
+        return cmd_registry;
+    }
+
+    Command& get(const std::string n) {
+        if(cmds.find(n) == cmds.end()) {
+            throw std::runtime_error("Command not found");
+        }
+        return *(cmds[n]);
+    }
+
+    Command& operator[](const std::string n) {
+        return get(n);
+    }
+
+
+    private:
+    CommandRegistry() {
+        cmds["wc"   ]  = Command::register_cmd<CmdWc>();       
+        cmds["date" ]  = Command::register_cmd<CmdDate>();
+        cmds["time" ]  = Command::register_cmd<CmdTime>();
+        cmds["echo" ]  = Command::register_cmd<CmdEcho>();
+        cmds["exit" ]  = Command::register_cmd<CmdExit>();
+        cmds["touch"]  = Command::register_cmd<CmdTouch>();
+        cmds["debug"]  = Command::register_cmd<CmdPrintTree>();
+    }
+
+    std::unordered_map<std::string, std::unique_ptr<Command> > cmds;  
+
+};
+
+
+}

include/cmd/cmd_base.hpp

@@ -0,0 +1,41 @@
+#pragma once
+
+#include <array>
+#include <string>
+#include <sstream>
+#include <unordered_map>
+#include <memory>
+
+#include "ast/ast.hpp"
+#include "cmd/arg.hpp"
+
+namespace fsh{
+
+class Command {
+
+    using FlagNode = std::optional<std::string>&;
+    using ArgNodes = std::vector<std::shared_ptr<CommandArgumentNode > >;
+
+    public:
+    void execute(FlagNode flag, ArgNodes& vec, std::istream& in, std::ostream& out);
+
+    template <typename T>
+    static std::unique_ptr<Command> register_cmd() {
+        std::unique_ptr<Command> cmd = std::make_unique<T>();
+        cmd->register_flags();
+        return cmd;
+    }
+
+    protected:
+
+    virtual void register_flags() {};
+    virtual void run(std::istream& in, std::ostream& out, ArgManager& args) {};
+
+    ArgFactory& get_factory() {return arg_factory;}
+
+    private:
+    ArgFactory arg_factory;
+};
+
+
+}

include/cmd/cmd_date.hpp

@@ -0,0 +1,21 @@
+#pragma once
+
+#include <ctime>
+#include <iomanip>
+
+#include "cmd/cmd_base.hpp"
+
+
+namespace fsh {
+
+class CmdDate : public Command {
+    
+    protected:
+    virtual void run(std::istream& in, std::ostream& out, ArgManager& args) override {
+        std::time_t time = std::time(nullptr);
+        out << std::asctime(std::localtime(&time));
+    }
+
+};
+
+}

include/cmd/cmd_echo.hpp

@@ -0,0 +1,26 @@
+#pragma once
+
+#include "cmd/cmd_base.hpp"
+
+namespace fsh {
+
+class CmdEcho : public Command {
+    
+    protected:
+    virtual void register_flags() override {
+        ArgFactory& factory = get_factory();
+        factory.add_input_rule();
+    }
+
+    virtual void run(std::istream& in, std::ostream& out, ArgManager& args) override {
+        std::string s;
+        std::string o;
+        while(getline(in, s)) {
+            o += s + "\n";
+        } 
+        out << o;
+    }
+
+};
+
+}

include/cmd/cmd_misc.hpp

@@ -0,0 +1,43 @@
+#pragma once
+
+#include <sstream>
+
+#include "cmd/cmd_base.hpp"
+#include "fsh.hpp"
+#include "ast/ast.hpp"
+
+
+namespace fsh {
+
+class CmdExit : public Command {
+    
+    protected:
+    virtual void run(std::istream& in, std::ostream& out, ArgManager& args) override {
+        fsh::instance().environment["EXITING"] = "1";
+    }
+
+};
+
+class CmdPrintTree : public Command {
+    
+    protected:
+
+    virtual void register_flags() override {
+        auto& factory = get_factory();
+        factory.add_input_rule();
+    }
+
+
+    virtual void run(std::istream& in, std::ostream& out, ArgManager& args) override {
+        std::string line;
+        std::getline(in, line);
+
+        auto tokens = Lexer::process(line);
+        auto ast = AstFactory::generate_ast(tokens);
+
+        ast->print(0);
+    }
+
+};
+
+}

include/cmd/cmd_time.hpp

@@ -0,0 +1,20 @@
+#pragma once
+
+#include <ctime>
+#include <iomanip>
+
+#include "cmd/cmd_base.hpp"
+
+namespace fsh {
+
+class CmdTime : public Command {
+    
+    protected:
+    virtual void run(std::istream& in, std::ostream& out, ArgManager& args) override {
+        std::time_t time = std::time(nullptr);
+        out << std::put_time(std::localtime(&time), "%T%n");
+    }
+
+};
+
+}

include/cmd/cmd_touch.hpp

@@ -0,0 +1,24 @@
+#pragma once
+
+#include "cmd/cmd_base.hpp"
+
+namespace fsh {
+
+class CmdTouch : public Command {
+    
+    protected:
+    virtual void register_flags() override {
+        ArgFactory& factory = get_factory();
+        factory.add_rule<std::string>(true);
+    }
+
+    virtual void run(std::istream& in, std::ostream& out, ArgManager& args) override {
+        auto name = *(args.get<std::string>(0));
+        if(std::ifstream(name, std::ios::in).good() || !std::ofstream(name, std::ios::out).is_open()) {
+            throw std::runtime_error("File exists");
+        }
+    }
+
+};
+
+}

include/cmd/cmd_wc.hpp

@@ -0,0 +1,37 @@
+#pragma once
+
+#include "cmd/cmd_base.hpp"
+
+namespace fsh {
+
+class CmdWc : public Command {
+    
+    protected:
+    virtual void register_flags() override {
+        ArgFactory& factory = get_factory();
+        factory.add_input_rule();
+        factory.add_rule("-c");
+        factory.add_rule("-w");
+    }
+
+    virtual void run(std::istream& in, std::ostream& out, ArgManager& args) override {
+        int i = 0;
+        char c;
+        if(args.get("-c")) {
+            while(in.get(c)) i++;
+            out << i << "\n";
+        } else if (args.get("-w")) {
+            bool prev_space = true;
+            while(in.get(c)) {
+                if(std::isspace(c)) { prev_space = true; }
+                else if(prev_space) { prev_space = false; i++;}
+            }
+            out << i << "\n";
+        } else {
+            throw std::invalid_argument("Didn't provide flag");
+        }
+    }
+
+};
+
+}