report warnings&errors consistently in edit-sources

[apt.git] / apt-private / private-sources.cc

diff --git a/apt-private/private-sources.cc b/apt-private/private-sources.cc
index 5f61a23abff1a74234a7374c2bbc93cf685be8a0..7e64d5d7f4e001c00a445f57d3c1a53c6abf24ea 100644 (file)
--- a/apt-private/private-sources.cc
+++ b/apt-private/private-sources.cc
@@ -7,6 +7,7 @@
 #include <apt-pkg/cmndline.h>
 #include <apt-pkg/error.h>
 #include <apt-pkg/fileutl.h>
 #include <apt-pkg/cmndline.h>
 #include <apt-pkg/error.h>
 #include <apt-pkg/fileutl.h>

+#include <apt-pkg/cachefile.h>

 
 #include <apt-private/private-output.h>
 #include <apt-private/private-sources.h>
 
 #include <apt-private/private-output.h>
 #include <apt-private/private-sources.h>


@@ -20,17 +21,19 @@
 #include <apti18n.h>
 
 /* Interface discussion with donkult (for the future):
 #include <apti18n.h>
 
 /* Interface discussion with donkult (for the future):

-  apt [add-{archive,release,component}|edit|change-release|disable]-sources 
+  apt [add-{archive,release,component}|edit|change-release|disable]-sources

  and be clever and work out stuff from the Release file
 */
 
  and be clever and work out stuff from the Release file
 */
 

-// EditSource - EditSourcesList                                        /*{{{*/
-// ---------------------------------------------------------------------
+// EditSource - EditSourcesList                                                /*{{{*/
+class APT_HIDDEN ScopedGetLock {
+public:
+   int fd;
+   ScopedGetLock(std::string const &filename) : fd(GetLock(filename)) {}
+   ~ScopedGetLock() { close(fd); }
+};

 bool EditSources(CommandLine &CmdL)
 {
 bool EditSources(CommandLine &CmdL)
 {

-   bool res;
-   pkgSourceList sl;
-

    std::string sourceslist;
    if (CmdL.FileList[1] != NULL)
    {
    std::string sourceslist;
    if (CmdL.FileList[1] != NULL)
    {


@@ -44,30 +47,45 @@ bool EditSources(CommandLine &CmdL)
    if (FileExists(sourceslist))
        before.FromFile(sourceslist);
 
    if (FileExists(sourceslist))
        before.FromFile(sourceslist);
 

-   int lockfd = GetLock(sourceslist);
-   if (lockfd < 0)
+   ScopedGetLock lock(sourceslist);
+   if (lock.fd < 0)

       return false;
 
       return false;
 

+   bool res;
+   bool file_changed = false;

    do {
    do {

-      EditFileInSensibleEditor(sourceslist);
-      _error->PushToStack();
-      res = sl.Read(sourceslist);
-      if (!res) {
+      if (EditFileInSensibleEditor(sourceslist) == false)
+        return false;
+      if (FileExists(sourceslist) && !before.VerifyFile(sourceslist))
+      {
+        file_changed = true;
+        pkgCacheFile::RemoveCaches();
+      }
+      pkgCacheFile CacheFile;
+      res = CacheFile.BuildCaches(nullptr);
+      if (res == false || _error->empty(GlobalError::WARNING) == false) {

         std::string outs;
         strprintf(outs, _("Failed to parse %s. Edit again? "), sourceslist.c_str());
          // FIXME: should we add a "restore previous" option here?
         std::string outs;
         strprintf(outs, _("Failed to parse %s. Edit again? "), sourceslist.c_str());
          // FIXME: should we add a "restore previous" option here?

-         res = !YnPrompt(outs.c_str(), true);
+         if (YnPrompt(outs.c_str(), true) == false)
+        {
+           if (res == false && _error->PendingError() == false)
+           {
+              CacheFile.Close();
+              pkgCacheFile::RemoveCaches();
+              res = CacheFile.BuildCaches(nullptr);
+           }
+           break;
+        }

       }
       }

-      _error->RevertToStack();

    } while (res == false);
    } while (res == false);

-   close(lockfd);

 
 

-   if (FileExists(sourceslist) && !before.VerifyFile(sourceslist)) {
+   if (res == true && file_changed == true)
+   {

       ioprintf(
          std::cout, _("Your '%s' file changed, please run 'apt-get update'."),
          sourceslist.c_str());
    }
       ioprintf(
          std::cout, _("Your '%s' file changed, please run 'apt-get update'."),
          sourceslist.c_str());
    }

-
-   return true;
+   return res;

 }
                                                                        /*}}}*/
 }
                                                                        /*}}}*/