// file      : odb/validator.cxx
// copyright : Copyright (c) 2009-2011 Code Synthesis Tools CC
// license   : GNU GPL v3; see accompanying LICENSE file

#include <odb/gcc.hxx>

#include <iostream>

#include <odb/traversal.hxx>
#include <odb/common.hxx>
#include <odb/context.hxx>
#include <odb/validator.hxx>

using namespace std;

namespace
{
  // Resolve null overrides.
  //
  static void
  override_null (semantics::node& n, string const& prefix = "")
  {
    string p (prefix.empty () ? prefix : prefix + '-');

    if (n.count (p + "null") && n.count (p + "not-null"))
    {
      if (n.get<location_t> (p + "null-location") <
          n.get<location_t> (p + "not-null-location"))
      {
        n.remove (p + "null");
        n.remove (p + "null-location");
      }
      else
      {
        n.remove (p + "not-null");
        n.remove (p + "not-null-location");
      }
    }
  }

  //
  // Pass 1.
  //

  struct data_member: traversal::data_member
  {
    data_member (bool& valid)
        : valid_ (valid)
    {
    }

    virtual void
    traverse (type& m)
    {
      if (context::transient (m))
        return;

      count_++;
      semantics::names* hint;
      semantics::type& t (context::utype (m, hint));

      if (t.fq_anonymous (hint))
      {
        cerr << m.file () << ":" << m.line () << ":" << m.column () << ":"
             << " error: unnamed type in data member declaration" << endl;

        cerr << m.file () << ":" << m.line () << ":" << m.column () << ":"
             << " info: use 'typedef' to name this type" << endl;

        valid_ = false;
      }

      // Make sure id or inverse member is not marked readonly since we
      // depend on these three sets not having overlaps. Once we support
      // composite ids, we will also need to make sure there are no
      // nested readonly members (probably move it to pass 2 and use
      // column_count()).
      //
      if (m.count ("readonly"))
      {
        if (m.count ("id"))
        {
          cerr << m.file () << ":" << m.line () << ":" << m.column () << ":"
               << " error: object id should not be declared readonly" << endl;

          valid_ = false;
        }

        if (m.count ("inverse"))
        {
          cerr << m.file () << ":" << m.line () << ":" << m.column () << ":"
               << " error: inverse object pointer should not be declared "
               << "readonly" << endl;

          valid_ = false;
        }
      }

      // Resolve null overrides.
      //
      override_null (m);
      override_null (m, "value");
    }

    bool& valid_;
    size_t count_;
  };

  // Find special members (id, version).
  //
  struct special_members: traversal::class_
  {
    special_members (class_kind kind,
                     bool& valid,
                     semantics::data_member*& id,
                     semantics::data_member*& optimistic)
        : kind_ (kind), member_ (valid, id, optimistic)
    {
      if (kind != class_view)
        *this >> inherits_ >> *this;

      *this >> names_ >> member_;
    }

    virtual void
    traverse (semantics::class_& c)
    {
      // Skip transient bases.
      //
      switch (kind_)
      {
      case class_object:
        {
          if (!context::object (c))
            return;
          break;
        }
      case class_view:
        {
          break;
        }
      case class_composite:
        {
          if (!context::composite (c))
            return;
          break;
        }
      case class_other:
        {
          assert (false);
          break;
        }
      }

      // Views don't have bases.
      //
      if (kind_ != class_view)
        inherits (c);

      names (c);
    }

  private:
    struct member: traversal::data_member
    {
      member (bool& valid,
              semantics::data_member*& id,
              semantics::data_member*& optimistic)
          : valid_ (valid), id_ (id), optimistic_ (optimistic)
      {
      }

      virtual void
      traverse (semantics::data_member& m)
      {
        if (m.count ("id"))
        {
          if (id_ != 0)
          {
            cerr << m.file () << ":" << m.line () << ":" << m.column () << ":"
                 << " error: multiple object id members" << endl;

            semantics::data_member& i (*id_);

            cerr << i.file () << ":" << i.line () << ":" << i.column ()
                 << ": info: previous id member is declared here" << endl;

            valid_ = false;
          }
          else
            id_ = &m;
        }

        if (m.count ("version"))
        {
          if (optimistic_ != 0)
          {
            cerr << m.file () << ":" << m.line () << ":" << m.column () << ":"
                 << " error: multiple version members" << endl;

            semantics::data_member& o (*optimistic_);

            cerr << o.file () << ":" << o.line () << ":" << o.column ()
                 << ": info: previous version member is declared here" << endl;

            valid_ = false;
          }
          else
            optimistic_ = &m;
        }
      }

