Sinipelto
/
banan-os
forked from Bananymous/banan-os
1
0
Fork 0
Code Pull Requests Activity

Shell: Cleanup code by defining argument types as nested types

This commit is contained in:
Bananymous 2024-10-15 23:44:06 +03:00
parent c40f244dff
commit 8317bb13ca
3 changed files with 29 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
userspace/programs/Shell/CommandTypes.cpp
View File

@ -139,12 +139,3 @@ BAN::ErrorOr<BAN::String> CommandArgument::evaluate(Execute& execute) const
return evaluated; return evaluated;
} }
BAN::ErrorOr<BAN::Vector<BAN::String>> SingleCommand::evaluate_arguments(Execute& execute) const
{
BAN::Vector<BAN::String> result;
TRY(result.reserve(arguments.size()));
for (const auto& arugment : arguments)
TRY(result.push_back(TRY(arugment.evaluate(execute))));
return result;
}

4
userspace/programs/Shell/CommandTypes.h
View File

@ -24,6 +24,8 @@
struct CommandTree; struct CommandTree;
class Execute; class Execute;
struct CommandArgument
{
struct FixedString struct FixedString
{ {
COMMAND_RULE5(FixedString, value); COMMAND_RULE5(FixedString, value);
@ -42,8 +44,6 @@ struct BuiltinVariable
BAN::String value; BAN::String value;
}; };
struct CommandArgument
{
using ArgumentPart = using ArgumentPart =
BAN::Variant< BAN::Variant<
FixedString, FixedString,

20
userspace/programs/Shell/TokenParser.cpp
View File

@ -171,7 +171,7 @@ BAN::ErrorOr<CommandArgument::ArgumentPart> TokenParser::parse_backslash(bool is
auto token = read_token(); auto token = read_token();
FixedString fixed_string; CommandArgument::FixedString fixed_string;
switch (token.type()) switch (token.type())
{ {
case Token::Type::EOF_: case Token::Type::EOF_:
@ -218,14 +218,14 @@ BAN::ErrorOr<CommandArgument::ArgumentPart> TokenParser::parse_dollar()
[](BAN::String& string) -> BAN::ErrorOr<CommandArgument::ArgumentPart> [](BAN::String& string) -> BAN::ErrorOr<CommandArgument::ArgumentPart>
{ {
if (string.empty()) if (string.empty())
return CommandArgument::ArgumentPart(EnvironmentVariable()); return CommandArgument::ArgumentPart(CommandArgument::EnvironmentVariable());
if (isdigit(string.front())) if (isdigit(string.front()))
{ {
size_t number_len = 1; size_t number_len = 1;
while (number_len < string.size() && isdigit(string[number_len])) while (number_len < string.size() && isdigit(string[number_len]))
number_len++; number_len++;
BuiltinVariable builtin; CommandArgument::BuiltinVariable builtin;
TRY(builtin.value.append(string.sv().substring(0, number_len))); TRY(builtin.value.append(string.sv().substring(0, number_len)));
for (size_t i = 0; i < number_len; i++) for (size_t i = 0; i < number_len; i++)
string.remove(0); string.remove(0);
@ -243,7 +243,7 @@ BAN::ErrorOr<CommandArgument::ArgumentPart> TokenParser::parse_dollar()
case '?': case '?':
case '!': case '!':
{ {
BuiltinVariable builtin; CommandArgument::BuiltinVariable builtin;
TRY(builtin.value.push_back(string.front())); TRY(builtin.value.push_back(string.front()));
string.remove(0); string.remove(0);
return CommandArgument::ArgumentPart(BAN::move(builtin)); return CommandArgument::ArgumentPart(BAN::move(builtin));
@ -255,7 +255,7 @@ BAN::ErrorOr<CommandArgument::ArgumentPart> TokenParser::parse_dollar()
while (env_len < string.size() && (isalnum(string[env_len]) || string[env_len] == '_')) while (env_len < string.size() && (isalnum(string[env_len]) || string[env_len] == '_'))
env_len++; env_len++;
EnvironmentVariable environment; CommandArgument::EnvironmentVariable environment;
TRY(environment.value.append(string.sv().substring(0, env_len))); TRY(environment.value.append(string.sv().substring(0, env_len)));
for (size_t i = 0; i < env_len; i++) for (size_t i = 0; i < env_len; i++)
string.remove(0); string.remove(0);
@ -263,7 +263,7 @@ BAN::ErrorOr<CommandArgument::ArgumentPart> TokenParser::parse_dollar()
return CommandArgument::ArgumentPart(BAN::move(environment)); return CommandArgument::ArgumentPart(BAN::move(environment));
} }
FixedString fixed_string; CommandArgument::FixedString fixed_string;
TRY(fixed_string.value.push_back('$')); TRY(fixed_string.value.push_back('$'));
return CommandArgument::ArgumentPart(BAN::move(fixed_string)); return CommandArgument::ArgumentPart(BAN::move(fixed_string));
}; };
@ -281,7 +281,7 @@ BAN::ErrorOr<CommandArgument::ArgumentPart> TokenParser::parse_dollar()
case Token::Type::SingleQuote: case Token::Type::SingleQuote:
case Token::Type::Whitespace: case Token::Type::Whitespace:
{ {
FixedString fixed_string; CommandArgument::FixedString fixed_string;
TRY(fixed_string.value.push_back('$')); TRY(fixed_string.value.push_back('$'));
return CommandArgument::ArgumentPart(BAN::move(fixed_string)); return CommandArgument::ArgumentPart(BAN::move(fixed_string));
} }
@ -289,7 +289,7 @@ BAN::ErrorOr<CommandArgument::ArgumentPart> TokenParser::parse_dollar()
{ {
consume_token(); consume_token();
BuiltinVariable builtin_variable; CommandArgument::BuiltinVariable builtin_variable;
TRY(builtin_variable.value.push_back('$')); TRY(builtin_variable.value.push_back('$'));
return CommandArgument::ArgumentPart(BAN::move(builtin_variable)); return CommandArgument::ArgumentPart(BAN::move(builtin_variable));
} }
@ -348,7 +348,7 @@ BAN::ErrorOr<CommandArgument::ArgumentPart> TokenParser::parse_single_quote()
{ {
ASSERT(read_token().type() == Token::Type::SingleQuote); ASSERT(read_token().type() == Token::Type::SingleQuote);
FixedString fixed_string; CommandArgument::FixedString fixed_string;
for (auto token = read_token();; token = read_token()) for (auto token = read_token();; token = read_token())
{ {
switch (token.type()) switch (token.type())
@ -380,6 +380,8 @@ BAN::ErrorOr<CommandArgument::ArgumentPart> TokenParser::parse_single_quote()
BAN::ErrorOr<CommandArgument> TokenParser::parse_argument() BAN::ErrorOr<CommandArgument> TokenParser::parse_argument()
{ {
using FixedString = CommandArgument::FixedString;
const auto token_type = peek_token().type(); const auto token_type = peek_token().type();
if (!can_parse_argument_from_token_type(token_type)) if (!can_parse_argument_from_token_type(token_type))
return unexpected_token_error(token_type); return unexpected_token_error(token_type);