#include "Sources.hpp"
#include "Commands.hpp"
#include "Cache.hpp"

#include <iostream>
#include <fstream>

#include <glibmm.h>
#include <giomm.h>

using namespace net::lliurex::apt;
using namespace std;
using namespace std::placeholders;

class Application
{
	

	public:

	/*! mandatory sources */
	vector<Sources> conf;
	
	/*! optional sources */
	vector<Sources> optional;
	
		
	
	Application(int argc,char * argv[])
	{
	
		//this should be call before calling any other command
		command::Init();
	
		if(argc<2)
			PrintUsage();
		else
		{
			string arg(argv[1]);
			bool found = false;
			
			if(arg=="list")
			{
				conf = command::LoadConf();
				command::List(conf);
				found=true;	
			}
			
			if(arg=="show")
			{
				command::Show();
				found=true;
			}
			
			if(arg=="check")
			{
				conf = command::LoadConf();
				Sources src(Sources::SourcesList);
				
				vector<command::Match> status = command::Check(src,conf);
				
				for(command::Match m : status)
				{
					string txt="";
					
					switch(m.status)
					{
						case FindLineStatus::Found:
							txt="complete";	
						break;
						
						case FindLineStatus::Partial:
							txt="partial";
						break;
					}
					
					if(txt!="")
					{
						cout<<txt<<" ["<<m.sources->name["C"]<<"]"<<endl;
					}

				}
				
				found=true;
			}
			
			if(arg=="set")
			{
				conf = command::LoadConf();
				
				vector<string> targets;
				
				for(int n=2;n<argc;n++)
				{
					targets.push_back(argv[n]);
				}
				
				Sources src(Sources::SourcesList);
				src=command::Set(src,conf,targets);
				src.Print();
				Write(src,Sources::SourcesList);
				
				found=true;
			}
			
			if(arg=="reset")
			{
				conf = command::LoadConf();
				
				vector<string> targets;
				
				for(int n=2;n<argc;n++)
				{
					targets.push_back(argv[n]);
				}
				
				Sources src(Sources::SourcesList);
				src=command::Set(src,conf,targets,true);
				src.Print();
				Write(src,Sources::SourcesList);
				found=true;
			}
			
			if(arg=="add")
			{
				if(argc<=3)
				{
					Sources src(Sources::SourcesList);
					src=command::Add(src,argv[2]);
					src.Print();
					src.Write(Sources::SourcesList);
					found = true;
				}
			}
			
			
			
			if(!found)
			{
				PrintUsage();
			}
		}
	}
	
	/*!
		There is nothing to free
	*/
	~Application()
	{
	
	}
		
	
	/*!
		Just prints out usage to stdout
	*/
	void PrintUsage()
	{
		cout<<"Usage:"<<endl;
		cout<<"llx-apt-cmd [COMMAND]"<<endl;
		
		cout<<"Available commands:"<<endl;
		
		cout<<"show\tSummarize sources.list"<<endl;
		cout<<"list\tShows installed profiles"<<endl;
		cout<<"check\tCompares sources.list with installed profiles"<<endl;
		cout<<"set [PROFILE1] [PROFILE2] [...]\tSets given profiles to sources.list"<<endl;
		cout<<"reset [PROFILE1] [PROFILE2] [...]\tSame as set command, but it wipes out current sources.list content, instead of merging it"<<endl;
		cout<<"add [LINE]\tAppends a new line to sources.list"<<endl;
		
	}
	
	void Write(Sources & src,const string & filename)
	{
		fstream fs (filename,std::fstream::out);
		
		if(!fs.good())
			throw runtime_error("Failed to write sources.list file");
			
		for(int n=0;n<src.lines.size();n++)
		{
			fs<<src.lines[n].ToString()<<endl;
		}
		
		fs.close();
	}
	
	
};

int main(int argc,char * argv[])
{
	Application app(argc,argv);

	return 0;
}