37 lines
895 B
C++
37 lines
895 B
C++
#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");
|
|
}
|
|
}
|
|
|
|
};
|
|
|
|
} |