#pragma once

#include "cmd/cmd_base.hpp"
#include "util/stringliteral.hpp"
#include "fsh.hpp"
#include <string>

namespace fsh {

    class CmdPrompt : public Command {

      protected:
        virtual void register_flags() override {
            ArgFactory& factory = get_factory();
            factory.add_rule<StringLiteral>(1);
        }

        virtual void run(std::istream& in, std::ostream& out, ArgManager& args) override {

            std::string n_prompt = args.get<StringLiteral>(0).value();

            fsh::instance().environment["PROMPT"] = n_prompt;
        }
    };

}