X-Git-Url: https://git.draconx.ca/gitweb/upkg.git/blobdiff_plain/9d3cb178db367962f68521985ef9e239e1d03fa9..5fe7c42921be9f911805b4006f59a0910c9404d0:/src/module.c

diff --git a/src/module.c b/src/module.c
index 5edc5e2..2eeffc0 100644
--- a/src/module.c
+++ b/src/module.c
@@ -1,25 +1,61 @@
+/*
+ *  upkg: tool for manipulating Unreal Tournament packages.
+ *  Copyright (C) 2009 Nick Bowler
+ *
+ *  This program is free software; you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation; either version 2 of the License, or
+ *  (at your option) any later version.
+ *
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with this program; if not, write to the Free Software
+ *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ */
+
 #include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <ctype.h>
 #include <glib-object.h>
 #include <ltdl.h>
 
-#include "module.h"
+#include <uobject/module.h>
+#include <uobject/package.h>
+#include "avl.h"
 
 static unsigned initialized;
+static struct avl_table *package_tree;
 
-static void dl_print_errors(const char *prefix)
+static char *str_cpy_lower(char *dst, const char *src)
 {
-	const char *err;
-	while (err = lt_dlerror()) {
-		if (prefix) fprintf(stderr, "%s: ", prefix);
-		puts(lt_dlerror());
-	}
+	size_t i;
+
+	for (i = 0; src[i]; i++)
+		dst[i] = tolower(src[i]);
+	dst[i] = 0;
+
+	return dst;
 }
 
-int module_init(void)
+static int modcmp(const void *a, const void *b, void *_data)
+{
+	const GTypeModule *ma = a;
+	const GTypeModule *mb = b;
+
+	return strcmp(ma->name, mb->name);
+}
+
+int u_object_module_init(void)
 {
 	if (!initialized) {
-		if (lt_dlinit() != 0) {
-			dl_print_errors(__func__);
+		package_tree = avl_create(modcmp, NULL, NULL);
+		if (!package_tree) {
+			fprintf(stderr, "%s: failed to create package tree.\n", __func__);
 			return -1;
 		}
 
@@ -30,13 +66,42 @@ int module_init(void)
 	return 0;
 }
 
-int module_exit(void)
+int u_object_module_exit(void)
 {
 	if (--initialized)
 		return 0;
 
-	if (lt_dlexit() != 0) {
-		dl_print_errors(__func__);
-		return -1;
+	avl_destroy(package_tree, NULL);
+
+	return 0;
+}
+
+GType u_object_module_get_class(const char *package, const char *class)
+{
+	char buf[strlen(package) + strlen(class) + 1];
+	GTypeModule search = { .name = str_cpy_lower(buf, package) };
+
+	GTypeModule *mod = avl_find(package_tree, &search);
+	if (!mod) {
+		void **p;
+
+		mod = u_pkg_new(package);
+		if (!mod) {
+			return 0;
+		}
+
+		p = avl_probe(package_tree, mod);
+		if (!p) {
+			g_object_unref(mod);
+			return 0;
+		}
 	}
+
+	if (!g_type_module_use(mod))
+		return 0;
+
+	str_cpy_lower(buf+strlen(package), class);
+	buf[0] = toupper(buf[0]);
+	buf[strlen(package)] = toupper(buf[strlen(package)]);
+	return g_type_from_name(buf);
 }