///	
// Written in the D programming language.
/**
This is a build tool,compile *.d to exe or lib,and help to build dfl2 gui (or other you like).
now default DC is dmd ,default platform is windows.

If your DC is dmd, dco can start no config file. 

Compiler dco.d :dmd dco.d -release,and dco ↓,

Usage:
  "use before" run: dco -ini,  config some infos to 'dco.ini' file,';' or '#' means you can do as it.Then copy dco.ini to your PATH: such as dmd's config file:sc.ini.
  And dco.exe can auto copy itsself to EnvPath,such as dmd.exe 's path: dmd2\window\bin.
  after that,you can run the 'dco.exe'  anywhere.
  
For example:
    to get the debug version( -release to get another)

	build some *.d to lib or exe 			: dco ↓
	build for libs 	such as dfl,dgui        : dco  -lib
	build for app.d use dfl2				: dco  -gui
	build app.d use dfl2 for console		: dco  -con
	build  one app.d in many *.d    		: dco app.d
	build lib and copy to libs				: dco -lib -c
	build by custom							: dco -arg -addlib -lib -c

    if your exe's file works on console,you should add '-con' or '-console'. 
    
Copyright: Copyright FrankLIKE 2014-.

License:   $(LGPL-3.0).

Authors:   FrankLIKE

Source: $(dco.d)

version: v0.0.3
Created Time:2014-10-27
Modify Time:2014-10-31~2014-11-3
*/
module dco;
/// dco 
import	std.stdio;
import	std.datetime;
import	std.process; 
import	std.string;
import	std.file;
import	std.path;
import	std.exception;
import  std.json;
import std.exception;

string	strAddArgs,strAddArgsdfl = " -de -w -property -X ";
string	strDebug,strDebugDefault=" -debug";
string	strTargetLflags,strConsole=" -L/su:console:4 ",strWindows = " -L/SUBSYSTEM:WINDOWS ";
string	strTargetLib,SpecialLib = "dfl",strWinLibs=" ole32.lib oleAut32.lib gdi32.lib Comctl32.lib Comdlg32.lib advapi32.lib uuid.lib ws2_32.lib ";// $(DMDInstallDir)windows\\lib\\
string	strDFile;
string	strAddLib;
string	strOtherArgs;
string	strImportDefault = " -I$(DMDInstallDir)windows/import ";
string	strTargetPath,strTargetFileName,strTargetTypeSwitch,targetTypeDefault = "lib";
string	strDCEnv,strDCEnvFile;
SysTime sourceLastUpdateTime,targetTime;

bool	bUseSpecialLib =false,bDebug =true,bBuildSpecialLib =false;
bool	bCopy =false ,bDisplayBuildStr=false,bDisplayCopyInfo =true;
bool	bDCO = false; 	
bool	bForce = false;
bool 	bAssignTarget =false;

//ini
string configFile ="dco.ini";
string[string] configKeyValue;

//ini args
string strPackageName,strArgs,strTargetName,strTargetType ="exe",strDC,strDCStandardEnvBin ="dmd2\\windows\\bin",strLibs ,strImport,strLflags;
 
void main(string[] args)
{
	
	if(!findDCEnv()) return;
	// readInJson();
	if(!checkArgs(args))
	{
		if(!findFiles())
		{
			ShowUsage();
			return;
		}
	}
	if(args.length ==1)
	{
		if(!CheckBinFolderAndCopy()) return;
    }
	if(args.length == 2 && (toLower(args[1]) == "-h" || toLower(args[1]) == "-help"))
	{
		ShowUsage();
		return;
	}
	if(strPackageName =="")
	{
		strPackageName = strTargetName;
	}
 
	buildExe(args);
}

bool findDCEnv()
{
	 if(!readConfig(configFile)) return false;
	auto path = environment["PATH"];
	 
	ptrdiff_t i = path.indexOf(strDCStandardEnvBin);
	int len = strDCStandardEnvBin.length;
	string strNoDC = "Not found "~strDC~" in your computer,please setup it.";
	if(i != -1)
	{
		path = path[0 .. i+len].idup;

		ptrdiff_t j = path.lastIndexOf(";");
		strDCEnv = path[j+1 .. i+len];
	 
		if(exists(strDCEnv~"\\"~strDC.stripRight()~".exe"))
		{
			strDCEnvFile = strDCEnv ~ "\\dco.exe";
			return true;
		}
		else
		{
			writeln("\nMaybe FirWall stop the 'dco',that can't  know the 'dco.exe' exists.");
			return false;
		}
	}
 
	writeln(strNoDC);
	return false;
}

