Rename macros to profiles
[mspang/inapt.git] / inapt.cc
index 12c75f4..b3a6ee6 100644 (file)
--- a/inapt.cc
+++ b/inapt.cc
@@ -1,8 +1,10 @@
 #include <stdio.h>
 #include <stdlib.h>
+#include <getopt.h>
 #include <iostream>
 #include <cstdio>
 #include <fstream>
+#include <set>
 #include <apt-pkg/pkgcache.h>
 #include <apt-pkg/cachefile.h>
 #include <apt-pkg/dpkgdb.h>
 
 using namespace std;
 
-bool InstallPackages(pkgCacheFile &Cache,bool ShwKept = false,bool Ask = true,
+char *prog = NULL;
+
+static struct option opts[] = {
+    { NULL, 0, NULL, '\0' },
+};
+
+bool run_install(pkgCacheFile &Cache,bool ShwKept = false,bool Ask = true,
                      bool Safety = true)
 {
-   if (_config->FindB("APT::Get::Purge",false) == true)
+   if (_config->FindB("APT::Get::Purge", false) == true)
    {
       pkgCache::PkgIterator I = Cache->PkgBegin();
       for (; I.end() == false; I++)
       {
          if (I.Purge() == false && Cache[I].Mode == pkgDepCache::ModeDelete)
-            Cache->MarkDelete(I,true);
+            Cache->MarkDelete(I, true);
       }
    }
 
    if (Cache->BrokenCount() != 0)
    {
-      return _error->Error("Internal error, InstallPackages was called with broken packages!");
+      return _error->Error("Internal error, run_install was called with broken packages!");
    }
 
    if (Cache->DelCount() == 0 && Cache->InstCount() == 0 &&
@@ -170,9 +178,106 @@ bool InstallPackages(pkgCacheFile &Cache,bool ShwKept = false,bool Ask = true,
    }
 }
 
-int main(int argc, char *argv[]) {
+static void usage() {
+    fprintf(stderr, "Usage: %s [filename]\n", prog);
+    exit(2);
+}
+
+static bool test_macro(const char *macro, set<string> *defines) {
+    return (*macro != '!' && defines->find(macro) != defines->end())
+            || (*macro == '!' && defines->find(macro + 1) == defines->end());
+}
+
+static pkgCache::PkgIterator eval_pkg(inapt_package *package, pkgCacheFile &cachef) {
+    pkgCache::PkgIterator pkg;
+
+    pkgCache *cache = cachef;
+
+    if (!pkg.end()) fatal("omg"); /* TODO */
+
+    for (std::vector<std::string>::iterator i = package->alternates.begin(); i != package->alternates.end(); i++) {
+        pkg = cache->FindPkg(*i);
+
+        /* no such package */
+        if (pkg.end())
+            continue;
+
+        /* real package */
+        if (cachef[pkg].CandidateVer)
+            break;
+
+        /* virtual package */
+        if (pkg->ProvidesList) {
+            if (!pkg.ProvidesList()->NextProvides) {
+                pkgCache::PkgIterator tmp = pkg.ProvidesList().OwnerPkg();
+                if (package->action == inapt_action::INSTALL) {
+                    debug("selecting %s instead of %s", tmp.Name(), pkg.Name());
+                    pkg = tmp;
+                    break;
+                } else {
+                    debug("will not remove %s instead of virtual package %s", tmp.Name(), pkg.Name());
+                }
+            } else {
+                debug("%s is a virtual package", pkg.Name());
+            }
+        } else {
+            debug("%s is a virtual packages with no provides", pkg.Name());
+        }
+    }
+
+    if (pkg.end()) {
+        /* todo: report all errors at the end */
+        if (package->alternates.size() == 1) {
+            fatal("%s:%d: No such package: %s", package->filename, package->linenum, package->alternates[0].c_str());
+        } else {
+            std::vector<std::string>::iterator i = package->alternates.begin();
+            std::string message = *(i++);
+            while (i != package->alternates.end()) {
+                message.append(", ");
+                message.append(*(i++));
+            }
+            fatal("%s:%d: No alternative available: %s", package->filename, package->linenum, message.c_str());
+        }
+    }
+
+    return pkg;
+}
+
+static bool test_macros(vector<std::string> *macros, set<string> *defines) {
+    bool ok = true;
+    for (vector<std::string>::iterator j = macros->begin(); j < macros->end(); j++) {
+        if (!test_macro((*j).c_str(), defines)) {
+            ok = false;
+            break;
+        }
+    }
+    return ok;
+}
 
-    vector<inapt_action> actions;
+static void eval_action(inapt_action *action, set<string> *defines, vector<inapt_package *> *final_actions) {
+    for (vector<inapt_package *>::iterator i = action->packages.begin(); i < action->packages.end(); i++) {
+        if (test_macros(&(*i)->predicates, defines))
+            final_actions->push_back(*i);
+    }
+}
+
+static void eval_block(inapt_block *block, set<string> *defines, vector<inapt_package *> *final_actions) {
+    if (!block)
+        return;
+
+    for (vector<inapt_action *>::iterator i = block->actions.begin(); i < block->actions.end(); i++)
+        if (test_macros(&(*i)->predicates, defines))
+            eval_action(*i, defines, final_actions);
+
+    for (vector<inapt_conditional *>::iterator i = block->children.begin(); i < block->children.end(); i++) {
+        if (test_macro((*i)->condition, defines))
+            eval_block((*i)->then_block, defines, final_actions);
+        else
+            eval_block((*i)->else_block, defines, final_actions);
+    }
+}
+
+static void exec_actions(std::vector<inapt_package *> *final_actions) {
 
     pkgInitConfig(*_config);
     pkgInitSystem(*_config, _system);
@@ -190,12 +295,17 @@ int main(int argc, char *argv[]) {
     pkgCache *cache = cachef;
     pkgDepCache *DCache = cachef;
 
-    scanner(&actions);
+    for (vector<inapt_package *>::iterator i = final_actions->begin(); i != final_actions->end(); i++)
+        (*i)->pkg = eval_pkg(*i, cachef);
 
-    for (vector<inapt_action>::iterator i = actions.begin(); i < actions.end(); i++) {
-        switch(i->action) {
+    for (vector<inapt_package *>::iterator i = final_actions->begin(); i < final_actions->end(); i++) {
+        pkgCache::PkgIterator k = (*i)->pkg;
+        switch ((*i)->action) {
             case inapt_action::INSTALL:
-                DCache->MarkInstall(cache->FindPkg(i->package), true);
+                if (!k.CurrentVer() || cachef[k].Delete()) {
+                    printf("preinstall %s %s:%d\n", (*i)->pkg.Name(), (*i)->filename, (*i)->linenum);
+                    DCache->MarkInstall(k, true);
+                }
                 break;
             case inapt_action::REMOVE:
                 break;
@@ -204,15 +314,20 @@ int main(int argc, char *argv[]) {
         }
     }
 
-    for (vector<inapt_action>::iterator i = actions.begin(); i < actions.end(); i++) {
-        switch(i->action) {
+    for (vector<inapt_package *>::iterator i = final_actions->begin(); i < final_actions->end(); i++) {
+        pkgCache::PkgIterator k = (*i)->pkg;
+        switch ((*i)->action) {
             case inapt_action::INSTALL:
-                printf("install %s %s:%d\n", i->package, i->filename, i->linenum);
-                DCache->MarkInstall(cache->FindPkg(i->package), false);
+                if ((!k.CurrentVer() && !cachef[k].Install()) || cachef[k].Delete()) {
+                    printf("install %s %s:%d\n", (*i)->pkg.Name(), (*i)->filename, (*i)->linenum);
+                    DCache->MarkInstall(k, false);
+                }
                 break;
             case inapt_action::REMOVE:
-                printf("remove %s %s:%d\n", i->package, i->filename, i->linenum);
-                DCache->MarkDelete(cache->FindPkg(i->package), false);
+                if ((k.CurrentVer() && !cachef[k].Delete()) || cachef[k].Install()) {
+                    printf("remove %s %s:%d\n", (*i)->pkg.Name(), (*i)->filename, (*i)->linenum);
+                    DCache->MarkDelete(k, false);
+                }
                 break;
             default:
                 fatal("uninitialized action");
@@ -246,10 +361,8 @@ int main(int argc, char *argv[]) {
 
     pkgProblemResolver fix (DCache);
 
-    for (vector<inapt_action>::iterator i = actions.begin(); i < actions.end(); i++)
-           fix.Protect(cache->FindPkg(i->package));
-    for (vector<inapt_action>::iterator i = actions.begin(); i < actions.end(); i++)
-           fix.Protect(cache->FindPkg(i->package));
+    for (vector<inapt_package *>::iterator i = final_actions->begin(); i < final_actions->end(); i++)
+        fix.Protect((*i)->pkg);
     fix.Resolve();
 
     fprintf(stderr, "\n");
@@ -269,5 +382,43 @@ int main(int argc, char *argv[]) {
 
     fprintf(stderr, "\n");
 
-    InstallPackages(cachef);
+    run_install(cachef);
+}
+
+int main(int argc, char *argv[]) {
+    int opt;
+    char *filename = NULL;
+
+    set<string> defines;
+
+    prog = xstrdup(basename(argv[0]));
+    while ((opt = getopt_long(argc, argv, "p:", opts, NULL)) != -1) {
+        switch (opt) {
+            case 'p':
+                defines.insert(optarg);
+                break;
+            case '?':
+                usage();
+                break;
+            default:
+                fatal("error parsing arguments");
+        }
+    }
+
+    if (argc - optind == 1)
+        filename = argv[optind++];
+    else if (argc - optind > 0)
+        usage();
+
+    fprintf(stderr, "defines: ");
+    for (set<string>::iterator i = defines.begin(); i != defines.end(); i++)
+        fprintf(stderr, "%s  ", i->c_str());
+    fprintf(stderr, "\n");
+
+    inapt_block context;
+    vector<inapt_package *> final_actions;
+
+    parser(filename, &context);
+    eval_block(&context, &defines, &final_actions);
+    exec_actions(&final_actions);
 }