Remove cruft from Makefile
[mspang/inapt.git] / inapt.cc
index 984ca8a..6e235de 100644 (file)
--- a/inapt.cc
+++ b/inapt.cc
 
 #include "inapt.h"
 #include "util.h"
-#include "acqprogress.h"
+#include "contrib/acqprogress.h"
 
 char *prog = NULL;
 
 static struct option opts[] = {
-    { "auto-remove", 0, NULL, 'z' },
     { "simulate", 0, NULL, 's' },
     { "purge", 0, NULL, 'u' },
     { NULL, 0, NULL, '\0' },
 };
 
-bool run_install(pkgCacheFile &cache) {
+static bool run_install(pkgCacheFile &cache) {
    if (_config->FindB("Inapt::Purge", false))
       for (pkgCache::PkgIterator i = cache->PkgBegin(); !i.end(); i++)
          if (!i.Purge() && cache[i].Mode == pkgDepCache::ModeDelete)
@@ -48,7 +47,7 @@ bool run_install(pkgCacheFile &cache) {
    FileFd Lock;
    Lock.Fd(GetLock(_config->FindDir("Dir::Cache::Archives") + "lock"));
    if (_error->PendingError())
-       return _error->Error(("Unable to lock the download directory"));
+       return _error->Error("Unable to lock the download directory");
 
    unsigned int width = 80;
    AcqTextStatus status (width, 0);
@@ -56,9 +55,9 @@ bool run_install(pkgCacheFile &cache) {
 
    pkgSourceList List;
    if (List.ReadMainList() == false)
-      return _error->Error(("The list of sources could not be read."));
+      return _error->Error("The list of sources could not be read");
 
-   SPtr<pkgPackageManager> PM= _system->CreatePM(cache);
+   SPtr<pkgPackageManager> PM = _system->CreatePM(cache);
    if (PM->GetArchives(&Fetcher, &List, &Recs) == false ||
        _error->PendingError())
       return false;
@@ -67,13 +66,9 @@ bool run_install(pkgCacheFile &cache) {
      return false;
 
   bool Failed = false;
-  for (pkgAcquire::ItemIterator I = Fetcher.ItemsBegin(); I != Fetcher.ItemsEnd(); I++)
-  {
-     if ((*I)->Status != pkgAcquire::Item::StatDone || (*I)->Complete != true) {
-         fprintf(stderr,("Failed to fetch %s  %s\n"),(*I)->DescURI().c_str(),
-                 (*I)->ErrorText.c_str());
+  for (pkgAcquire::ItemIterator i = Fetcher.ItemsBegin(); i != Fetcher.ItemsEnd(); i++) {
+     if ((*i)->Status != pkgAcquire::Item::StatDone || (*i)->Complete != true)
          Failed = true;
-     }
   }
 
   if (Failed)
@@ -88,12 +83,13 @@ bool run_install(pkgCacheFile &cache) {
   return false;
 }
 
-void run_autoremove(pkgCacheFile &cache)
-{
+static void run_autoremove(pkgCacheFile &cache) {
+    bool purge = _config->FindB("Inapt::Purge", false);
+
     for (pkgCache::PkgIterator i = cache->PkgBegin(); !i.end(); i++) {
         if (cache[i].Garbage) {
             debug("autoremove: %s", i.Name());
-            cache->MarkDelete(i, 0);
+            cache->MarkDelete(i, purge);
         }
     }
 
@@ -102,13 +98,33 @@ void run_autoremove(pkgCacheFile &cache)
 }
 
 static void usage() {
-    fprintf(stderr, "Usage: %s [filename..]\n", prog);
+    fprintf(stderr, "Usage: %s [options] [filename..]\n", prog);
     exit(2);
 }
 
-static bool test_macro(const char *macro, std::set<std::string> *defines) {
-    return (*macro != '!' && defines->find(macro) != defines->end())
-            || (*macro == '!' && defines->find(macro + 1) == defines->end());
+static bool test_profile(const char *profile, std::set<std::string> *defines) {
+    return (*profile != '!' && defines->find(profile) != defines->end())
+            || (*profile == '!' && defines->find(profile + 1) == defines->end());
+}
+
+static bool test_anyprofile(std::string &profile, std::set<std::string> *defines) {
+    char *s = xstrdup(profile.c_str());
+    const char *c = strtok(s, "/");
+
+    if (test_profile(c, defines)) {
+        free(s);
+        return true;
+    }
+
+    while ((c = strtok(NULL, "/")) != NULL) {
+        if (test_profile(c, defines)) {
+            free(s);
+            return true;
+        }
+    }
+
+    free(s);
+    return false;
 }
 
 static pkgCache::PkgIterator eval_pkg(inapt_package *package, pkgCacheFile &cache) {
@@ -153,8 +169,7 @@ static pkgCache::PkgIterator eval_pkg(inapt_package *package, pkgCacheFile &cach
             std::vector<std::string>::iterator i = package->alternates.begin();
             std::string message = *(i++);
             while (i != package->alternates.end()) {
-                message.append(", ");
-                message.append(*(i++));
+                message.append(", ").append(*(i++));
             }
             _error->Error("%s:%d: No alternative available: %s", package->filename, package->linenum, message.c_str());
         }
@@ -163,10 +178,10 @@ static pkgCache::PkgIterator eval_pkg(inapt_package *package, pkgCacheFile &cach
     return pkg;
 }
 
-static bool test_macros(vector<std::string> *macros, std::set<std::string> *defines) {
+static bool test_profiles(vector<std::string> *profiles, std::set<std::string> *defines) {
     bool ok = true;
-    for (vector<std::string>::iterator j = macros->begin(); j < macros->end(); j++) {
-        if (!test_macro((*j).c_str(), defines)) {
+    for (vector<std::string>::iterator j = profiles->begin(); j < profiles->end(); j++) {
+        if (!test_anyprofile(*j, defines)) {
             ok = false;
             break;
         }
@@ -176,7 +191,7 @@ static bool test_macros(vector<std::string> *macros, std::set<std::string> *defi
 
 static void eval_action(inapt_action *action, std::set<std::string> *defines, std::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))
+        if (test_profiles(&(*i)->predicates, defines))
             final_actions->push_back(*i);
     }
 }
@@ -186,11 +201,11 @@ static void eval_block(inapt_block *block, std::set<std::string> *defines, std::
         return;
 
     for (vector<inapt_action *>::iterator i = block->actions.begin(); i < block->actions.end(); i++)
-        if (test_macros(&(*i)->predicates, defines))
+        if (test_profiles(&(*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))
+        if (test_profiles(&(*i)->predicates, defines))
             eval_block((*i)->then_block, defines, final_actions);
         else
             eval_block((*i)->else_block, defines, final_actions);
@@ -202,12 +217,12 @@ static void eval_profiles(inapt_block *block, std::set<std::string> *defines) {
         return;
 
     for (vector<inapt_profiles *>::iterator i = block->profiles.begin(); i < block->profiles.end(); i++)
-        if (test_macros(&(*i)->predicates, defines))
+        if (test_profiles(&(*i)->predicates, defines))
             for (vector<std::string>::iterator j = (*i)->profiles.begin(); j != (*i)->profiles.end(); j++)
                 defines->insert(*j);
 
     for (vector<inapt_conditional *>::iterator i = block->children.begin(); i < block->children.end(); i++) {
-        if (test_macro((*i)->condition, defines))
+        if (test_profiles(&(*i)->predicates, defines))
             eval_profiles((*i)->then_block, defines);
         else
             eval_profiles((*i)->else_block, defines);
@@ -236,15 +251,48 @@ static void dump_actions(pkgCacheFile &cache) {
     }
 }
 
+static bool sanity_check(std::vector<inapt_package *> *final_actions, pkgCacheFile &cache) {
+    bool okay = true;
+    std::map<std::string, inapt_package *> packages;
+
+    for (vector<inapt_package *>::iterator i = final_actions->begin(); i != final_actions->end(); i++) {
+        if (packages.find((*i)->pkg.Name()) != packages.end()) {
+            inapt_package *first = packages[(*i)->pkg.Name()];
+            inapt_package *current = *i;
+            _error->Error("Multiple directives for package %s at %s:%d and %s:%d",
+                    (*i)->pkg.Name(), first->filename, first->linenum, current->filename, current->linenum);
+            okay = false;
+            continue;
+        }
+        packages[(*i)->pkg.Name()] = *i;
+    }
+
+    for (pkgCache::PkgIterator i = cache->PkgBegin(); !i.end(); i++) {
+        if (cache[i].Delete() && (i->Flags & pkgCache::Flag::Essential || i->Flags & pkgCache::Flag::Important)) {
+            _error->Error("Removing essential package %s", i.Name());
+            okay = false;
+        }
+    }
+
+    return okay;
+}
+
+static void show_breakage(pkgCacheFile &cache) {
+    std::string broken;
+    for (pkgCache::PkgIterator i = cache->PkgBegin(); !i.end(); i++)
+        if (cache[i].NowBroken() || cache[i].InstBroken())
+            broken.append(" ").append(i.Name());
+
+    _error->Error("Broken packages:%s", broken.c_str());
+}
+
 static void exec_actions(std::vector<inapt_package *> *final_actions) {
     int marked = 0;
+    bool purge = _config->FindB("Inapt::Purge", false);
 
     pkgInitConfig(*_config);
     pkgInitSystem(*_config, _system);
 
-//     _config->Set("Debug::pkgProblemResolver", true);
-//     _config->Set("Debug::pkgAutoRemove", true);
-
     OpTextProgress prog;
     pkgCacheFile cache;
 
@@ -259,6 +307,7 @@ static void exec_actions(std::vector<inapt_package *> *final_actions) {
     if (_error->PendingError())
         return;
 
+    // preliminary loop (auto-installs, includes recommends - could do this manually)
     for (vector<inapt_package *>::iterator i = final_actions->begin(); i < final_actions->end(); i++) {
         pkgCache::PkgIterator k = (*i)->pkg;
         switch ((*i)->action) {
@@ -275,31 +324,36 @@ static void exec_actions(std::vector<inapt_package *> *final_actions) {
         }
     }
 
+    // secondary loop (removes package and reinstalls auto-removed packages)
     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:
                 if ((!k.CurrentVer() && !cache[k].Install()) || cache[k].Delete()) {
-                    debug("install %s %s:%d", (*i)->pkg.Name(), (*i)->filename, (*i)->linenum);
+                    debug("force install %s %s:%d", (*i)->pkg.Name(), (*i)->filename, (*i)->linenum);
                     cache->MarkInstall(k, false);
                 }
                 if (cache[k].Flags & pkgCache::Flag::Auto) {
-                    marked++;
                     debug("marking %s as manually installed", (*i)->pkg.Name());
                     cache->MarkAuto(k, false);
+                    marked++;
                 }
                 break;
             case inapt_action::REMOVE:
-                if ((k.CurrentVer() && !cache[k].Delete()) || cache[k].Install()) {
+                if ((k.CurrentVer() && !cache[k].Delete()) || cache[k].Install())
                     debug("remove %s %s:%d", (*i)->pkg.Name(), (*i)->filename, (*i)->linenum);
-                    cache->MarkDelete(k, false);
-                }
+
+                /* always mark so purge works */
+                cache->MarkDelete(k, purge);
                 break;
             default:
                 fatal("uninitialized action");
         }
     }
 
+    if (_error->PendingError())
+        return;
+
     dump_nondownloadable(cache);
     dump_actions(cache);
 
@@ -309,16 +363,17 @@ static void exec_actions(std::vector<inapt_package *> *final_actions) {
             fix.Protect((*i)->pkg);
         fix.Resolve();
 
-       if (_error->PendingError())
-          return;
-
-        dump_actions(cache);
+        if (cache->BrokenCount()) {
+            show_breakage(cache);
+            return;
+        }
     }
 
-    if (_config->FindB("Inapt::AutomaticRemove", false)) {
-        cache->MarkAndSweep();
-        run_autoremove(cache);
-    }
+    cache->MarkAndSweep();
+    run_autoremove(cache);
+
+    if (!sanity_check(final_actions, cache))
+        return;
 
     if (_config->FindB("Inapt::Simulate", false)) {
         pkgSimulate PM (cache);
@@ -358,13 +413,24 @@ static void auto_profiles(std::set<std::string> *defines) {
     defines->insert(uts.nodename);
 }
 
+static void set_option(char *opt) {
+    char *eq = strchr(opt, '=');
+    if (!eq)
+        fatal("invalid syntax for '%s': must be <option>=<value>", opt);
+
+    std::string option (opt, eq - opt);
+    std::string value (eq + 1);
+
+    _config->Set(option, value);
+}
+
 int main(int argc, char *argv[]) {
     int opt;
 
     std::set<std::string> defines;
 
     prog = xstrdup(basename(argv[0]));
-    while ((opt = getopt_long(argc, argv, "p:s", opts, NULL)) != -1) {
+    while ((opt = getopt_long(argc, argv, "p:o:sd", opts, NULL)) != -1) {
         switch (opt) {
             case 'p':
                 defines.insert(optarg);
@@ -372,15 +438,18 @@ int main(int argc, char *argv[]) {
             case '?':
                 usage();
                 break;
-            case 'z':
-                _config->Set("Inapt::AutomaticRemove", true);
-                break;
             case 's':
                 _config->Set("Inapt::Simulate", true);
                 break;
             case 'u':
                 _config->Set("Inapt::Purge", true);
                 break;
+            case 'd':
+                debug_level++;
+                break;
+            case 'o':
+                set_option(optarg);
+                break;
             default:
                 fatal("error parsing arguments");
         }
@@ -407,4 +476,6 @@ int main(int argc, char *argv[]) {
        _error->DumpErrors();
         exit(1);
     }
+
+    return 0;
 }