upkg: Add support for controlling which package is acted upon.

[upkg.git] / src / upkg.c

diff --git a/src/upkg.c b/src/upkg.c
index 19f1449a3e1aa53ab550438c43886d88909ad433..a82ab2c355764e3e683294c44a8c68bbd18ce82a 100644 (file)
--- a/src/upkg.c
+++ b/src/upkg.c
@@ -19,6 +19,8 @@
 #include <config.h>
 #include <stdio.h>
 #include <stdlib.h>
+#include <string.h>
+#include <errno.h>
 #include <getopt.h>
 
 #include <glib-object.h>
@@ -64,7 +66,7 @@ There is NO WARRANTY, to the extent permitted by law."
 
 void print_usage(FILE *f)
 {
-       fprintf(f, "Usage: %s [options]\n", progname);
+       fprintf(f, "Usage: %s [options] [object ...]\n", progname);
 }
 
 void print_help(void)
@@ -218,7 +220,7 @@ int package_info(struct upkg *pkg)
        if (verbose >= 1) print_upkg_exports(pkg);
        printf("Imports: %lu\n", pkg->import_count);
 
-       return EXIT_SUCCESS;
+       return 0;
 }
 
 static int export(struct upkg *pkg, GObject *obj, unsigned idx)
@@ -262,13 +264,13 @@ int package_export(struct upkg *pkg)
        const char *class, *package;
        GObject *obj;
        GType type;
-       int rc = EXIT_SUCCESS;
+       int ret = 0;
 
        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;
+                       return -1;
                }
 
                type = u_object_module_get_class(package, class);
@@ -277,21 +279,61 @@ int package_export(struct upkg *pkg)
                obj = g_object_new(type, NULL);
                if (U_OBJECT_IS_EXPORTABLE(obj)) {
                        if (export(pkg, obj, i) != 0) {
-                               rc = EXIT_FAILURE;
+                               ret = -1;
                        }
                }
                g_object_unref(obj);
        }
 
-       return rc;
+       return ret;
 }
 
-int main(int argc, char **argv)
+static int process_object(int mode, const char *objname)
 {
        GTypeModule *pkg;
+       int ret = 0;
+
+       pkg = u_pkg_open(objname);
+       if (!pkg || !g_type_module_use(pkg)) {
+               fprintf(stderr, "%s: %s: failed to open package.\n",
+                               progname, objname);
+               goto out;
+       }
+
+       if (!U_PKG(pkg)->pkg) {
+               if (u_pkg_is_native(pkg)) {
+                       fprintf(stderr, "%s: %s: not a UObject package.\n",
+                                       progname, objname);
+               } else {
+                       fprintf(stderr, "%s: %s: package not found.\n",
+                                       progname, objname);
+               }
+
+               goto out_unuse;
+       }
+
+       switch (mode) {
+       case MODE_INFO:
+               ret = package_info(U_PKG(pkg)->pkg);
+               break;
+       case MODE_EXPORT:
+               ret = package_export(U_PKG(pkg)->pkg);
+               break;
+       default:
+               abort();
+       }
+
+out_unuse:
+       g_type_module_unuse(pkg);
+out:
+       return ret;
+}
+
+int main(int argc, char **argv)
+{
        const char *pkgname = NULL;
        unsigned mode = MODE_INFO;
-       int opt, rc = EXIT_FAILURE;
+       int opt, rc;
 
        if (argc > 0) progname = argv[0];
 
@@ -330,40 +372,22 @@ int main(int argc, char **argv)
                }
        }
 
-       if (!pkgname) {
+       if (!pkgname && !argv[optind]) {
                print_usage(stderr);
                return EXIT_FAILURE;
        }
 
-       pkg = u_pkg_open(pkgname);
-       if (!pkg || !g_type_module_use(pkg)) {
-               fprintf(stderr, "%s: %s: failed to open package.\n",
-                               progname, pkgname);
-               return EXIT_FAILURE;
-       }
-
-       if (!U_PKG(pkg)->pkg) {
-               if (u_pkg_is_native(pkg)) {
-                       fprintf(stderr, "%s: %s: not a UObject package.\n",
-                                       progname, pkgname);
-               } else {
-                       fprintf(stderr, "%s: %s: package not found.\n",
-                                       progname, pkgname);
+       rc = EXIT_SUCCESS;
+       if (argv[optind]) {
+               for (int i = optind; i < argc; i++) {
+                       if (process_object(mode, argv[i]) != 0)
+                               rc = EXIT_FAILURE;
                }
-
-               return EXIT_FAILURE;
+       } else {
+               if (process_object(mode, pkgname) != 0)
+                       rc = EXIT_FAILURE;
        }
 
-       switch (mode) {
-       case MODE_INFO:
-               rc = package_info(U_PKG(pkg)->pkg);
-               break;
-       case MODE_EXPORT:
-               rc = package_export(U_PKG(pkg)->pkg);
-               break;
-       }
-
-       g_type_module_unuse(pkg);
        u_object_module_exit();
        u_pkg_vfs_exit();
        return rc;