// file      : tests/loader/driver.cxx -*- C++ -*-
// copyright : Copyright (c) 2014-2015 Code Synthesis Ltd
// license   : MIT; see accompanying LICENSE file

#include <vector>
#include <memory>    // shared_ptr
#include <string>
#include <cassert>
#include <iostream>
#include <exception>
#include <algorithm> // sort(), find()

#include <odb/session.hxx>
#include <odb/transaction.hxx>

#include <odb/pgsql/database.hxx>

#include <butl/process>
#include <butl/timestamp>  // timestamp_nonexistent
#include <butl/filesystem>

#include <brep/package>
#include <brep/package-odb>

using namespace std;
using namespace odb::core;
using namespace butl;
using namespace brep;

static inline bool
operator== (const dependency& a, const dependency& b)
{
  return a.name == b.name && !a.constraint == !b.constraint &&
    (!a.constraint || (a.constraint->operation == b.constraint->operation &&
                      a.constraint->version == b.constraint->version));
}

static bool
check_location (shared_ptr<package_version>& pv)
{
  if (pv->internal_repository == nullptr)
    return !pv->location;
  else
    return pv->location && *pv->location ==
      path (pv->name + "-" + pv->version.string () + ".tar.gz");
}

int
main (int argc, char* argv[])
{
  if (argc != 7)
  {
    cerr << "usage: " << argv[0]
         << " <loader_path> --db-host <host> --db-port <port>"
         << " <loader_conf_file>" << endl;

    return 1;
  }

  try
  {
    path cp (argv[6]);

    // Make configuration file path absolute to use it's directory as base for
    // internal repositories relative local paths.
    //
    if (cp.relative ())
      cp.complete ();

    // Update packages file timestamp to enforce loader to update
    // persistent state.
    //
    path p (cp.directory () / path ("internal/1/stable/packages"));
    char const* args[] = {"touch", p.string ().c_str (), nullptr};
    assert (process (args).wait ());

    timestamp srt (file_mtime (p));

    // Run the loader.
    //
    char const** ld_args (const_cast<char const**> (argv + 1));
    assert (process (ld_args).wait ());

    // Check persistent objects validity.
    //
    odb::pgsql::database db ("", "", "brep", argv[3], stoul (argv[5]));

    {
      session s;
      transaction t (db.begin ());

      assert (db.query<repository> ().size () == 3);
      assert (db.query<package_version> ().size () == 9);

      shared_ptr<repository> sr (db.load<repository> ("cppget.org/stable"));
      shared_ptr<repository> mr (db.load<repository> ("cppget.org/math"));
      shared_ptr<repository> cr (db.load<repository> ("cppget.org/misc"));

      // Verify 'stable' repository.
      //
      assert (sr->location.canonical_name () == "cppget.org/stable");
      assert (sr->location.string () ==
              "http://pkg.cppget.org/internal/1/stable");
      assert (sr->display_name == "stable");

      dir_path srp (cp.directory () / dir_path ("internal/1/stable"));
      assert (sr->local_path == srp.normalize ());

      assert (sr->packages_timestamp == srt);
      assert (sr->repositories_timestamp ==
              file_mtime (dir_path (sr->local_path) / path ("repositories")));
      assert (sr->internal);

      version fv1 ("1.0");
      shared_ptr<package_version> fpv1 (
        db.load<package_version> (
          package_version_id {
            "libfoo",
            {fv1.epoch, fv1.canonical_upstream, fv1.revision}}));
      assert (check_location (fpv1));

      version fv2 ("1.2.2");
      shared_ptr<package_version> fpv2 (
        db.load<package_version> (
          package_version_id {
            "libfoo",
            {fv2.epoch, fv2.canonical_upstream, fv2.revision}}));
      assert (check_location (fpv2));

      version fv3 ("1.2.3-4");
      shared_ptr<package_version> fpv3 (
        db.load<package_version> (
          package_version_id {
            "libfoo",
            {fv3.epoch, fv3.canonical_upstream, fv3.revision}}));
      assert (check_location (fpv3));

      version fv4 ("1.2.4");
      shared_ptr<package_version> fpv4 (
        db.load<package_version> (
          package_version_id {
            "libfoo",
            {fv4.epoch, fv4.canonical_upstream, fv4.revision}}));
      assert (check_location (fpv4));

      version xv ("1.0.0-1");
      shared_ptr<package_version> xpv (
        db.load<package_version> (
          package_version_id {
            "libstudxml",
            {xv.epoch, xv.canonical_upstream, xv.revision}}));
      assert (check_location (xpv));

      // Verify libstudxml package version.
      //
      assert (xpv->name == "libstudxml");
      assert (xpv->version == version ("1.0.0-1"));
      assert (xpv->summary == "Modern C++ XML API");
      assert (xpv->tags == strings ({"c++", "xml", "parser", "serializer",
              "pull", "streaming", "modern"}));
      assert (!xpv->description);
      assert (xpv->url == "http://www.codesynthesis.com/projects/libstudxml/");
      assert (!xpv->package_url);
      assert (xpv->email ==
              email ("studxml-users@codesynthesis.com",
                     "Public mailing list, posts by  non-members "
                     "are allowed but moderated."));
      assert (xpv->package_email &&
              *xpv->package_email == email ("boris@codesynthesis.com",
                                            "Direct email to the author."));

      assert (xpv->internal_repository.load () == sr);
      assert (xpv->external_repositories.empty ());
      assert (xpv->priority == priority::low);
      assert (xpv->changes.empty ());

      assert (xpv->license_alternatives.size () == 1);
      assert (xpv->license_alternatives[0].size () == 1);
      assert (xpv->license_alternatives[0][0] == "MIT");

      assert (xpv->dependencies.size () == 2);
      assert (xpv->dependencies[0].size () == 1);
      assert (xpv->dependencies[0][0] ==
              (dependency {
                 "libexpat",
                 brep::optional<dependency_constraint> (
                   dependency_constraint{
                     comparison::ge, version ("2.0.0")})}));

      assert (xpv->dependencies[1].size () == 1);
      assert (xpv->dependencies[1][0] ==
              (dependency {
                "libgenx", brep::optional<dependency_constraint> ()}));

      assert (xpv->requirements.empty ());

      // Verify libfoo package versions.
      //
      assert (fpv1->name == "libfoo");
      assert (fpv1->version == version ("1.0"));
      assert (fpv1->summary == "The Foo Library");
      assert (fpv1->tags.empty ());
      assert (!fpv1->description);
      assert (fpv1->url == "http://www.example.com/foo/");
      assert (!fpv1->package_url);
      assert (fpv1->email == "foo-users@example.com");
      assert (!fpv1->package_email);

      assert (fpv1->internal_repository.load () == sr);
      assert (fpv1->external_repositories.size () == 1);
      assert (fpv1->external_repositories[0].load () == cr);
      assert (fpv1->priority == priority::low);
      assert (fpv1->changes.empty ());

      assert (fpv1->license_alternatives.size () == 1);
      assert (fpv1->license_alternatives[0].size () == 1);
      assert (fpv1->license_alternatives[0][0] == "MIT");

      assert (fpv1->dependencies.empty ());
      assert (fpv1->requirements.empty ());

      assert (fpv2->name == "libfoo");
      assert (fpv2->version == version ("1.2.2"));
      assert (fpv2->summary == "The Foo library");
      assert (fpv2->tags == strings ({"c++", "foo"}));
      assert (!fpv2->description);
      assert (fpv2->url == "http://www.example.com/foo/");
      assert (!fpv2->package_url);
      assert (fpv2->email == "foo-users@example.com");
      assert (!fpv2->package_email);

      assert (fpv2->internal_repository.load () == sr);
      assert (fpv2->external_repositories.empty ());
      assert (fpv2->priority == priority::low);
      assert (fpv2->changes.empty ());

      assert (fpv2->license_alternatives.size () == 1);
      assert (fpv2->license_alternatives[0].size () == 1);
      assert (fpv2->license_alternatives[0][0] == "MIT");

      assert (fpv2->dependencies.size () == 2);
      assert (fpv2->dependencies[0].size () == 1);
      assert (fpv2->dependencies[1].size () == 1);

      assert (fpv2->dependencies[0][0] ==
              (dependency {
                 "libbar",
                 brep::optional<dependency_constraint> (
                   dependency_constraint{comparison::le, version ("2.4.0")})}));

      assert (fpv2->dependencies[1][0] ==
              (dependency {
                 "libexp",
                 brep::optional<dependency_constraint> (
                   dependency_constraint{
                     comparison::eq, version ("1+1.2")})}));

      assert (fpv2->requirements.empty ());

      assert (fpv3->name == "libfoo");
      assert (fpv3->version == version ("1.2.3-4"));
      assert (fpv3->summary == "The Foo library");
      assert (fpv3->tags == strings ({"c++", "foo"}));
      assert (!fpv3->description);
      assert (fpv3->url == "http://www.example.com/foo/");
      assert (!fpv3->package_url);
      assert (fpv3->email == "foo-users@example.com");
      assert (!fpv3->package_email);

      assert (fpv3->internal_repository.load () == sr);
      assert (fpv3->external_repositories.empty ());
      assert (fpv3->priority == priority::low);

      assert (fpv3->changes.empty ());

      assert (fpv3->license_alternatives.size () == 1);
      assert (fpv3->license_alternatives[0].size () == 1);
      assert (fpv3->license_alternatives[0][0] == "MIT");

      assert (fpv3->dependencies.size () == 1);
      assert (fpv3->dependencies[0].size () == 1);
      assert (fpv3->dependencies[0][0] ==
              (dependency {
                 "libmisc",
                 brep::optional<dependency_constraint> (
                   dependency_constraint{
                     comparison::ge, version ("2.0.0")})}));

      assert (fpv4->name == "libfoo");
      assert (fpv4->version == version ("1.2.4"));
      assert (fpv4->summary == "The Foo Library");
      assert (fpv4->tags == strings ({"c++", "foo"}));
      assert (*fpv4->description == "Very good foo library.");
      assert (fpv4->url == "http://www.example.com/foo/");
      assert (!fpv4->package_url);
      assert (fpv4->email == "foo-users@example.com");
      assert (!fpv4->package_email);

      assert (fpv4->internal_repository.load () == sr);
      assert (fpv4->external_repositories.empty ());
      assert (fpv4->priority == priority::low);
      assert (fpv4->changes == "some changes 1\nsome changes 2");

      assert (fpv4->license_alternatives.size () == 1);
      assert (fpv4->license_alternatives[0].comment ==
              "Permissive free software license.");
      assert (fpv4->license_alternatives[0].size () == 1);
      assert (fpv4->license_alternatives[0][0] == "MIT");

      assert (fpv4->dependencies.size () == 1);
      assert (fpv4->dependencies[0].size () == 1);
      assert (fpv4->dependencies[0][0] ==
              (dependency {
                 "libmisc",
                 brep::optional<dependency_constraint> (
                   dependency_constraint{
                     comparison::ge, version ("2.0.0")})}));

      // Verify 'math' repository.
      //
      assert (mr->location.canonical_name () == "cppget.org/math");
      assert (mr->location.string () ==
              "http://pkg.cppget.org/internal/1/math");
      assert (mr->display_name == "math");

      dir_path mrp (cp.directory () / dir_path ("internal/1/math"));
      assert (mr->local_path == mrp.normalize ());

      assert (mr->packages_timestamp ==
              file_mtime (dir_path (mr->local_path) / path ("packages")));
      assert (mr->repositories_timestamp ==
              file_mtime (dir_path (mr->local_path) / path ("repositories")));
      assert (mr->internal);

      version ev ("1+1.2");
      shared_ptr<package_version> epv (
        db.load<package_version> (
          package_version_id {
            "libexp",
            {ev.epoch, ev.canonical_upstream, ev.revision}}));
      assert (check_location (epv));

      version fv5 ("1.2.4-1");
      shared_ptr<package_version> fpv5 (
        db.load<package_version> (
          package_version_id {
            "libfoo",
            {fv5.epoch, fv5.canonical_upstream, fv5.revision}}));
      assert (check_location (fpv5));

      // Verify libfoo package versions.
      //
      assert (fpv5->name == "libfoo");
      assert (fpv5->version == version ("1.2.4-1"));
      assert (fpv5->summary == "The Foo Math Library");
      assert (fpv5->tags == strings ({"c++", "foo", "math"}));
      assert (*fpv5->description ==
              "A modern C++ library with easy to use linear algebra and "
              "optimization tools. There are over 100 functions in total with "
              "an extensive test suite. The API is similar to MATLAB.");
      assert (fpv5->url == "http://www.example.com/foo/");
      assert (fpv5->package_url &&
              *fpv5->package_url == "http://www.example.com/foo/pack");
      assert (fpv5->email == "foo-users@example.com");
      assert (fpv5->package_email &&
              *fpv5->package_email == "pack@example.com");

      assert (fpv5->internal_repository.load () == mr);
      assert (fpv5->external_repositories.size () == 1);
      assert (fpv5->external_repositories[0].load () == cr);

      assert (fpv5->priority == priority::high);
      assert (fpv5->priority.comment == "Due to critical bug fix.");

      const char ch[] = R"DLM(1.2.4-1
 * applied patch for critical bug-219
 * regenerated documentation

1.2.4
 * test suite extended significantly)DLM";

      assert (fpv5->changes == ch);

      assert (fpv5->license_alternatives.size () == 2);
      assert (fpv5->license_alternatives[0].comment ==
              "If using with GNU TLS.");
      assert (fpv5->license_alternatives[0].size () == 2);
      assert (fpv5->license_alternatives[0][0] == "LGPLv2");
      assert (fpv5->license_alternatives[0][1] == "MIT");
      assert (fpv5->license_alternatives[1].comment ==
              "If using with OpenSSL.");
      assert (fpv5->license_alternatives[1].size () == 1);
      assert (fpv5->license_alternatives[1][0] == "BSD");

      assert (fpv5->dependencies.size () == 2);
      assert (fpv5->dependencies[0].size () == 2);
      assert (fpv5->dependencies[0].comment ==
              "Crashes in range [1.1, 2.3.0].");

      assert (fpv5->dependencies[0][0] ==
              (dependency {
                 "libmisc",
                 brep::optional<dependency_constraint> (
                   dependency_constraint{comparison::lt, version ("1.1")})}));

      assert (fpv5->dependencies[0][1] ==
              (dependency {
                 "libmisc",
                 brep::optional<dependency_constraint> (
                   dependency_constraint{
                     comparison::gt, version ("2.3.0")})}));

      assert (fpv5->dependencies[1].size () == 1);
      assert (fpv5->dependencies[1].comment == "Newer - better.");

      assert (fpv5->dependencies[1][0] ==
              (dependency {"libstudxml", nullopt}));

      requirements& fpvr5 (fpv5->requirements);
      assert (fpvr5.size () == 4);

      assert (fpvr5[0] == strings ({"linux", "windows", "macosx"}));
      assert (!fpvr5[0].conditional);
      assert (fpvr5[0].comment == "Symbian is coming.");

      assert (fpvr5[1] == strings ({"c++11"}));
      assert (!fpvr5[1].conditional);
      assert (fpvr5[1].comment.empty ());

      assert (fpvr5[2] == strings ({"VC++"}));
      assert (fpvr5[2].conditional);
      assert (fpvr5[2].comment == "12.0 or later if targeting Windows.");

      assert (fpvr5[3].empty ());
      assert (fpvr5[3].conditional);
      assert (fpvr5[3].comment ==
              "libc++ standard library if using Clang on Mac OS X.");

      // Verify libexp package version.
      //
      assert (epv->name == "libexp");
      assert (epv->version == version ("1+1.2"));
      assert (epv->summary == "The exponent");
      assert (epv->tags == strings ({"c++", "exponent"}));
      assert (!epv->description);
      assert (epv->url == "http://www.exp.com");
      assert (!epv->package_url);
      assert (epv->email == email ("users@exp.com"));
      assert (!epv->package_email);

      assert (epv->internal_repository.load () == mr);
      assert (epv->external_repositories.empty ());
      assert (epv->priority == priority (priority::low));
      assert (epv->changes.empty ());

      assert (epv->license_alternatives.size () == 1);
      assert (epv->license_alternatives[0].size () == 1);
      assert (epv->license_alternatives[0][0] == "MIT");

      assert (epv->dependencies.size () == 1);
      assert (epv->dependencies[0].size () == 1);
      assert (epv->dependencies[0][0] ==
              (dependency {
                "libmisc", brep::optional<dependency_constraint> ()}));

      assert (epv->requirements.empty ());

      // Verify 'misc' repository.
      //
      assert (cr->location.canonical_name () == "cppget.org/misc");
      assert (cr->location.string () ==
              "http://pkg.cppget.org/external/1/misc");
      assert (cr->display_name.empty ());

      dir_path crp (cp.directory () / dir_path ("external/1/misc"));
      assert (cr->local_path == crp.normalize ());

      assert (cr->packages_timestamp ==
              file_mtime (dir_path (cr->local_path) / path ("packages")));
      assert (cr->repositories_timestamp == timestamp_nonexistent);
      assert (!cr->internal);

      version bv ("2.3.5");
      shared_ptr<package_version> bpv (
        db.load<package_version> (
          package_version_id {
            "libbar",
            {bv.epoch, bv.canonical_upstream, bv.revision}}));
      assert (check_location (bpv));

      version fv0 ("0.1");
      shared_ptr<package_version> fpv0 (
        db.load<package_version> (
          package_version_id {
            "libfoo",
            {fv0.epoch, fv0.canonical_upstream, fv0.revision}}));
      assert (check_location (fpv0));

      // Verify libbar package version.
      //
      assert (bpv->name == "libbar");
      assert (bpv->version == version ("2.3.5"));
      assert (bpv->summary.empty ());
      assert (bpv->tags.empty ());
      assert (!bpv->description);
      assert (bpv->url.empty ());
      assert (!bpv->package_url);
      assert (bpv->email.empty ());
      assert (!bpv->package_email);

      assert (bpv->internal_repository == nullptr);
      assert (bpv->external_repositories.size () == 1);
      assert (bpv->external_repositories[0].load () == cr);

      assert (bpv->priority == priority ());
      assert (bpv->changes.empty ());

      assert (bpv->license_alternatives.empty ());
      assert (bpv->dependencies.empty ());
      assert (bpv->requirements.empty ());

      // Verify libfoo package versions.
      //
      assert (fpv0->name == "libfoo");
      assert (fpv0->version == version ("0.1"));
      assert (fpv0->summary.empty ());
      assert (fpv0->tags.empty ());
      assert (!fpv0->description);
      assert (fpv0->url.empty ());
      assert (!fpv0->package_url);
      assert (fpv0->email.empty ());
      assert (!fpv0->package_email);

      assert (fpv0->internal_repository.load () == nullptr);
      assert (fpv0->external_repositories.size () == 1);
      assert (fpv0->external_repositories[0].load () == cr);
      assert (fpv0->priority == priority::low);
      assert (fpv0->changes.empty ());

      assert (fpv0->license_alternatives.empty ());

      assert (fpv0->dependencies.empty ());
      assert (fpv0->requirements.empty ());

      // Change package version summary, update the object persistent
      // state, rerun loader and ensure the model were not rebuilt.
      //
      bpv->summary = "test";
      db.update (bpv);

      t.commit ();
    }

    assert (process (ld_args).wait ());
    transaction t (db.begin ());

    version bv ("2.3.5");
    shared_ptr<package_version> bpv (
      db.load<package_version> (
        package_version_id {
          "libbar",
          {bv.epoch, bv.canonical_upstream, bv.revision}}));

    assert (bpv->summary == "test");

    t.commit ();
  }
  // Fully qualified to avoid ambiguity with odb exception.
  //
  catch (const std::exception& e)
  {
    cerr << e.what () << endl;
    return 1;
  }
}