bool readConfig(string configFile)
{ 
	try
	{ 
		string strConfigPath = thisExePath();
		strConfigPath = strConfigPath[0..strConfigPath.lastIndexOf("\\")].idup;
		strConfigPath ~= "\\" ~ configFile;
 
		if(!exists(strConfigPath)) 
		{
			writeln("if you known dco.ini is there,maybe your 'FireWall' stop to access the 'dco.ini',please stop it.Otherwise,dco not found dco.ini, it will help you to  create a init dco.ini file ,but you should input something in it.");
			initNewConfigFile();
			return false;
		}  
	    
		auto file = File(strConfigPath); 
		scope(failure) file.close();
		auto range = file.byLine();
		foreach (line; range)
		{
			if (!line.init && line[0] != '#' && line[0] != ';' && line.indexOf("=") != -1)
			{ 
				ptrdiff_t i =line.indexOf("=");
				configKeyValue[line.strip()[0..i].idup] = line.strip()[i+1..$].idup;
			}
		}
	 
		 file.close();
  
		strDC = configKeyValue.get("DC","dmd"); 
		 
		strDCStandardEnvBin = configKeyValue.get("DCStandardEnvBin","dmd2\\windows\\bin"); 
		SpecialLib = configKeyValue.get("SpecialLib","dfl");  
		strImport = configKeyValue.get("importPath","");
		strLflags = configKeyValue.get("lflags","/su:console:4"); 
		return true;
  }
  catch(Exception e) 
  {
		writeln(" Read ini file err,you should input something in ini file.",e.msg);
		return false;
  }
}


bool checkArgs(string[] args)
{
	string c;
	int p;
	bool bfindD =false;
	foreach(int i,arg;args)
	{
		c = toLower(arg);
		p = c.indexOf('-');
		if(p != -1)
		{
			c = c[p+1 .. $];
		}
		if(i ==0) continue;
		if(c.indexOf(".d") != -1)
		{
			bfindD = true;
		 
		}
		else if(c == "force")
		{
			bForce = true;
		}
		else if(c.indexOf("of") != -1)
		{
			bAssignTarget = true;
			strTargetName = c[(c.indexOf("of")+1)..$];
		}
		else if(c == strPackageName || c == strPackageName~"lib")
		{
			bAssignTarget = true;
			bBuildSpecialLib = true;
			strTargetTypeSwitch = " -" ~ targetTypeDefault;
			strTargetName = c~".lib";
		}
		else if (c == "ini")
		{
			initNewConfigFile();
			return false;
		}
	}
	return bfindD;
}

bool CheckBinFolderAndCopy() 
{
	if(checkIsUpToDate())
	{
		writeln(strTargetName ~" file is up to date.");
		return false;
	}
	return true;
}

bool checkIsUpToDate()
{
	 getTargetInfo();
     if(exists(strTargetFileName))
     {
		targetTime = getTargetTime(strTargetFileName);
 
        if(strTargetFileName.indexOf("dco.exe") != -1)
        {
			if(!checkIsUpToDate(strDCEnvFile ,targetTime))
			{
				string strcopy ="copy dco.exe " ~ strDCEnvFile;
				writeln(strcopy);
				auto status = system(strcopy);
				if(status !=0)
				{
					writeln("copy failed.");
				}
			 //copy(strTargetFileName,strDCEnvFile);
			}
 	    }
 		 
		bool bUpToDate = (targetTime >= sourceLastUpdateTime);
		 
		if(!bUpToDate || bForce)
		{
			removeExe(strTargetFileName);
		}
 		return bUpToDate;
    }
 
    return false;
}
SysTime getTargetTime(string strPathFile)
{
	 return DirEntry(strPathFile).timeLastModified;
}
void removeExe(string strPathExe)
{
    if(exists(strPathExe))
	{
		if(system("del " ~ strPathExe) == 0) 
        {
			writeln(strPathExe ~ ", remove  ok!");
			return;
		}
	}
}

bool checkIsUpToDate(string strPathFile,SysTime targettime)
{
	 if(!exists(strPathFile)) return false;
    auto testFile = DirEntry(strPathFile);
    auto createTime = testFile.timeLastModified;
   
    return (targettime <= createTime);
}

