#include <config.h>

#include <apt-pkg/cdrom.h>
#include <apt-pkg/error.h>
#include <apt-pkg/fileutl.h>

#include <algorithm>
#include <string>
#include <vector>
#include <stddef.h>
#include <iostream>

#include <gtest/gtest.h>

#include "file-helpers.h"

class Cdrom : public pkgCdrom {
   public:
      bool FindPackages(std::string const &CD,
	    std::vector<std::string> &List,
	    std::vector<std::string> &SList,
	    std::vector<std::string> &SigList,
	    std::vector<std::string> &TransList,
	    std::string &InfoDir) {
	 std::string const startdir = SafeGetCWD();
	 EXPECT_FALSE(startdir.empty());
	 bool const result = pkgCdrom::FindPackages(CD, List, SList, SigList, TransList, InfoDir, NULL, 0);
	 std::sort(List.begin(), List.end());
	 std::sort(SList.begin(), SList.end());
	 std::sort(SigList.begin(), SigList.end());
	 std::sort(TransList.begin(), TransList.end());
	 EXPECT_EQ(0, chdir(startdir.c_str()));
	 return result;
      }

      using pkgCdrom::DropRepeats;
};

TEST(CDROMTest,FindPackages)
{
   std::string path;
   createTemporaryDirectory("findpackage", path);

   createDirectory(path, ".disk");
   createDirectory(path, "pool");
   createDirectory(path, "dists/stable/main/binary-i386");
   createDirectory(path, "dists/stable/main/source");
   createDirectory(path, "dists/stable/contrib/binary-amd64");
   createDirectory(path, "dists/stable/contrib/binary-all");
   createDirectory(path, "dists/unstable/main/binary-i386");
   createDirectory(path, "dists/unstable/main/i18n");
   createDirectory(path, "dists/unstable/main/source");
   createDirectory(path, "dists/broken/non-free/source");
   createFile(path, "dists/broken/.aptignr");
   createFile(path, "dists/stable/main/binary-i386/Packages");
   createFile(path, "dists/stable/main/binary-i386/Packages.bz2");
   createFile(path, "dists/stable/main/source/Sources.xz");
   createFile(path, "dists/stable/contrib/binary-amd64/Packages");
   createFile(path, "dists/stable/contrib/binary-amd64/Packages.gz");
   createFile(path, "dists/stable/contrib/binary-all/Packages");
   createFile(path, "dists/unstable/main/binary-i386/Packages.xz");
   createFile(path, "dists/unstable/main/binary-i386/Packages.lzma");
   createFile(path, "dists/unstable/main/i18n/Translation-en");
   createFile(path, "dists/unstable/main/i18n/Translation-de.bz2");
   createFile(path, "dists/unstable/main/source/Sources.xz");
   createFile(path, "dists/broken/non-free/source/Sources.gz");
   createFile(path, "dists/stable/Release.gpg");
   createFile(path, "dists/stable/Release");
   createFile(path, "dists/unstable/InRelease");
   createFile(path, "dists/broken/Release.gpg");
   createLink(path, "dists/unstable", "dists/sid");

   Cdrom cd;
   std::vector<std::string> Packages, Sources, Signatur, Translation;
   std::string InfoDir;
   EXPECT_TRUE(cd.FindPackages(path, Packages, Sources, Signatur, Translation, InfoDir));
   EXPECT_EQ(4, Packages.size());
   EXPECT_EQ(path + "/dists/sid/main/binary-i386/", Packages[0]);
   EXPECT_EQ(path + "/dists/stable/contrib/binary-amd64/", Packages[1]);
   EXPECT_EQ(path + "/dists/stable/main/binary-i386/", Packages[2]);
   EXPECT_EQ(path + "/dists/unstable/main/binary-i386/", Packages[3]);
   EXPECT_EQ(3, Sources.size());
   EXPECT_EQ(path + "/dists/sid/main/source/", Sources[0]);
   EXPECT_EQ(path + "/dists/stable/main/source/", Sources[1]);
   EXPECT_EQ(path + "/dists/unstable/main/source/", Sources[2]);
   EXPECT_EQ(3, Signatur.size());
   EXPECT_EQ(path + "/dists/sid/", Signatur[0]);
   EXPECT_EQ(path + "/dists/stable/", Signatur[1]);
   EXPECT_EQ(path + "/dists/unstable/", Signatur[2]);
   EXPECT_EQ(4, Translation.size());
   EXPECT_EQ(path + "/dists/sid/main/i18n/Translation-de", Translation[0]);
   EXPECT_EQ(path + "/dists/sid/main/i18n/Translation-en", Translation[1]);
   EXPECT_EQ(path + "/dists/unstable/main/i18n/Translation-de", Translation[2]);
   EXPECT_EQ(path + "/dists/unstable/main/i18n/Translation-en", Translation[3]);
   EXPECT_EQ(path + "/.disk/", InfoDir);

   cd.DropRepeats(Packages, "Packages");
   cd.DropRepeats(Sources, "Sources");
   _error->PushToStack();
   cd.DropRepeats(Signatur, "InRelease");
   cd.DropRepeats(Signatur, "Release.gpg");
   _error->RevertToStack();
   _error->DumpErrors();
   cd.DropRepeats(Translation, "");

   EXPECT_EQ(3, Packages.size());
   EXPECT_EQ(path + "/dists/stable/contrib/binary-amd64/", Packages[0]);
   EXPECT_EQ(path + "/dists/stable/main/binary-i386/", Packages[1]);
   EXPECT_EQ(path + "/dists/unstable/main/binary-i386/", Packages[2]);
   EXPECT_EQ(2, Sources.size());
   EXPECT_EQ(path + "/dists/stable/main/source/", Sources[0]);
   EXPECT_EQ(path + "/dists/unstable/main/source/", Sources[1]);
   EXPECT_EQ(2, Signatur.size());
   EXPECT_EQ(path + "/dists/stable/", Signatur[0]);
   EXPECT_EQ(path + "/dists/unstable/", Signatur[1]);
   EXPECT_EQ(2, Translation.size());
   EXPECT_EQ(path + "/dists/unstable/main/i18n/Translation-de", Translation[0]);
   EXPECT_EQ(path + "/dists/unstable/main/i18n/Translation-en", Translation[1]);

   removeDirectory(path);
}
