SimpleCToC.h

#pragma once

#include <iostream>
#include <vector>
#include <fstream>
#include "keywords.h"
#include "ArgParser.h"
using namespace std;
string helpstr = "Welcome to scc, the simpleC compiler.\nflags:\n-h,--help display this message.\n-t,--transpile: default mode, transpiles SimpleC to C\n-c,--compile: transpile, then compile SimpleC with clang.\n-r,--run: compile with clang, then run the output.\nusage:\nscc [inputfile] [-t, -c, -r] (extra flags)";
void replaceAll(std::string& str, const std::string& from, const std::string& to) {
    if (from.empty())
        return;
    size_t start_pos = 0;
    while ((start_pos = str.find(from, start_pos)) != std::string::npos) {
        str.replace(start_pos, from.length(), to);
        start_pos += to.length();
    }
}
string readFile(const string& fileName)
{
    ifstream ifs(fileName.c_str(), ios::in | ios::binary | ios::ate);

    ifstream::pos_type fileSize = ifs.tellg();
    if (fileSize < 0)
        return std::string();

        ifs.seekg(0, ios::beg);

    vector<char> bytes(fileSize);
    ifs.read(&bytes[0], fileSize);

    return string(&bytes[0], fileSize);
}

vector<string> split(const string& str, const string& delim)
{
vector<string> tokens;
size_t prev = 0, pos = 0;
do
{
    pos = str.find(delim, prev);
    if (pos == string::npos) pos = str.length();
    string token = str.substr(prev, pos - prev);
    if (!token.empty()) tokens.push_back(token);
    prev = pos + delim.length();
} while (pos < str.length() && prev < str.length());
return tokens;
}
std::string Process(std::string strtp) {
    InitKeywords();
    string tmp = strtp;
    vector<string> spltmp = split(tmp, string("\n"));
    for (int it = 0; it != spltmp.size(); it++)
    {
        string line = spltmp[it];
        if(line.find("#include") != string::npos){
          spltmp[it].append("\n");
        }
        if (line.find("Call ") != string::npos)
        {
            spltmp[it].append(";");
        }
        if (line.find(" = ") != string::npos)
        {
            spltmp[it].append(";");
        }        
        if (line.find("declare ") != string::npos)
        {
            vector<string> splline = split(spltmp[it], string(" "));
            string nline = string(splline[1].append(string(" ").append(splline[2])).append("("));
            for (int  i = 3; i < splline.size(); i++)
            {
                nline.append(string(" ") + splline[i]);
            }
            spltmp[it] = nline;
            spltmp[it].append("){");
        }
        for(int i = 0; i <= 9; i++)
        {
        	keyword var = keywords[i];
            replaceAll(spltmp[it], var.scword, var.cword);
        }
    }
    string rtnval = string(); rtnval +=string("#include <boopes.h>\n");
    for (int it = 0; it != spltmp.size(); it++)
    {
        rtnval += spltmp[it];
    }
    return rtnval;
}
int run(char** argv, int argc){
    int code = 0;
    std::string command = std::string(" ");
    command += "clang ";
    Arguments args = getArgs(argv, argc);
    if (args.isHelp) { printf(helpstr.c_str()); return 0; }
    for(std::string i : args.iFiles){
        command += i + ".c";
        command += " ";
    }
    if(args.oFile == ""){args.oFile = "a.out";}
    command += "-o ";
    command += args.oFile;
    printf("output file: %s\n", args.oFile.c_str());

    printf("%s\n", to_string(args.ctype).c_str());
    switch(args.ctype){
        case TRANSPILE:
            for(std::string s : args.iFiles){
                ofstream of = ofstream(s + ".c");
                of << Process(readFile(s));
                of.close();
            }
            break;
        case COMPILE:
            for(std::string s : args.iFiles){
                ofstream of = ofstream(s + ".c");
                of << Process(readFile(s));
                of.close();
                command += s + ".c";
            }
            code = system(command.c_str());
            return code;
            break;
        case RUN:
	    for(std::string s : args.iFiles){
	    ofstream of = ofstream(s + ".c");
		of << Process(readFile(s));
		of.close();
	    }
	    printf("%s", command.c_str());
            code = system(command.c_str());
            code += system("./a.out");
            return code;
            break;
        default:
            std::cout << "ERROR: Compile Type Not Specified, Quitting." << std::endl;
            exit(1);
            break;
    }
    return code;
}