void buildExe(string[] args)
{
	string c;
	int p;
	foreach(int i,arg;args)
	{
		c = toLower(arg);
		p = c.indexOf('-');
		if(p != -1)
		{
			c = c[p+1 .. c.length];
		}
		switch(c)
    	{
			case "h","help":
				ShowUsage();
				break;
			case "gui":
				strTargetLflags = strWindows;
				bUseSpecialLib = true;
    			strAddArgs = strAddArgsdfl;
				break;
			case "win","windows","winexe":
				strTargetLflags = strWindows;
				break;
    		case "debug":
				bDebug = true;
				break;
			case "release":
				bDebug = false;
				strDebug = " -" ~ c.idup;
				break;

    		case "console","con","exe":
    			strTargetLflags = strConsole;
    			break;
			case "all":
				bUseSpecialLib = false;
				strAddLib = strLibs;
    			strAddArgs = strAddArgsdfl;
    			strImport = strImportDefault;
    			strTargetLflags = strConsole;
    			break;
			case "addlib":
    			strAddLib = strLibs~" ";
    			strImport = strImportDefault;
    			strTargetLflags = strConsole;
    			break;
			case "arg":
    			strAddArgs = strAddArgsdfl;
    			break;
			case "lib":
				strTargetTypeSwitch = " -" ~ targetTypeDefault;
				break;
			case "dfl","dfllib":
				 bBuildSpecialLib = true;
				strTargetTypeSwitch = " -" ~ targetTypeDefault;
				break;
			case "c","copy":
				bCopy = true;
				break;
			case "force":
				bForce = true;
				break;
			case "init":
				
				break;
    		default:
    		if(i ==0) continue;
				if(c.indexOf(".d") != -1)
				{
					strDFile ~= " ";
					strDFile ~= arg;
				}
				else
				{
					strOtherArgs ~= " ";
					strOtherArgs ~= arg;
				}
    		break;
    	}
	}

   strTargetLib = bDebug ? SpecialLib ~ "_debug.lib" : SpecialLib ~ ".lib";
 
   if(bBuildSpecialLib)
   {
	   strOtherArgs = " -of" ~ strTargetLib;
	   strAddLib = strLibs;
   }
 
	if(bUseSpecialLib)
	{
			if(SpecialLib == "dfl")
	{
		strLibs =strWinLibs;
	}
		strAddLib = strTargetLib ~" " ~ strLibs;
	}
  
	buildExe();
}

void buildExe()
{
	if(bForce)
	{
		removeExe(strTargetFileName);
	}
	strDC ~= " ";
	strDC ~= strTargetTypeSwitch;
	string strCommon = strAddLib ~ strTargetLflags ~ strDFile ~ strDebug;
 
    string buildstr = strDC ~ strAddArgsdfl ~ strOtherArgs ~ " " ~ strImportDefault ~ strCommon ~ "\r\n";
	buildstr = bUseSpecialLib ? buildstr : strDC ~ strAddArgs ~ strOtherArgs ~ " " ~ strImport ~ strCommon;
	if(bDisplayBuildStr)
	{
		writeln(buildstr);
	}
 
	StopWatch sw;
	sw.start();
	auto status = system(buildstr);
	sw.stop();
   
   if (status != 0)
	{
		writeln("Compilation failed:\n", status);
	}
	else
	{
		writeln("\nCompile time :" , sw.peek().msecs/1000.0,"secs");

		if(bCopy)
		{
			copyFile();
		}
	}
	writeln("End.");
}

void copyFile()
{
	string strcopy;
	if(!exists(strTargetFileName)) return;
	if( sourceLastUpdateTime >= getTargetTime(strTargetFileName))
	{
		writeln(strTargetFileName," is up to date.");
      return;
    }
	if(strTargetFileName.indexOf("exe") != -1)
	{
		//copy(strTargetFileName,strDCEnv); //
		strcopy = "copy " ~ strTargetFileName~" " ~ strDCEnv;
	}
	else
	{
		string strDCLibPath = strDCEnv[0..(strDCEnv.length - "bin".length)].idup ~ "lib"; 
		//copy(strDCEnv,strDCLibPath);
		strcopy = "copy " ~ strTargetFileName ~ " " ~ strDCLibPath;
	}
	if(bDisplayCopyInfo)
	{
		writeln(strcopy);
	}
		 
	auto status = system(strcopy);
	if(status !=0)
	{
		writeln("Copy failed.");
	}
}
  
bool findFiles()
{ 
	bool bPackage = false; 
	auto packages = dirEntries(".","{package.d,all.d}",SpanMode.depth);
	if(packages.count >0) bPackage = true;
	auto dFiles = dirEntries(".","*.{d,di}",SpanMode.depth);
	int icount =0;
    SysTime fileTime;
    DirEntry rootDE ;
  
	foreach(d; dFiles)
	{
	    if(!bAssignTarget)
	    {
			if(icount == 0)
			{
				strTargetName = d.name[(d.name.lastIndexOf("\\")+1) .. d.name.lastIndexOf(".")];
				strTargetName ~= "." ~ strTargetType; 
			}
		}
		if(icount ==0 )
		{
			ReadDFile(d,bPackage);
		}
		
		strDFile ~= " ";
		strDFile ~= d.name[2 ..$].idup;
		
		//sourceLastUpdateTime 
		rootDE = DirEntry(d);
        if(rootDE.timeLastModified > fileTime)
        {
        	fileTime = rootDE.timeLastModified;
        } 
        icount++;
	}
    sourceLastUpdateTime = fileTime;
    
	strDFile = strDFile.stripRight().idup;
	
	if(icount <= 0)  
	{
		writeln("Not found any *.d files in current folder.If there is a 'source' or 'src' folder,dco will find the '*.d' from there.");
		 return false;
	}
	bCopy = (strDFile.indexOf("dco.d") != -1) ? true : false;
	return true;
}