      bool& valid_;
      semantics::data_member*& id_;
      semantics::data_member*& optimistic_;
    };

    class_kind kind_;
    member member_;
    traversal::names names_;
    traversal::inherits inherits_;
  };

  //
  //
  struct value_type: traversal::type
  {
    value_type (bool& valid): valid_ (valid) {}

    virtual void
    traverse (semantics::type& t)
    {
      // Resolve null overrides.
      //
      override_null (t);
      override_null (t, "value");
    }

    bool& valid_;
  };

  //
  //
  struct class1: traversal::class_
  {
    class1 (bool& valid,
            options const& ops,
            semantics::unit& unit,
            value_type& vt)
        : valid_ (valid),
          options_ (ops),
          unit_ (unit),
          vt_ (vt),
          member_ (valid)
    {
      *this >> names_ >> member_;
    }

    virtual void
    traverse (type& c)
    {
      if (context::object (c))
        traverse_object (c);
      else if (context::view (c))
        traverse_view (c);
      else
      {
        if (context::composite (c))
          traverse_composite (c);

        vt_.dispatch (c);
      }
    }

    virtual void
    traverse_object (type& c)
    {
      // Check that the callback function exist.
      //
      if (c.count ("callback"))
      {
        string name (c.get<string> ("callback"));
        tree decl (
          lookup_qualified_name (
            c.tree_node (), get_identifier (name.c_str ()), false, false));

        if (decl == error_mark_node || TREE_CODE (decl) != BASELINK)
        {
          cerr << c.file () << ":" << c.line () << ":" << c.column () << ": "
               << "error: unable to resolve member function '" << name << "' "
               << "specified with '#pragma db callback' for class '"
               << context::class_name (c) << "'" << endl;

          valid_ = false;
        }

        // Figure out if we have a const version of the callback. OVL_*
        // macros work for both FUNCTION_DECL and OVERLOAD.
        //
        for (tree o (BASELINK_FUNCTIONS (decl)); o != 0; o = OVL_NEXT (o))
        {
          tree f (OVL_CURRENT (o));
          if (DECL_CONST_MEMFUNC_P (f))
          {
            c.set ("callback-const", true);
            break;
          }
        }

        //@@ Would be nice to check the signature of the function(s)
        //   instead of postponing it until the C++ compilation. Though
        //   we may still get C++ compilation errors because of const
        //   mismatch.
        //
      }

      // Check bases.
      //
      bool base (false);

      for (type::inherits_iterator i (c.inherits_begin ());
           i != c.inherits_end ();
           ++i)
      {
        type& b (i->base ());

        if (context::object (b))
          base = true;
        else if (context::view (b) || context::composite (b))
        {
          // @@ Should we use hint here?
          //
          string name (context::class_fq_name (b));

          cerr << c.file () << ":" << c.line () << ":" << c.column () << ":"
               << " error: base class '" << name << "' is a view or value type"
               << endl;

          cerr << c.file () << ":" << c.line () << ":" << c.column () << ":"
               << " info: object types cannot derive from view or value "
               << "types"
               << endl;

          cerr << b.file () << ":" << b.line () << ":" << b.column () << ":"
               << " info: class '" << name << "' is defined here" << endl;

          valid_ = false;
        }
      }

      // Check special members.
      //
      semantics::data_member* id (0);
      semantics::data_member* optimistic (0);
      {
        special_members t (class_object, valid_, id, optimistic);
        t.traverse (c);
      }

      if (id == 0)
      {
        // An object without an id should either be abstract or
        // explicitly marked as such.
        //
        if (!(c.count ("id") || context::abstract (c)))
        {
          cerr << c.file () << ":" << c.line () << ":" << c.column () << ":"
               << " error: no data member designated as an object id" << endl;

          cerr << c.file () << ":" << c.line () << ":" << c.column () << ":"
               << " info: use '#pragma db id' to specify an object id member"
               << endl;

          cerr << c.file () << ":" << c.line () << ":" << c.column () << ":"
               << " info: or explicitly declare that this persistent class "
               << "has no object id" << endl;

          valid_ = false;
        }
      }
      else
      {
        c.set ("id-member", id);

        // Complain if an id member has a default value (default value
        // for the id's type is ok -- we will ignore it).
        //
        if (id->count ("default"))
        {
          cerr << id->file () << ":" << id->line () << ":" << id->column ()
               << ": error: object id member cannot have default value"
               << endl;

          valid_ = false;
        }

        // Automatically mark the id member as not null. If we already have
        // an explicit null pragma for this member, issue an error.
        //
        if (id->count ("null"))
        {
          cerr << id->file () << ":" << id->line () << ":" << id->column ()
               << ": error: object id member cannot be null" << endl;

          valid_ = false;
        }
        else
          id->set ("not-null", true);
      }

      if (optimistic != 0)
      {
        semantics::data_member& m (*optimistic);

        // Make sure we have the class declared optimistic.
        //
        if (&optimistic->scope () == &c && !c.count ("optimistic"))
        {
          cerr << m.file () << ":" << m.line () << ":" << m.column () << ":"
               << " error: version data member in a class not declared "
               << "optimistic" << endl;

          cerr << c.file () << ":" << c.line () << ":" << c.column () << ":"
               << " info: use '#pragma db optimistic' to declare this "
               << "class optimistic" << endl;

          valid_ = false;
        }

        // Make sure we have object id.
        //
        if (id == 0)
        {
          cerr << c.file () << ":" << c.line () << ":" << c.column () << ":"
               << " error: optimistic class without an object id" << endl;

          valid_ = false;
        }

        // Make sure id and version members are in the same class. The
        // current architecture relies on that.
        //
        if (id != 0 && &id->scope () != &optimistic->scope ())
        {
          cerr << c.file () << ":" << c.line () << ":" << c.column () << ":"
               << " error: object id and version members are in different "
               << "classes" << endl;

          cerr << c.file () << ":" << c.line () << ":" << c.column () << ":"
               << " info: object id and version members must be in the same "
               << "class" << endl;

          cerr << id->file () << ":" << id->line () << ":" << id->column ()
               << ": info: object id member is declared here" << endl;

          cerr << m.file () << ":" << m.line () << ":" << m.column () << ":"
               << " error: version member is declared here" << endl;

          valid_ = false;
        }

        // Make sure this class is not readonly.
        //
        if (c.count ("readonly"))
        {
          cerr << c.file () << ":" << c.line () << ":" << c.column () << ":"
               << " error: optimistic class cannot be readonly" << endl;

          valid_ = false;
        }


        // This takes care of also marking derived classes as optimistic.
        //
        c.set ("optimistic-member", optimistic);
      }
      else
      {
        // Make sure there is a version member if the class declared
        // optimistic.
        //
        if (c.count ("optimistic"))
        {
          cerr << c.file () << ":" << c.line () << ":" << c.column () << ":"
               << " error: optimistic class without a version member" << endl;

          cerr << c.file () << ":" << c.line () << ":" << c.column () << ":"
               << " info: use '#pragma db version' to declare on of the "
               << "data members as a version" << endl;

          valid_ = false;
        }
      }

      // Check members.
      //
      member_.count_ = 0;
      names (c);

      if (member_.count_ == 0 && !base)
      {
        cerr << c.file () << ":" << c.line () << ":" << c.column () << ":"
             << " error: no persistent data members in the class" << endl;

        valid_ = false;
      }
    }

