X-Git-Url: https://git.draconx.ca/gitweb/upkg.git/blobdiff_plain/6bce0a077c3ffc8f74b4402d56bfbe7cbd6cba04..a6f1f69cc4874e2f2f878289bfc0f677220dfda9:/src/upkg.c

diff --git a/src/upkg.c b/src/upkg.c
index 61f67a4..4a76fa2 100644
--- a/src/upkg.c
+++ b/src/upkg.c
@@ -28,11 +28,22 @@
 #include "module.h"
 #include "uobject.h"
 #include "exportable.h"
+#include "loadable.h"
 
-static const char *progname = "upkg";
+enum {
+	MODE_INFO,
+	MODE_EXPORT,
+	MODE_MAX
+};
+
+int verbose = 0;
 
-static const char *sopts = "VH";
+static const char *progname = "upkg";
+static const char *sopts = "ixvVH";
 static const struct option lopts[] = {
+	{ "info",     0, NULL, 'i' },
+	{ "export",   0, NULL, 'x' },
+	{ "verbose",  0, NULL, 'v' },
 	{ "version",  0, NULL, 'V' },
 	{ "help",     0, NULL, 'H' },
 	{ 0 }
@@ -48,14 +59,14 @@ There is NO WARRANTY, to the extent permitted by law."
 	);
 }
 
-void print_usage(void)
+void print_usage(FILE *f)
 {
-	printf("Usage: %s [options] package\n", progname);
+	fprintf(f, "Usage: %s [options] package\n", progname);
 }
 
 void print_help(void)
 {
-	print_usage();
+	print_usage(stdout);
 	puts("Detailed help coming soon.  Until then, I'll just list my options.");
 	for (unsigned i = 0; lopts[i].name; i++) {
 		const struct option *o = &lopts[i];
@@ -98,15 +109,138 @@ void print_guid(unsigned char guid[static 16])
 	}
 }
 
