// file      : qt/driver.cxx
// author    : Constantin Michael <constantin@codesynthesis.com>
// copyright : not copyrighted - public domain

#include <iostream>

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

#include "database.hxx" // create_database

#include "employee.hxx"
#include "employee-odb.hxx"

using namespace std;
using namespace odb::core;

ostream& operator << (ostream& os, const QString& s)
{
  return os << s.toStdString ();
}

int
main (int argc, char* argv[])
{
  try
  {
    auto_ptr<database> db (create_database (argc, argv));

    // Create a few persistent objects.
    //
    {
      // Simple Tech Ltd.
      //
      {
        QSharedPointer<employer> er (new employer ("Simple Tech Ltd"));

        QSharedPointer<employee> john (
          new employee ("John",
                        "Doe",
                        QDate (1974, 5, 23),
                        QByteArray ("\0xF1\0x00\0x34\0x45\0x00\0xDE", 6),
                        er));

        QSharedPointer<employee> jane (
          new employee ("Jane",
                        "Smith",
                        QDate (1983, 1, 18),
                        QByteArray ("\0xD7\0x00\0x14", 3),
                        er));

        john->emails ().insert ("john_d@example.com");
        john->emails ().insert ("john.doe@simple.com");
        jane->emails ().insert ("jane_s@example.com");
        jane->emails ().insert ("jane.smith@simple.com");

        // Set the inverse side of the employee-employer relationship.
        //
        er->employees ().push_back (john);
        er->employees ().push_back (jane);

        transaction t (db->begin ());

        db->persist (er);

        db->persist (john);
        db->persist (jane);

        t.commit ();
      }

      // Complex Systems Inc.
      //
      {
        QSharedPointer<employer> er (new employer ("Complex Systems Inc"));

        QSharedPointer<employee> john (
          new employee ("John",
                        "Smith",
                        QDate (1954, 8, 1),
                        QByteArray ("\0x23\0xFD\0x8F\0x00", 4),
                        er));

        QSharedPointer<employee> jane (
          new employee ("Jane",
                        "Smith",
                        QDate (1976, 12, 31),
                        QByteArray ("0x00\0x32\0x00\0x01\0x00", 5),
                        er));

        john->emails ().insert ("john_d@example.com");
        john->emails ().insert ("john.doe@complex.com");
        jane->emails ().insert ("jane_s@example.com");
        jane->emails ().insert ("jane.smith@complex.com");

        // Set the inverse side of the employee-employer relationship.
        //
        er->employees ().push_back (john);
        er->employees ().push_back (jane);

        transaction t (db->begin ());

        db->persist (er);

        db->persist (john);
        db->persist (jane);

        t.commit ();
      }
    }


    // Load Simple Tech Ltd and print its employees.
    //
    {
      session s;
      transaction t (db->begin ());

      QSharedPointer<employer> stl (db->load<employer> ("Simple Tech Ltd"));

      employees& es (stl->employees ());

      for (employees::iterator i (es.begin ()); i != es.end (); ++i)
      {
        QLazyWeakPointer<employee>& lwp (*i);

        // Load and lock the employee and his employer.
        //
        QSharedPointer<employee> p (lwp.load ());
        QSharedPointer<employer> pe (p->employer ().load ());

        cout << p->first () << " " << p->last ()  << endl
             << "  born: " << p->born ().toString () << endl;

        for (emails::const_iterator j (p->emails ().begin ()),
               e (p->emails ().end ()); j != e; ++j)
        {
          cout << "  email: " << *j << endl;
        }

        cout  << "  public key length: " << p->public_key ().size () << endl
              << "  employer: "
              << pe->name () << endl
              << endl;
      }

      t.commit ();
    }

    // Search for Complex Systems Inc employees.
    //
    {
      typedef odb::query<employee> query;
      typedef odb::result<employee> result;

      session s;
      transaction t (db->begin ());

      result r (db->query<employee> (
                  query::employer::name == "Complex Systems Inc"));

      for (result::iterator i (r.begin ()); i != r.end (); ++i)
        cout << i->first () << " " << i->last () << endl;

      t.commit ();
    }
  }
  catch (const odb::exception& e)
  {
    cerr << e.what () << endl;
    return 1;
  }
}