    virtual void
    traverse_view (type& c)
    {
      // Views require query support.
      //
      if (!options_.generate_query ())
      {
        cerr << c.file () << ":" << c.line () << ":" << c.column () << ":"
             << " error: query support is required when using views"
             << endl;

        cerr << c.file () << ":" << c.line () << ":" << c.column () << ":"
             << " info: use the --generate-query option to enable query "
             << "support"
             << endl;

        valid_ = false;
      }

      // Check bases.
      //
      for (type::inherits_iterator i (c.inherits_begin ());
           i != c.inherits_end ();
           ++i)
      {
        type& b (i->base ());

        if (context::object (b) ||
            context::view (b) ||
            context::composite (b))
        {
          // @@ Should we use hint here?
          //
          string name (context::class_fq_name (b));

          cerr << c.file () << ":" << c.line () << ":" << c.column () << ":"
               << " error: base class '" << name << "' is an object, "
               << "view, or value type"
               << endl;

          cerr << c.file () << ":" << c.line () << ":" << c.column () << ":"
               << " info: view types cannot derive from view, object or "
               << "value types"
               << endl;

          cerr << b.file () << ":" << b.line () << ":" << b.column () << ":"
               << " info: class '" << name << "' is defined here" << endl;

          valid_ = false;
        }
      }

      // Check id.
      //
      semantics::data_member* id (0);
      semantics::data_member* optimistic (0);
      {
        special_members t (class_view, valid_, id, optimistic);
        t.traverse (c);
      }

      if (id != 0)
      {
        cerr << id->file () << ":" << id->line () << ":" << id->column ()
             << ": error: view type data member cannot be designated as an "
             << "object id" << endl;

        valid_ = false;
      }

      if (optimistic != 0)
      {
        semantics::data_member& o (*optimistic);

        cerr << o.file () << ":" << o.line () << ":" << o.column ()
             << ": error: view type data member cannot be designated as a "
             << "version" << endl;

        valid_ = false;
      }

      // Check members.
      //
      member_.count_ = 0;
      names (c);

      if (member_.count_ == 0)
      {
        cerr << c.file () << ":" << c.line () << ":" << c.column () << ":"
             << " error: no persistent data members in the class" << endl;

        valid_ = false;
      }
    }

