#include <stdio.h>
#include <stdlib.h>
#include <sstream>

#include "../../memory.h"
#include "../../loader.h"
#include "../../assembler.h"
#include "../../expression.h"


#define GTASM_MAJOR_VERSION "1.0"
#define GTASM_MINOR_VERSION "6R"
#define GTASM_VERSION_STR "gtasm v" GTASM_MAJOR_VERSION "." GTASM_MINOR_VERSION


int main(int argc, char* argv[])
{
    if(argc != 2  &&  argc != 3)
    {
        fprintf(stderr, "%s\n", GTASM_VERSION_STR);
        fprintf(stderr, "Usage:   gtasm <input filename> <optional include path>\n");
        return 1;
    }

    std::string name = std::string(argv[1]);
    if(name.find(".vasm") == name.npos  &&  name.find(".gasm") == name.npos   &&  name.find(".asm") == name.npos  &&  name.find(".s") == name.npos)
    {
        fprintf(stderr, "Wrong file extension in %s : must be one of : '.vasm' or '.gasm' or '.asm' or '.s'\n", name.c_str());
        return 1;
    }

    Loader::initialise();
    Expression::initialise();
    Assembler::initialise();

    // Optional include path
    uint16_t address = DEFAULT_EXEC_ADDRESS;
    std::string includepath = (argc == 3) ? std::string(argv[2]) : ".";
    Assembler::setIncludePath(includepath);

    // Path and name
    size_t slash = name.find_last_of("\\/");
    std::string path = (slash != std::string::npos) ? name.substr(0, slash) : ".";
    Expression::replaceText(path, "\\", "/");
    name = (slash != std::string::npos) ? name.substr(slash + 1) : name;
    std::string filename = path + "/" + name;
    Loader::setFilePath(filename);

    if(!Assembler::assemble(filename, address)) return 1;

    // Create gt1 format
    Loader::Gt1File gt1File;
    gt1File._loStart = LO_BYTE(address);
    gt1File._hiStart = HI_BYTE(address);
    Loader::Gt1Segment gt1Segment;
    gt1Segment._loAddress = LO_BYTE(address);
    gt1Segment._hiAddress = HI_BYTE(address);

    bool hasRomCode = false;
    Assembler::ByteCode byteCode;
    while(!Assembler::getNextAssembledByte(byteCode))
    {
        if(byteCode._isRomAddress) hasRomCode = true; 

        // Custom address
        if(byteCode._isCustomAddress)
        {
            if(gt1Segment._dataBytes.size())
            {
                // Previous segment
                gt1Segment._segmentSize = uint8_t(gt1Segment._dataBytes.size());
                gt1File._segments.push_back(gt1Segment);
                gt1Segment._dataBytes.clear();
            }

            address = byteCode._address;
            gt1Segment._isRomAddress = byteCode._isRomAddress;
            gt1Segment._loAddress = LO_BYTE(address);
            gt1Segment._hiAddress = HI_BYTE(address);
        }

        gt1Segment._dataBytes.push_back(byteCode._data);
    }

    // Last segment
    if(gt1Segment._dataBytes.size())
    {
        gt1Segment._segmentSize = uint8_t(gt1Segment._dataBytes.size());
        gt1File._segments.push_back(gt1Segment);
    }

    // Don't save gt1 file for any asm files that contain native rom code
    std::string gt1FileName;
    if(!hasRomCode  &&  !saveGt1File(filename, gt1File, gt1FileName)) return 1;

    Loader::printGt1Stats(gt1FileName, gt1File, false);

    return 0;
}