1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
#include "Shell.class.h"
#include <Binding/Sys.ns.h>
#include <Binding/Process.class.h>
APP(Shell);
Shell::Shell() : ShellApp("Shell.app", "The Melon command line interpreter"), cwd(FS::cwdNode()) {
}
void Shell::setupVars() {
shell_var_t tmp;
tmp.readonly = true;
tmp.value = String::number(pr.getUid());
m_vars.set("UID", tmp);
tmp.value = String::number(pr.getPid());
m_vars.set("PID", tmp);
tmp.value = "/";
m_vars.set("CWD", tmp);
}
int Shell::run() {
String dn = FS::dirname(pr.argv(0));
FSNode shellDir = (dn.empty() ? FS::cwdNode() : FS::find(dn, FS::cwdNode()));
String appletsDir = FS::find("Applets", shellDir).path() + "/";
setupVars();
cwd = FS::cwdNode();
while (1) {
outvt << MVT::setfgcolor(PROMPTS_COLOR) << "{" << MVT::setfgcolor(PROMPTV_COLOR) << cwd.getName() <<
MVT::setfgcolor(PROMPTS_COLOR) << "}: " << MVT::setfgcolor(ENTRY_COLOR) << FLUSH;
m_vars["CWD"].value = cwd.path();
String s = invt.get();
outvt << MVT::setfgcolor(NORMAL_COLOR) << FLUSH;
if (s.contains(EOF)) return 0;
if (s.empty()) continue;
while (s[0] == WChar(" ") or s[0] == WChar("\t")) {
s = s.substr(1, s.size() - 1);
}
if (s.empty()) continue;
if (s[0] == WChar("#")) continue;
//Parse string
Vector<String> cmd;
cmd.push(String());
bool inQuote = false;
for (u32int i = 0; i < s.size(); i++) {
if (s[i] == WChar("'")) {
inQuote = !inQuote;
} else if (s[i] == WChar("\\")) {
i++;
cmd.back() += s[i];
} else if (s[i] == WChar(" ") and !inQuote) {
cmd.push(String());
} else {
cmd.back() += s[i];
}
}
//Look for variables to replace
for (u32int i = 0; i < cmd.size(); i++) {
if (cmd[i][0] == WChar("$")) {
String k = cmd[i].substr(1);
if (m_vars.has(k)) {
cmd[i] = m_vars[k].value;
} else {
outvt << "Unknown variable : " << k << ENDL;
}
}
}
//Run command
if (cmd[0] == "exit") { // * * * INTEGRATED COMMANDS * * * //
if (cmd.size() == 1) return 0;
return cmd[1].toInt();
} else if (cmd[0] == "help") {
Vector<String> args;
args.push("cat");
args.push(shellDir.path());
if (args.back() != "/") args.back() += "/";
args.back() += "Help.txt";
if (!appRun(appletsDir + "cat", args)) outvt << "Could not find cat command to display help file\n";
} else if (cmd[0] == "echo") {
for (u32int i = 1; i < cmd.size(); i++) {
if (i > 1) outvt << " ";
outvt << cmd[i];
}
outvt << ENDL;
} else if (cmd[0] == "cd") {
if (cmd.size() != 2) {
outvt << "Invalid argument count.\n";
} else {
FSNode ncwd = FS::find(cmd[1], cwd);
if (!ncwd.valid()) {
outvt << "No such directory : " << cmd[1] << ENDL;
} else if (ncwd.type() != NT_DIRECTORY) {
outvt << "Not a directory : " << cmd[1] << ENDL;
} else {
ncwd.setCwd();
cwd = ncwd;
}
}
} else { // * * * CALL AN EXTERNAL COMMAND * * * //
if (FS::basename(cmd[0]) != cmd[0]) {
if (!appRun(cmd[0], cmd))
outvt << "No such file : " << cmd[0] << "\n";
} else {
if (!appRun(appletsDir + cmd[0], cmd))
outvt << "Unknown command : " << cmd[0] << "\n";
}
}
}
}
bool Shell::appRun(const String& app, Vector<String>& args) {
Process p = Process::run(app);
if (p.valid()) {
p.setInVT(invt);
p.setOutVT(outvt);
for (u32int i = 1; i < args.size(); i++) {
p.pushArg(args[i]);
}
p.start();
s32int v = p.wait();
outvt << "Return value : " << (s64int)v << ENDL;
return true;
}
return false;
}
|
