PA2-zkouska/pkg.cpp

#include <iostream>
#include <cassert>
#include <vector>
#include <set>
#include <map>
#include <string>
#include <unordered_map>
#include <memory>
#include <unordered_set>
#include <queue>
#include <list>
#include <sstream>
#include <cstring>
 
 #include <stack>

using namespace std;

class CPkg{
public:
    CPkg( std::string n )
    : _name(std::move(n)) {}
    CPkg& addDep( std::string n ) {
        _deps.emplace_back( std::move(n) );
        return *this;
    }
    bool operator<( const CPkg& oth ) const {
        return _name < oth._name;
    }

    std::string _name;
    std::vector< std::string > _deps;
private:
 
};

class CPkgSys{
public:
    CPkgSys& addPkg(const CPkg& p) {
        packages[p._name] = p._deps;
        return *this;
    }

    std::set<std::string> install(std::list<std::string> what) {
        std::set<std::string> d = installed;
        std::queue<std::string> q;
        std::set<std::string> r;

        for (const auto& i : what) {
            q.push(i);
            if (d.emplace(i).second)
                r.emplace(i);
        }

        while (q.size()) {
            auto i = q.front();
            q.pop();
            auto it = packages.find(i);

            if (it == packages.end())
                throw std::invalid_argument("Package not found.");
            
            for ( const auto& j : it->second ) {
                if (d.emplace(j).second) {
                    r.emplace(j);
                    q.push(j);
                }
            }
        }
        std::swap( installed, d );
        return r;
    }

private:
    std::unordered_map<std::string, std::vector<std::string>> packages;
    std::set<std::string> installed;
};

int main(void){
    CPkgSys s;
    stringstream ss;
    s.addPkg(CPkg("ssh").addDep("sudo").addDep("apt"))
    .addPkg(CPkg("sudo").addDep("git").addDep("c++"));
    s.addPkg(CPkg("apt"))
    .addPkg(CPkg("c++").addDep("c").addDep("asm").addDep("fortran"));
    s.addPkg(CPkg("git"))
    .addPkg(CPkg("c").addDep("kekw"))
    .addPkg(CPkg("kekw"))
    .addPkg(CPkg("asm"))
    .addPkg(CPkg("fortran"));
    s.addPkg(CPkg("python").addDep("bash").addDep("sadge"))
    .addPkg(CPkg("karel").addDep("python"))
    .addPkg(CPkg("bash").addDep("sadge"))
    .addPkg(CPkg("sadge"))
    .addPkg(CPkg("cython").addDep("dev"));
    s.addPkg(CPkg("perl"));
 
    // ss << s;
    // assert(ss.str() == "");
    ss.clear();
    ss.str("");
 
    set<string> t1 = s.install(list<string> {"sudo"});
    assert(t1 == (set<string> {"asm", "c", "c++", "fortran", "git", "kekw", "sudo"}));
    set<string> t2 = s.install(list<string> {"ssh", "c++"});
    assert(t2 == (set<string> {"apt", "ssh"}));
 
    // ss << s;
    // assert(ss.str() == "apt, asm, c, c++, fortran, git, kekw, ssh, sudo");
    // ss.clear();
    // ss.str("");
 
    try{
        set<string> e = s.install(list<string> {"karel", "cython"});
        assert("Sem ses nemel dostat debilku" == nullptr);
    }
    catch(const invalid_argument & e){
        assert(strcmp("Package not found.", e.what()) == 0);
    }
    set<string> t3 = s.install(list<string> {"karel", "fortran", "git"});
    assert(t3 == (set<string> {"bash", "karel", "python", "sadge"}));
 
    s.addPkg(CPkg("java").addDep("utils"))
    .addPkg(CPkg("utils").addDep("VB"))
    .addPkg(CPkg("VB").addDep("java"));
 
    set<string> t4 = s.install(list<string> {"java", "perl"});
    assert(t4 == (set<string> {"VB", "java", "perl", "utils"}));
 
    // ss << s;
    // assert(ss.str() == "VB, apt, asm, bash, c, c++, fortran, git, java, karel, kekw, perl, python, sadge, ssh, sudo, utils");
    // ss.clear();
    // ss.str("");
 
    CPkgSys k;
    k.addPkg(CPkg("ssh").addDep("sudo").addDep("apt"))
    .addPkg(CPkg("sudo").addDep("git"));
    k.addPkg(CPkg("apt"));
    k.addPkg(CPkg("git"))
    .addPkg(CPkg("c").addDep("kekw"))
    .addPkg(CPkg("kekw"))
    .addPkg(CPkg("asm"))
    .addPkg(CPkg("fortran"));
    k.addPkg(CPkg("python").addDep("bash").addDep("sadge"))
    .addPkg(CPkg("karel").addDep("python"))
    .addPkg(CPkg("bash").addDep("sadge"))
    .addPkg(CPkg("sadge"));
    k.addPkg(CPkg("perl").addDep("no"));
 
    set<string> t5 = k.install(list<string> {"asm"});
    assert(t5 == (set<string> {"asm"}));
    set<string> t6 = k.install(list<string> {"python", "ssh"});
    assert(t6 == (set<string> {"apt", "bash", "git", "python", "sadge", "ssh", "sudo"}));
 
    try{
        set<string> t7 = k.install(list<string> {"perl", "c"});
        assert("Sem ses nemel dostat debilku" == nullptr);
    }
    catch(const invalid_argument & e){
        assert(strcmp("Package not found.", e.what()) == 0);
    }
    set<string> t8 = k.install(list<string> {"c", "ssh", "karel"});
 
    assert(t8 == (set<string> {"c", "karel", "kekw"}));
 
    // ss << k;
    // assert(ss.str() == "apt, asm, bash, c, git, karel, kekw, python, sadge, ssh, sudo");
    // ss.clear();
    // ss.str("");
    std::cout << "success!\n";
}