Refactor code to use funny stuff

This commit is contained in:
Maxwell Jeffress 2024-12-27 20:55:19 +11:00
parent 313ecb962f
commit 2910d630d0

View File

@ -1,355 +1,384 @@
#include <iostream>
#include <cstdlib>
#include <fstream>
#include <vector>
#include <string>
#include <cstring>
#include <map>
#include <variant>
#include <vector>
using namespace std;
bool checkForDec(string str) {
enum class VarType {
INTEGER,
DECIMAL,
STRING,
UNKNOWN
};
struct Variable {
VarType type;
variant<int, double, string> value;
Variable() : type(VarType::UNKNOWN) {}
Variable(VarType t, const variant<int, double, string>& v) : type(t), value(v) {}
string toString() const {
switch(type) {
case VarType::INTEGER:
return to_string(get<int>(value));
case VarType::DECIMAL:
return to_string(get<double>(value));
case VarType::STRING:
return get<string>(value);
default:
return "unknown";
}
}
};
class Interpreter {
private:
map<string, Variable> variables;
bool isVerbose = false;
const vector<string> builtInFunctions = {
"exit", "log", "type", "run", "verbose", "str", "dec", "int"
};
const vector<string> operators = {
"+", "-", "*", "/"
};
const vector<string> incrementors = {
"++", "--"
};
const vector<string> modifiers = {
"+=", "-=", "*=", "/="
};
const vector<string> comparitors = {
"==", ">", ">=", "=<", "!=", "!<", "!>", "!=<", "!>="
};
bool checkForDec(const string& str) {
try {
if (stoi(str) * 1.0 == stod(str)) {
return false;
} else {
return true;
}
} catch (const std::invalid_argument& e) {
return false;
} catch (const std::out_of_range& e) {
return stoi(str) * 1.0 != stod(str);
} catch (...) {
return false;
}
}
bool checkForInt(string str) {
bool checkForInt(const string& str) {
try {
if (stoi(str) * 1.0 == stod(str)) {
return true;
} else {
return false;
}
} catch (const std::invalid_argument& e) {
return false;
} catch (const std::out_of_range& e) {
return stoi(str) * 1.0 == stod(str);
} catch (...) {
return false;
}
}
bool checkForStr(string str) {
//if (string[0] == '"' && string.back() == '"') return true;
}
string encodeInt(int input) {
}
string encodeDec(double input) {
}
string encodeStr(string input) {
}
int decodeInt(string input) {
}
double decodeDec(string input) {
}
string decodeStr(string input) {
}
string findType(string str, vector<string> terms, vector<string> list) {
for (int i = 0; i < list.size(); i++) {
if (str == terms[i]) {
return list[i];
}
}
return "Error";
}
void error(string str) {
void error(const string& str) {
cout << "\u001b[31m Error running code: " << str << "\u001b[39m" << endl;
}
bool isVerbose = false;
void verbose(string str) {
void verbose(const string& str) {
if (isVerbose) cout << "\u001b[36m Info: " << str << "\u001b[39m" << endl;
}
int main(int argc, char *argv[]) {
cout << "Oktolang interpreter\n";
vector<string> vars;
vector<string> varTypes;
vector<string> varContents;
vector<string> functions = {"exit", "log", "type", "run", "verbose", "str", "dec", "int"};
bool running = true;
string in;
int lineCounter = 0;
int maxLines = 0;
vector<string> lines;
bool fromFile = false;
if (argc > 1) {
fromFile = true;
ifstream inFile(argv[1]);
string inLines;
while (getline(inFile, inLines)) {
if (inLines[0] == '#') continue;
lines.push_back(inLines);
maxLines ++;
}
}
while (running) {
if (argc == 1) {
cout << "> ";
getline(cin, in);
} else {
if (lineCounter == maxLines) break;
in = lines[lineCounter];
lineCounter ++;
}
vector<string> terms = {""};
vector<string> types;
int termNumber = 0;
vector<pair<string, string>> tokenize(const string& input) {
vector<pair<string, string>> tokens;
string currentToken;
bool stringOpen = false;
// Split everything up
verbose("Splitting everything up...");
for (int i = 0; i < in.length(); i++) {
if (in[i] == ' ' && !stringOpen) {
if (!terms[termNumber].empty()) {
termNumber++;
terms.push_back("");
for (char c : input) {
if (c == ' ' && !stringOpen) {
if (!currentToken.empty()) {
tokens.push_back({currentToken, getTokenType(currentToken)});
currentToken.clear();
}
}
else {
terms[termNumber] += in[i];
}
if (in[i] == '"') {
stringOpen = !stringOpen;
}
}
// Assign types to everything
verbose("Assigning types...");
for (int i = 0; i < terms.size(); i++) {
bool typeFound = false;
string keyword = terms[i];
for (int x = 0; x < functions.size(); x++) {
if (keyword == functions[x]) {
types.push_back("function");
typeFound = true;
}
}
if (!typeFound) {
for (int x = 0; x < vars.size(); x++) {
if (keyword == vars[x]) {
types.push_back("variable");
typeFound = true;
}
}
}
if (!typeFound) {
if (keyword == "+") { types.push_back("plus"); }
else if (keyword == "-") { types.push_back("minus"); }
else if (keyword == "*") { types.push_back("multiply"); }
else if (keyword == "/") { types.push_back("divide"); }
else if (keyword == "++") { types.push_back("increment"); }
else if (keyword == "--") { types.push_back("decrement"); }
else if (keyword == "=") { types.push_back("equals"); }
else if (keyword[0] == '"') { types.push_back("str"); }
else if (checkForInt(keyword)) { types.push_back("int"); }
else if (checkForDec(keyword)) { types.push_back("dec"); }
else { types.push_back("unknown"); }
}
}
// Start interpreting code
verbose("Running code...");
if (terms[0] == "exit") {
verbose("Exit command run");
if (terms.size() < 2) {
return 0;
} else if (stoi(terms[1])) {
return stoi(terms[1]);
} else if (terms[1] == "0") {
return 0;
} else {
error("Exit arguments must be integers");
currentToken += c;
}
} else if (terms[0] == "log") {
if (c == '"') stringOpen = !stringOpen;
}
if (!currentToken.empty()) {
tokens.push_back({currentToken, getTokenType(currentToken)});
}
return tokens;
}
auto handleVariable(Variable var) {
}
double doDecMath() {
}
string getTokenType(const string& token) {
for (const auto& func : builtInFunctions) {
if (token == func) return "function";
}
for (const auto& inc : incrementors) {
if (token == inc) return "incrementor";
}
for (const auto& mod : modifiers) {
if (token == mod) return "modifier";
}
for (const auto& op : operators) {
if (token == op) return "operator";
}
for (const auto& comp : comparitors) {
if (token == comp) return "comparitor";
}
if (variables.find(token) != variables.end()) return "variable";
if (token == "=") return "equals";
if (token[0] == '"') return "str";
if (checkForInt(token)) return "int";
if (checkForDec(token)) return "dec";
return "unknown";
}
public:
auto executeCommand(const string& input) {
auto tokens = tokenize(input);
if (tokens.empty()) return 0;
if (tokens[0].first == "log") {
verbose("Log command run");
if (terms.size() > 1) {
if (types[1] == "str" || types[1] == "int" || types[1] == "dec") {
verbose("Logging directly");
cout << terms[1] << endl;
} else if (types[1] == "variable") {
verbose("Logging a variable");
for (int i = 0; i < vars.size(); i++) {
if (vars[i] == terms[1]) {
verbose("Logging variable " + vars[i] + " with type " + varTypes[i] + " and value " + varContents[i]);
cout << varContents[i] << endl;
break;
if (tokens.size() < 2) {
error("log requires an argument");
return 0;
}
const auto& [token, type] = tokens[1];
if (type == "str" || type == "int" || type == "dec") {
cout << token << endl;
}
else if (type == "variable") {
auto it = variables.find(token);
if (it != variables.end()) {
cout << it->second.toString() << endl;
}
}
} else if (types[1] == "function") {
else if (type == "function") {
error("for now, functions cannot be logged");
} else if (types[1] == "unknown") {
}
else if (type == "unknown") {
error("that type is unknown");
}
} else {
error("log requires an argument");
return 0;
}
} else if (terms[0] == "type") {
verbose("Type command run");
if (terms.size() < 2) {
else if (tokens[0].first == "exit") {
if (tokens.size() < 2) {
exit(0);
} else if (tokens.size() > 1) {
const auto& [token, type] = tokens[1];
if (type == "int") {
exit(stoi(token));
} else {
error("exit argument must be integer");
}
}
return 0;
}
else if (tokens[0].first == "type") {
if (tokens.size() < 2) {
error("type requires an argument");
} else {
cout << types[1] << endl;
return 0;
}
} else if (terms[0] == "str") {
verbose("String command run");
if (terms.size() < 2) {
error("variable must have a name");
} else {
if (terms.size() == 3) {
error("when defining a variable, set what the variable means");
} else if (terms[2] != "=") {
error("when defining a variable, use '='");
} else if (types[3] != "str") {
error("you've initialised a string, but set it's value to something else");
} else if (types[1] != "unknown") {
error("variable is already initialised");
} else {
verbose("String is being defined...");
vars.push_back(terms[1]);
varTypes.push_back("str");
varContents.push_back(terms[3]);
if (isVerbose) {
for (int i = 0; i < vars.size(); i++) {
if (vars[i] == terms[1]) {
verbose("Variable " + vars[i] + " with type " + varTypes[i] + " defined as " + varContents[i]);
break;
const auto& [token, type] = tokens[1];
cout << type << endl;
return 0;
}
}
}
}
}
} else if (terms[0] == "dec") {
verbose("Decimal command run");
if (terms.size() < 2) {
error("variable must have a name");
} else {
if (terms.size() == 3) {
error("when defining a variable, set what the variable means");
} else if (terms[2] != "=") {
error("when defining a variable, use '='");
} else if (!(types[3] == "dec" || types[3] == "int")) {
error("you've initialised a decimal, but set it's value to something else");
} else {
verbose("Decimal is being defined...");
vars.push_back(terms[1]);
varTypes.push_back("dec");
varContents.push_back(terms[3]);
if (isVerbose) {
for (int i = 0; i < vars.size(); i++) {
if (vars[i] == terms[1]) {
verbose("Variable " + vars[i] + " with type " + varTypes[i] + " defined as " + varContents[i]);
break;
}
}
}
}
}
} else if (terms[0] == "int") {
verbose("Integer command run");
if (terms.size() < 2) {
error("variable must have a name");
} else {
if (terms.size() == 3) {
error("when defining an integer, set what the variable means");
} else if (terms[2] != "=") {
error("when defining a variable, use '='");
} else if (types[3] != "int") {
error("you've initialised a number, but set it's value to something else");
} else {
verbose("Number is being defined...");
vars.push_back(terms[1]);
varTypes.push_back("int");
varContents.push_back(terms[3]);
if (isVerbose) {
for (int i = 0; i < vars.size(); i++) {
if (vars[i] == terms[1]) {
verbose("Variable " + vars[i] + " with type " + varTypes[i] + " defined as " + varContents[i]);
break;
}
}
}
}
}
} else if (terms[0] == "run") {
if (terms.size() < 2) {
else if (tokens[0].first == "run") {
if (tokens.size() < 2) {
error("run requires an argument");
} else {
string inputStr = terms[1];
return 0;
}
string inputStr = tokens[1].first;
char input[inputStr.length() + 1];
strcpy(input, inputStr.c_str());
int returnCode = system(input);
return 0;
}
} else if (terms[0] == "verbose") {
else if (tokens[0].first == "str") {
verbose("String command run");
if (tokens.size() < 2) {
error("variable must have a name");
return 0;
}
if (tokens.size() < 4) {
error("when defining a variable, set what the variable means");
return 0;
}
const auto& varName = tokens[1].first;
if (tokens[2].first != "=") {
error("when defining a variable, use '='");
return 0;
}
if (variables.find(varName) != variables.end()) {
error("variable is already initialized");
return 0;
}
if (tokens[3].second != "str") {
error("you've initialized a string, but set it's value to a different type");
return 0;
}
string value = tokens[3].first;
value = value.substr(1, value.length() - 2);
variables[varName] = Variable(VarType::STRING, value);
verbose("String variable " + varName + "defined as " + value);
return 0;
}
else if (tokens[0].first == "int") {
verbose("Integer command run");
if (tokens.size() < 2) {
error("variable must have a name");
return 0;
}
if (tokens.size() < 4) {
error("when defining a variable, set what the variable means");
return 0;
}
const auto& varName = tokens[1].first;
if (tokens[2].first != "=") {
error("when defining a variable, use '='");
return 0;
}
if (variables.find(varName) != variables.end()) {
error("variable is already initialized");
return 0;
}
if (tokens[3].second != "int") {
error("you've initialized an integer, but set it's value to a different type");
return 0;
}
int value = stoi(tokens[3].first);
variables[varName] = Variable(VarType::INTEGER, value);
verbose("Integer variable " + varName + "defined as " + to_string(value));
return 0;
}
else if (tokens[0].first == "dec") {
verbose("Decimal command run");
if (tokens.size() < 2) {
error("variable must have a name");
return 0;
}
if (tokens.size() < 4) {
error("when defining a variable, set what the variable means");
return 0;
}
const auto& varName = tokens[1].first;
if (tokens[2].first != "=") {
error("when defining a variable, use '='");
return 0;
}
if (variables.find(varName) != variables.end()) {
error("variable is already initialized");
return 0;
}
if (tokens[3].second != "dec") {
error("you've initialized a decimal, but set it's value to a different type");
return 0;
}
double value = stod(tokens[3].first);
variables[varName] = Variable(VarType::DECIMAL, value);
verbose("Decimal variable " + varName + "defined as " + to_string(value));
return 0;
}
else if (tokens[0].first == "verbose") {
isVerbose = !isVerbose;
} else if (terms[0] == "help") {
cout << "Oktolang Help\nBuilt In Functions:\n help: This current help function\n log: Log something to the command line\n type: Find the type of the input\n str: Define a string\n num: Define a number\n";
} else {
bool variableFound = false;
for (int i = 0; i < vars.size(); i++) {
if (vars[i] == terms[0]) {
variableFound = true;
verbose("Found variable " + vars[i] + " with value " + varContents[i] + " and type " + varTypes[i]);
if (varTypes[i] == "int" || varTypes[i] == "dec") {
if (types[1] == "increment") {
verbose("Incrementing variable " + vars[i] + " with value " + varContents[i]);
if (varTypes[i] == "dec") varContents[i] = to_string(stod(varContents[i]) + 1);
if (varTypes[i] == "int") varContents[i] = to_string(stoi(varContents[i]) + 1);
verbose("New value of " + vars[i] + " is " + varContents[i]);
} else if (types[1] == "equals") {
if (types[3] == "add") {
verbose("Adding " + vars[2] + " and " + vars[4]);
if (types[2] == "variable") {
for (int i = 0; i < vars.size(); i++) {
if (types[2] == vars[i]) {
terms[2] = vars[i];
return 0;
}
break;
else if (tokens[0].first == "help") {
cout << "Oktolang Help\nBuilt In Functions:\n help: This current help function\n log: Log something to the command line\n type: Find the type of the input\n str: Define a string\n int: Define a whole number\n dec: Define a number with a decimal place\n";
return 0;
}
else if (variables.find(tokens[0].first) != variables.end()) {
if (tokens.size() < 2) {
error("Expected an operator");
return 0;
}
if (types[4] == "variable") {
for (int i = 0; i < vars.size(); i++) {
if (types[4] == vars[i]) {
terms[4] = vars[i];
}
break;
}
}
verbose("Adding " + vars[2] + " and " + vars[4]);
}
}
}
break;
}
}
if (!variableFound && terms[0] != "") {
error("I don't know how that works");
if (fromFile) {
return 1;
auto& var = variables[tokens[0].first];
if (var.type == VarType::INTEGER) {
if (tokens[1].second == "incrementor") {
if (tokens[1].first == "++") {
int currentValue = get<int>(var.value);
var.value = currentValue + 1;
} else if (tokens[1].second == "--") {
int currentValue = get<int>(var.value);
var.value = currentValue - 1;
}
} else if (tokens[1].second == "operator") {}
} else if (var.type == VarType::DECIMAL) {
if (tokens[1].second == "incrementor") {
if (tokens[1].first == "++") {
double currentValue = get<double>(var.value);
var.value = currentValue + 1;
} else if (tokens[1].second == "--") {
double currentValue = get<double>(var.value);
var.value = currentValue - 1;
}
}
}
return 0;
}
if (!tokens[0].first.empty()) {
error("I don't know how that works");
return 0;
}
return 0;
}
};
int main(int argc, char *argv[]) {
cout << "Oktolang interpreter\n";
Interpreter interpreter;
if (argc > 1) {
ifstream inFile(argv[1]);
string line;
while (getline(inFile, line)) {
if (line[0] == '#') continue;
if (interpreter.executeCommand(line) != 0) return 1;
}
} else {
string input;
while (true) {
cout << "> ";
getline(cin, input);
if (interpreter.executeCommand(input) != 0) break;
}
}
return 0;
}