+void print_upkg_exports(struct upkg *pkg)
+{
+	for (unsigned i = 0; i < pkg->export_count; i++) {
+		const char *name, *package, *class;
+
+		name  = upkg_export_name(pkg, i);
+		class = upkg_export_class(pkg, i, &package);
+
+		if (!name || !class)
+			continue;
+
+		printf("%u - %s (%s.%s)\n", i+1, name, package, class);
+	}
+}
+
+int package_info(struct upkg *pkg)
+{
+	printf("Version: %u\n",  pkg->version);
+	printf("License: %u\n",  pkg->license);
+	printf("GUID: ");
+	print_guid(pkg->guid);
+
+	printf("Flags:   %lx\n", pkg->flags);
+	if (verbose >= 1) {
+		print_upkg_flags("\t", pkg->flags);
+	}
+
+	if (verbose >= 1) {
+		printf("Names:   %lu\n", pkg->name_count);
+		if (verbose >= 2) {
+			for (unsigned long i = 0; i < pkg->name_count; i++) {
+				printf("\t%s\n", upkg_get_name(pkg, i));
+			}
+		}
+	}
+
+	printf("Exports: %lu\n", pkg->export_count);
+	if (verbose >= 1) print_upkg_exports(pkg);
+	printf("Imports: %lu\n", pkg->import_count);
+
+	return EXIT_SUCCESS;
+}
+
+static int export(struct upkg *pkg, GObject *obj, unsigned idx)
+{
+	struct upkg_file *f = upkg_export_open(pkg, idx);
+	char name[256];
+	FILE *of;
+	int rc = -1;
+
+	if (u_object_deserialize(obj, f) != 0) {
+		goto out;
+	}
+
+	if (U_OBJECT_IS_LOADABLE(obj) && u_object_load(obj) != 0) {
+		goto out;
+	}
+
+	u_object_export_name(obj, name, sizeof name);
+
+	printf("exporting %s to %s\n", upkg_export_name(pkg, idx), name);
+	of = fopen(name, "wb");
+	if (!of) {
+		perror(name);
+		goto out;
+	}
+
+	rc = u_object_export(obj, of);
+
+	if (fclose(of) != 0) {
+		perror(name);
+		rc = -1;
+		goto out;
+	}
+
+	if (U_OBJECT_IS_LOADABLE(obj)) {
+		u_object_unload(obj);
+	}
+
+out:
+	upkg_export_close(f);
+	return rc;
+}
+
+int package_export(struct upkg *pkg)
+{
+	const char *class, *package;
+	GObject *obj;
+	GType type;
+	int rc = EXIT_SUCCESS;
+
+	for (unsigned i = 0; i < pkg->export_count; i++) {
+		class = upkg_export_class(pkg, i, &package);
+		if (!class) {
+			fprintf(stderr, "error getting class information.\n");
+			return EXIT_FAILURE;
+		}
+
+		type = module_get_class(package, class);
+		if (!type) continue;
+
+		obj = g_object_new(type, NULL);
+		if (U_OBJECT_IS_EXPORTABLE(obj)) {
+			if (export(pkg, obj, i) != 0) {
+				rc = EXIT_FAILURE;
+			}
+		}
+		g_object_unref(obj);
+	}
+
+	return rc;
+}
+
 int main(int argc, char **argv)
 {
 	struct upkg *pkg;
-	int opt;
+	unsigned mode = MODE_INFO;
+	int opt, rc = EXIT_FAILURE;
 
 	if (argc > 0) progname = argv[0];
 
 	while ((opt = getopt_long(argc, argv, sopts, lopts, NULL)) != -1) {
 		switch (opt) {
+		case 'i':
+			mode = MODE_INFO;
+			break;
+		case 'x':
+			mode = MODE_EXPORT;
+			break;
+		case 'v':
+			verbose++;
+			break;
 		case 'V':
 			print_version();
 			return EXIT_SUCCESS;
@@ -114,57 +248,35 @@ int main(int argc, char **argv)
 			print_help();
 			return EXIT_SUCCESS;
 		default:
-			print_usage();
+			print_usage(stderr);
 			return EXIT_FAILURE;
 		}
 	}
 
-	if (argc < 2) {
-		fprintf(stderr, "usage: upkg file\n");
+	if (argv[optind] == NULL) {
+		print_usage(stderr);
 		return EXIT_FAILURE;
 	}
 
 	if (module_init() != 0)
 		return EXIT_FAILURE;
 
-	pkg = upkg_fopen(argv[1]);
+	pkg = upkg_fopen(argv[optind]);
 	if (!pkg) {
 		fprintf(stderr, "failed to open package!\n");
 		return EXIT_FAILURE;
 	}
 
-	printf("Version: %u\n",  pkg->version);
-	printf("License: %u\n",  pkg->license);
-	printf("GUID: ");
-	print_guid(pkg->guid);
-
-	printf("Flags:   %lx\n", pkg->flags);
-	print_upkg_flags("\t", pkg->flags);
-
-	printf("Names:   %lu\n", pkg->name_count);
-	for (unsigned long i = 0; i < pkg->name_count; i++) {
-		printf("\t%s\n", upkg_get_name(pkg, i));
+	switch (mode) {
+	case MODE_INFO:
+		rc = package_info(pkg);
+		break;
+	case MODE_EXPORT:
+		rc = package_export(pkg);
+		break;
 	}
 
-	printf("Exports: %lu\n", pkg->export_count);
-	printf("Imports: %lu\n", pkg->import_count);
-
-	GObject *music = g_object_new(module_get_class("Engine", "Music"), NULL);
-	if (!music)
-		return EXIT_FAILURE;
-	struct upkg_file *f = upkg_export_open(pkg, 0);
-	u_object_deserialize(music, f);
-
-	char name[256];
-	u_object_export_name(music, name, sizeof name);
-	FILE *fp = fopen(name, "wb");
-	if (!fp) return EXIT_FAILURE;
-	u_object_export(music, fp);
-	fclose(fp);
-	printf("Wrote %s\n", name);
-
 	upkg_close(pkg);
 	module_exit();
-
-	return 0;
+	return rc;
 }