void getTargetInfo()
{
	  string root_path = getcwd();
    string strPath;
	auto dFiles = dirEntries(root_path,"*.{lib,exe}",SpanMode.shallow);
	int i =0;
	foreach(d;dFiles)
	{
		i++;
		strTargetFileName =d;
	   strTargetType = d.name[d.name.lastIndexOf(".")+1..$];
	   break;

	}
	if(i ==0)
	{
		if(strTargetName.indexOf("." ~ strTargetType) == -1)
		{
			strTargetName = strTargetName ~ "." ~ strTargetType;
		}
		strTargetFileName = root_path ~ "\\" ~ strTargetName;
	}
	return;
}
 
void ShowUsage()
{
	writeln("
dco written by FrankLIKE.
Usage:
	dco [<switches...>] <files...>
		  
	for example: dco ↓ 
	         or: dco app.d 
		 
build for dfl2:	dco ↓ 
			or: dco -gui
			or:	dco *.d -gui
build for other: dco -lib
		     or: dco *.d -lib
			 or: dco *.d -release
	         or: dco *.d -arg -addlib

Switches:
    -h	       Print help(usage: -h,or -help).
    -c	       Copy new exe or lib to 'windows/bin' or 'lib' Folder
		(-copy also is ok). 
    -release   Build files's Release version(Default version is 'debug').
    -gui       Make a Windows GUI exe without a console(For DFL).
    -win       Make a Windows GUI exe without a console
		(For any other: the same to -winexe,-windows).
    -lib       Build lib files.
    -ini	   Create the ini file for config. 
    -all       Build files by args,libs(Default no dfl_debug.lib) in Console.
    -arg       Build files by args(-de -w -property -X).
    -addlib    Build files by add libs(ole32.lib oleAut32.lib gdi32.lib 
		Comctl32.lib Comdlg32.lib advapi32.lib uuid.lib ws2_32.lib).
    ");
} 

void ReadDFile(string dFile,bool bPackage)
{ 
	 auto file = File(dFile); 
	 scope(failure)  file.close();
	 auto range = file.byLine();
	 int icount = 0;
    foreach (line; range)
    {
        if (!line.init && line.indexOf("import") != -1)
        { 
          bUseSpecialLib = !bPackage;
          bBuildSpecialLib = bPackage;
          if(bUseSpecialLib) strTargetLflags = strWindows;
          
        	if(line.indexOf("dfl") != -1)
        	{
        		SpecialLib = "dfl";
				break;
			}
			else if(line.indexOf("dgui") != -1)
			{
				strArgs = strAddArgsdfl = " -g -de -w -property -X ";
				SpecialLib = "dgui";
				break;
			}
        }
        icount++;
        if(icount >100) break;
    }
    file.close(); 
}

void initNewConfigFile()
{
	auto ini = File("dco.ini","w"); 
	ini.writeln(";DC=dmd");
	ini.writeln("DC=");
	ini.writeln(";DCStandardEnvBin=dmd2\\windows\\bin");
	ini.writeln("DCStandardEnvBin=");
	ini.writeln(";SpecialLib=dfl");
	ini.writeln("SpecialLib=");
	ini.writeln(";importPath=-I$(DMDInstallDir)windows/import");
	ini.writeln("importPath=");
	ini.writeln(";lflags=-L/su:console:4");
	ini.writeln(";lflags=-L/SUBSYSTEM:WINDOWS");
	ini.writeln("lflags=");
	ini.close();
 /*
	auto pid = spawnProcess("notepad.exe dco.ini");
    auto dmd = tryWait(pid);
	if (dmd.terminated)
	{
		if (dmd.status == 0) writeln("open dco.ini succeeded!");
		else writeln("open dco.ini failed");
	}
	else writeln("Still opening...");
	*/
}

void readInJson()
{
	if(!exists("dub.json") & !exists("package.json") ) return;
	/*
	strPackageName = configKeyValue.get("name","");
	strArgs = configKeyValue.get("args",strAddArgs);
	strLibs = configKeyValue.get("libs","");
	strTargetName = configKeyValue.get("targetName","");
	strTargetType = configKeyValue.get("targetType",strTargetType); 
	*/
}