use environment variable if no path given

diff --git a/zpm-init.c b/zpm-init.c
index b63007f4f2227941ff641ee1db79af0826776a7c..9a0f839f0ee11c8703e623805d754ff858d71a68 100644 (file)
--- a/zpm-init.c
+++ b/zpm-init.c
@@ -5,12 +5,19 @@
 int main(int ac, char **av){
        struct zpm pkg;
        int rv;
+       char *path;
 
-       if (ac < 2) {
-               fprintf(stderr, "usage: path\n");
+       path = getenv("ZPMDB");
+
+       if (ac > 1) {
+               path = av[1];
+       }
+
+       if (!path) {
+               fprintf(stderr, "usage: zpm-init [path]\n");
                return 1;
        }
-       rv = zpm_init(&pkg, av[1]);
+       rv = zpm_init(&pkg, path);
        if (rv) {
                zpm_close(&pkg);
        }