    virtual void
    traverse_composite (type& c)
    {
      bool base (false);

      for (type::inherits_iterator i (c.inherits_begin ());
           i != c.inherits_end ();
           ++i)
      {
        type& b (i->base ());

        if (context::composite (b))
          base = true;
        else if (context::object (b) || context::view (b))
        {
          // @@ Should we use hint here?
          //
          string name (context::class_fq_name (b));

          cerr << c.file () << ":" << c.line () << ":" << c.column () << ":"
               << " error: base class '" << name << "' is a view or object "
               << "type"
               << endl;

          cerr << c.file () << ":" << c.line () << ":" << c.column () << ":"
               << " info: composite value types cannot derive from object "
               << "or view types" << endl;

          cerr << b.file () << ":" << b.line () << ":" << b.column () << ":"
               << " info: class '" << name << "' is defined here" << endl;

          valid_ = false;
        }
      }

      // Check id.
      //
      semantics::data_member* id (0);
      semantics::data_member* optimistic (0);
      {
        special_members t (class_composite, valid_, id, optimistic);
        t.traverse (c);
      }

      if (id != 0)
      {
        cerr << id->file () << ":" << id->line () << ":" << id->column ()
             << ": error: value type data member cannot be designated as an "
             << "object id" << endl;

        valid_ = false;
      }

      if (optimistic != 0)
      {
        semantics::data_member& o (*optimistic);

        cerr << o.file () << ":" << o.line () << ":" << o.column ()
             << ": error: value type data member cannot be designated as a "
             << "version" << endl;

        valid_ = false;
      }

      // Check members.
      //
      member_.count_ = 0;
      names (c);

      if (member_.count_ == 0 && !base)
      {
        cerr << c.file () << ":" << c.line () << ":" << c.column () << ":"
             << " error: no persistent data members in the class" << endl;

        valid_ = false;
      }
    }

    bool& valid_;
    options const& options_;
    semantics::unit& unit_;
    value_type& vt_;

    data_member member_;
    traversal::names names_;
  };

  struct typedefs1: typedefs
  {
    typedefs1 (traversal::declares& d)
        : typedefs (true), declares_ (d)
    {
    }

    void
    traverse (semantics::typedefs& t)
    {
      if (check (t))
        traversal::typedefs::traverse (t);
      else
        declares_.traverse (t);
    }

  private:
    traversal::declares& declares_;
  };

  //
  // Pass 2.
  //

  struct object_no_id_members: object_members_base
  {
    object_no_id_members (bool& valid)
        : object_members_base (false, false, true), valid_ (valid), dm_ (0)
    {
    }

    virtual void
    traverse_simple (semantics::data_member& m)
    {
      if (m.count ("inverse"))
      {
        semantics::data_member& dm (dm_ != 0 ? *dm_ : m);

        cerr << dm.file () << ":" << dm.line () << ":" << dm.column () << ":"
             << " error: inverse object pointer member '" << member_prefix_
             << m.name () << "' in an object without an object id" << endl;

        valid_ = false;
      }
    }

    virtual void
    traverse_container (semantics::data_member& m, semantics::type&)
    {
      semantics::data_member& dm (dm_ != 0 ? *dm_ : m);

      cerr << dm.file () << ":" << dm.line () << ":" << dm.column () << ":"
           << " error: container member '" << member_prefix_ << m.name ()
           << "' in an object without an object id" << endl;

      valid_ = false;
    }

    virtual void
    traverse_composite (semantics::data_member* m, semantics::class_& c)
    {
      semantics::data_member* old_dm (dm_);

      if (dm_ == 0)
        dm_ = m;

      object_members_base::traverse_composite (m, c);

      dm_ = old_dm;
    }

