103 lines
2.2 KiB
C++
103 lines
2.2 KiB
C++
|
#include "cmd.h"
|
||
|
|
#include "fatal.h"
|
||
|
|
|
||
|
|
|
||
|
|
void Cmd::registerYourself(CmdServer *cmdServer) {
|
||
|
|
if (cmdServer) {
|
||
|
|
cmdServer->registerCmd(this);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
CmdServer::CmdServer(uint16_t port) :
|
||
|
|
m_server(port), m_client(255), cmd(""), params(""), cmdListIdx(0)
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
void CmdServer::begin() {
|
||
|
|
m_server.begin();
|
||
|
|
}
|
||
|
|
|
||
|
|
void CmdServer::exec() {
|
||
|
|
static uint8_t state = 0;
|
||
|
|
bool done = false;
|
||
|
|
|
||
|
|
m_client = m_server.available();
|
||
|
|
|
||
|
|
if (m_client) {
|
||
|
|
int chi;
|
||
|
|
while ((chi = m_client.read()) != -1) {
|
||
|
|
char ch = (char) chi;
|
||
|
|
switch (state) {
|
||
|
|
case 0:
|
||
|
|
if ((ch != ' ') && (ch != '\n')) {
|
||
|
|
cmd += ch;
|
||
|
|
} else if (ch == '\n') {
|
||
|
|
done = true;
|
||
|
|
} else {
|
||
|
|
state = 1;
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
case 1:
|
||
|
|
if (ch != '\n') {
|
||
|
|
params += ch;
|
||
|
|
} else {
|
||
|
|
done = true;
|
||
|
|
state = 0;
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (done) {
|
||
|
|
parseCommand();
|
||
|
|
cmd = "";
|
||
|
|
params = "";
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void CmdServer::registerCmd(Cmd *cmdObj) {
|
||
|
|
if (cmdListIdx < NUM_OF_COMMANDS) {
|
||
|
|
cmdList[cmdListIdx] = cmdObj;
|
||
|
|
cmdListIdx++;
|
||
|
|
} else {
|
||
|
|
fatal(FATAL_NOT_ENOUGH_CMD_SLOTS);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void CmdServer::parseCommand() {
|
||
|
|
//m_client.println("cmd: " + cmd);
|
||
|
|
//m_client.println("params: " + params);
|
||
|
|
|
||
|
|
if (cmd.equalsIgnoreCase("quit")) {
|
||
|
|
m_client.println("good bye");
|
||
|
|
m_client.flush();
|
||
|
|
m_client.stop();
|
||
|
|
} else if (cmd.equalsIgnoreCase("help")) {
|
||
|
|
for (uint8_t i = 0; i < cmdListIdx; i++) {
|
||
|
|
Cmd *c = cmdList[i];
|
||
|
|
m_client.println(c->getCmdName() + " " + c->getHelp());
|
||
|
|
}
|
||
|
|
m_client.println("HELP help");
|
||
|
|
m_client.println("QUIT quit");
|
||
|
|
} else {
|
||
|
|
// m_client.print("cmdListIdx: "); m_client.println(cmdListIdx);
|
||
|
|
bool found = false;
|
||
|
|
for (uint8_t i = 0; i < cmdListIdx; i++) {
|
||
|
|
// m_client.println("Check: " + cmdList[i]->getCmdName());
|
||
|
|
if (cmdList[i]->getCmdName().equalsIgnoreCase(cmd)) {
|
||
|
|
found = true;
|
||
|
|
// m_client.println("cmd found");
|
||
|
|
cmdList[i]->setClient(&m_client);
|
||
|
|
cmdList[i]->setServer(&m_server);
|
||
|
|
String res = cmdList[i]->exec(params);
|
||
|
|
m_client.println(res);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if (! found)
|
||
|
|
m_client.println("command not found");
|
||
|
|
}
|
||
|
|
}
|