1 files changed, 155 insertions, 0 deletions
diff --git a/odb/mysql/header.cxx b/odb/mysql/header.cxx
new file mode 100644
index 0000000..a4111ec
--- /dev/null
+++ b/odb/mysql/header.cxx
@@ -0,0 +1,155 @@
+// file      : odb/mysql/header.cxx
+// author    : Boris Kolpackov <boris@codesynthesis.com>
+// copyright : Copyright (c) 2009-2010 Code Synthesis Tools CC
+// license   : GNU GPL v2; see accompanying LICENSE file
+
+#include <odb/mysql/common.hxx>
+#include <odb/mysql/header.hxx>
+
+namespace
+{
+  struct class_: traversal::class_, context
+  {
+    class_ (context& c)
+        : context (c)
+    {
+    }
+
+    virtual void
+    traverse (type& c)
+    {
+      if (c.file () != unit.file ())
+        return;
+
+      if (!c.count ("object"))
+        return;
+
+      string const& type (c.fq_name ());
+
+      // Find the id member and type.
+      //
+      id_member t (*this);
+      t.traverse (c);
+
+      if (t.member () == 0)
+      {
+        cerr << c.file () << ":" << c.line () << ":" << c.column ()
+             << " error: no data member designated as object id" << endl;
+
+        cerr << c.file () << ":" << c.line () << ":" << c.column ()
+             << " info: use '#pragma odb id' to specify object id member"
+             << endl;
+
+        throw generation_failed ();
+      }
+
+      semantics::data_member& id (*t.member ());
+      semantics::type& id_type (id.type ());
+
+      if (id_type.anonymous ())
+      {
+        // Can be a template-id (which we should handle eventually) or an
+        // anonymous type in member declaration (e.g., struct {...} m_;).
+        //
+        cerr << id.file () << ":" << id.line () << ":" << id.column ()
+             << " error: unnamed type in data member declaration" << endl;
+
+        cerr << id.file () << ":" << id.line () << ":" << id.column ()
+             << " info: use 'typedef' to name this type"
+             << endl;
+
+        throw generation_failed ();
+      }
+
+      os << "// " << c.name () << endl
+         << "//" << endl;
+
+      os << "template <>" << endl
+         << "class access::object_traits< " << type << " >: " << endl
+         << "  public access::object_memory< " << type << " >," << endl
+         << "  public access::object_factory< " << type << " >"
+         << "{"
+         << "public:" << endl;
+
+      // object_type & shared_ptr
+      //
+      os << "typedef " << type << " object_type;";
+
+      // id_type
+      //
+      os << "typedef " << id_type.fq_name () << " id_type;"
+         << endl;
+
+      // id_source
+      //
+      os << "static const odb::id_source id_source = odb::ids_assigned;"
+         << endl;
+
+      // id ()
+      //
+      os << "static id_type" << endl
+         << "id (const object_type&);"
+         << endl;
+
+      // persist ()
+      //
+      os << "static void" << endl
+         << "persist (database&, object_type&);"
+         << endl;
+
+      // store ()
+      //
+      os << "static void" << endl
+         << "store (database&, object_type&);"
+         << endl;
+
+      // erase ()
+      //
+      os << "static void" << endl
+         << "erase (database&, const id_type&);"
+         << endl;
+
+      // find ()
+      //
+      os << "static pointer_type" << endl
+         << "find (database&, const id_type&);"
+         << endl;
+
+      os << "static bool" << endl
+         << "find (database&, const id_type&, object_type&);";
+
+      os << "};";
+    }
+  };
+}
+
+namespace mysql
+{
+  void
+  generate_header (context& ctx)
+  {
+    traversal::unit unit;
+    traversal::defines unit_defines;
+    traversal::namespace_ ns;
+    class_ c (ctx);
+
+    unit >> unit_defines >> ns;
+    unit_defines >> c;
+
+    traversal::defines ns_defines;
+
+    ns >> ns_defines >> ns;
+    ns_defines >> c;
+
+    ctx.os << "#include <odb/core.hxx>" << endl
+           << "#include <odb/traits.hxx>" << endl
+           << endl;
+
+    ctx.os << "namespace odb"
+           << "{";
+
+    unit.dispatch (ctx.unit);
+
+    ctx.os << "}";
+  }
+}