  private:
    bool& valid_;
    semantics::data_member* dm_; // Direct view data member.
  };

  struct view_members: object_members_base
  {
    view_members (bool& valid)
        : object_members_base (false, false, true), valid_ (valid), dm_ (0)
    {
    }

    virtual void
    traverse_simple (semantics::data_member& m)
    {
      if (context::object_pointer (utype (m)))
      {
        semantics::data_member& dm (dm_ != 0 ? *dm_ : m);

        cerr << dm.file () << ":" << dm.line () << ":" << dm.column () << ":"
             << " error: view data member '" << member_prefix_ << m.name ()
             << "' is an object pointer" << endl;

        cerr << dm.file () << ":" << dm.line () << ":" << dm.column () << ":"
             << ": info: views cannot contain object pointers" << endl;

        valid_ = false;
      }
    }

    virtual void
    traverse_container (semantics::data_member& m, semantics::type&)
    {
      semantics::data_member& dm (dm_ != 0 ? *dm_ : m);

      cerr << dm.file () << ":" << dm.line () << ":" << dm.column () << ":"
           << " error: view data member '" << member_prefix_ << m.name ()
           << "' is a container" << endl;

      cerr << dm.file () << ":" << dm.line () << ":" << dm.column () << ":"
           << ": info: views cannot contain containers" << endl;

      valid_ = false;
    }

    virtual void
    traverse_composite (semantics::data_member* m, semantics::class_& c)
    {
      semantics::data_member* old_dm (dm_);

      if (dm_ == 0)
        dm_ = m;

      object_members_base::traverse_composite (m, c);

      dm_ = old_dm;
    }

  private:
    bool& valid_;
    semantics::data_member* dm_; // Direct view data member.
  };

  //
  //
  struct class2: traversal::class_
  {
    class2 (bool& valid, options const& ops, semantics::unit& unit)
        : valid_ (valid),
          options_ (ops),
          unit_ (unit),
          object_no_id_members_ (valid),
          view_members_ (valid)
    {
    }

    virtual void
    traverse (type& c)
    {
      if (context::object (c))
        traverse_object (c);
      else if (context::view (c))
        traverse_view (c);
      else if (context::composite (c))
        traverse_composite (c);
    }

    virtual void
    traverse_object (type& c)
    {
      if (context::id_member (c) == 0 && !context::abstract (c))
      {
        // Make sure we don't have any containers or inverse pointers.
        //
        object_no_id_members_.traverse (c);
      }
    }

    virtual void
    traverse_view (type& c)
    {
      // Make sure we don't have any containers or object pointers.
      //
      view_members_.traverse (c);
    }

    virtual void
    traverse_composite (type&)
    {
    }

    bool& valid_;
    options const& options_;
    semantics::unit& unit_;

    object_no_id_members object_no_id_members_;
    view_members view_members_;
  };
}

void validator::
validate (options const& ops,
          semantics::unit& u,
          semantics::path const&,
          unsigned short pass)
{
  auto_ptr<context> ctx (create_context (cerr, u, ops, 0));

  bool valid (true);

  if (pass == 1)
  {
    traversal::unit unit;
    traversal::defines unit_defines;
    traversal::declares unit_declares;
    typedefs1 unit_typedefs (unit_declares);
    traversal::namespace_ ns;
    value_type vt (valid);
    class1 c (valid, ops, u, vt);

    unit >> unit_defines >> ns;
    unit_defines >> c;
    unit >> unit_declares >> vt;
    unit >> unit_typedefs >> c;

    traversal::defines ns_defines;
    traversal::declares ns_declares;
    typedefs1 ns_typedefs (ns_declares);

    ns >> ns_defines >> ns;
    ns_defines >> c;
    ns >> ns_declares >> vt;
    ns >> ns_typedefs >> c;

    unit.dispatch (u);
  }
  else
  {
    traversal::unit unit;
    traversal::defines unit_defines;
    typedefs unit_typedefs (true);
    traversal::namespace_ ns;
    class2 c (valid, ops, u);

    unit >> unit_defines >> ns;
    unit_defines >> c;
    unit >> unit_typedefs >> c;

    traversal::defines ns_defines;
    typedefs ns_typedefs (true);

    ns >> ns_defines >> ns;
    ns_defines >> c;
    ns >> ns_typedefs >> c;

    unit.dispatch (u);
  }

  if (!valid)
    throw failed ();
}