/*
    Copyright 2012 Wolfgang Thaller.

    This file is part of Retro68.

    Retro68 is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    Retro68 is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with Retro68.  If not, see <http://www.gnu.org/licenses/>.
*/

#include <iostream>
#include <unistd.h>
#include <algorithm>
#include <vector>
#include <string>
#include <functional>
#include <errno.h>
#include <fstream>

using namespace std::placeholders;

#ifdef USE_BOOST_REGEX
#include <boost/regex.hpp>
namespace rx = boost;
#else
#include <regex>
namespace rx = std;
#endif

int main(int argc, char *argv[])
{
	std::vector<std::string> argv2;
	std::vector<const char*> argv3;
	
	std::copy(argv, argv+argc, std::back_inserter(argv2));
	argv2[0] += ".real";
	
	std::string inputFileName;
	std::string tempFileName;
	
	char tempFileTemplate[] = "/tmp/filteredXXXXXX.s";
	int fd = mkstemps(tempFileTemplate, 2);
	tempFileName = tempFileTemplate;
	
	for(auto& p : argv2)
	{
		if(p.substr(p.length()-2) == ".s")
		{
			inputFileName = p;
			p = tempFileName;
			//std::cerr << "Temp file: " << tempFileName << std::endl;
		}
	}
	
	{
		std::ifstream in(inputFileName);
		std::ofstream out(tempFileName);
		
		std::string wordS = "[0-9a-f][0-9a-f][0-9a-f][0-9a-f]";
		rx::regex jsr("\t(jsr|jra) __magic_inline_(" + wordS + "(_" + wordS + ")*)");
		rx::regex word(wordS);
		rx::regex rts("\trts");
		rx::regex instruction("\t[a-z]+.*");
		rx::regex macsbug("# macsbug symbol");

		bool hadRts = false;
		bool macsbugSymbol = false, macsbugSymbol1;
		while(in)
		{
			std::string line;
			std::getline(in, line);
			if(!in)
				break;
			
			macsbugSymbol1 = false;

			rx::smatch match;
			// ******* 1. __magic_inline hack for Toolbox calls
			// 
			// PrepareHeaders.hs converts calls to ONEWORDINLINE, TWOWORDINLINE etc. functions
			// into "jsr __magic_inline_a123" or "jsr __magic_inline_1234_5678", etc.
			// This is converted to a series of dc.w
			if(rx::regex_match(line, match, jsr))
			{
				const rx::sregex_iterator end;
				for (rx::sregex_iterator p(line.cbegin(), line.cend(), word);
					p != end;
					++p)
				{
					out << "\tdc.w 0x" << p->str() << std::endl;
				}
				if(match[1] == "jra")
				{
					out << "rts\n";
					hadRts = true;
				}
			}

			// ******* 2. strip unneeded extra rts from "# macsbug symbol" paragraphs
			//
			// GCC is patche to add something like this to the end of every function:
			// # macsbug symbol
			//     rts
			//     .byte 132
			//     .ascii	"main"
			//     .align 2,0
			//     .short 0
			//     .size	main, .-main
			//
			// The rts is usually redundant as most functions already end in RTS.
			// The following removes the RTS if there has already been an RTS immediately before.
			else if(rx::regex_match(line, macsbug))
			{
				out << line << std::endl;
				macsbugSymbol1 = true;
			}
			else if(rx::regex_match(line, rts))
			{
				if(!macsbugSymbol || !hadRts)
					out << line << std::endl;
				hadRts = true;
			}
			else if(rx::regex_match(line, instruction))
			{
				hadRts = false;
				out << line << std::endl;
			}

			// Pass through everything else.
			else
				out << line << std::endl;

			macsbugSymbol = macsbugSymbol1;
		}
	}
	
	close(fd);
	
	std::transform(argv2.begin(), argv2.end(), std::back_inserter(argv3), 
		[](const std::string& str) { return str.c_str(); });

	argv3.push_back(NULL);
	
	execvp(argv3[0], const_cast<char*const*>( argv3.data() ));
	perror("execvp");
	
	return 1;
}