source: lliurex-apt/trunk/fuentes/src/cli.cpp @ 923

#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;
}