LIBGUESTFS_PATH implementation.
[libguestfs.git] / fish / fish.c
index a252516..0fc14ce 100644 (file)
@@ -39,24 +39,26 @@ static void interactive (void);
 static void shell_script (void);
 static void script (int prompt);
 static void cmdline (char *argv[], int optind, int argc);
-static void issue_command (const char *cmd, char *argv[]);
+static int issue_command (const char *cmd, char *argv[]);
 
 /* Currently open libguestfs handle. */
 guestfs_h *g;
 int g_launched = 0;
 
 int quit = 0;
+int verbose = 0;
 
-void
+int
 launch (void)
 {
   if (!g_launched) {
     if (guestfs_launch (g) == -1)
-      exit (1);
+      return -1;
     if (guestfs_wait_ready (g) == -1)
-      exit (1);
+      return -1;
     g_launched = 1;
   }
+  return 0;
 }
 
 static void
@@ -67,7 +69,7 @@ usage (void)
           "guestfish lets you edit virtual machine filesystems\n"
           "Copyright (C) 2009 Red Hat Inc.\n"
           "Usage:\n"
-          "  guestfish [--options] cmd [: cmd ...]\n"
+          "  guestfish [--options] cmd [: cmd : cmd ...]\n"
           "or for interactive use:\n"
           "  guestfish\n"
           "or from a shell script:\n"
@@ -76,13 +78,14 @@ usage (void)
           "  ...\n"
           "  EOF\n"
           "Options:\n"
-          "  -h|--cmd-help       List available commands\n"
-          "  -h|--cmd-help cmd   Display detailed help on 'cmd'\n"
-          "  -a image            Add image\n"
-          "  -m dev[:mnt]        Mount dev on mnt (if omitted, /)\n"
-          /*"  --ro|-r             All mounts are read-only\n"*/
-          "  -v|--verbose        Verbose messages\n"
-          "For more information, see the manpage guestfish(1).\n");
+          "  -h|--cmd-help        List available commands\n"
+          "  -h|--cmd-help cmd    Display detailed help on 'cmd'\n"
+          "  -a|--add image       Add image\n"
+          "  -m|--mount dev[:mnt] Mount dev on mnt (if omitted, /)\n"
+          "  -n|--no-sync         Don't autosync\n"
+        /*"  --ro|-r              All mounts are read-only\n"*/
+          "  -v|--verbose         Verbose messages\n"
+          "For more information,  see the manpage guestfish(1).\n");
 }
 
 int
@@ -94,6 +97,7 @@ main (int argc, char *argv[])
     { "cmd-help", 2, 0, 'h' },
     { "help", 0, 0, '?' },
     { "mount", 1, 0, 'm' },
+    { "no-sync", 0, 0, 'n' },
     { "verbose", 0, 0, 'v' },
     { 0, 0, 0, 0 }
   };
@@ -111,6 +115,16 @@ main (int argc, char *argv[])
     exit (1);
   }
 
+  guestfs_set_autosync (g, 1);
+
+  /* If developing, add . to the path.  Note that libtools interferes
+   * with this because uninstalled guestfish is a shell script that runs
+   * the real program with an absolute path.  Detect that too.
+   */
+  if (argv[0] &&
+      (argv[0][0] != '/' || strstr (argv[0], "/.libs/lt-") != NULL))
+    guestfs_set_path (g, ".:" GUESTFS_DEFAULT_PATH);
+
   for (;;) {
     c = getopt_long (argc, argv, options, long_options, NULL);
     if (c == -1) break;
@@ -151,6 +165,15 @@ main (int argc, char *argv[])
       mps = mp->next;
       break;
 
+    case 'n':
+      guestfs_set_autosync (g, 0);
+      break;
+
+    case 'v':
+      verbose++;
+      guestfs_set_verbose (g, verbose);
+      break;
+
     case '?':
       usage ();
       exit (0);
@@ -163,7 +186,7 @@ main (int argc, char *argv[])
 
   /* If we've got mountpoints, we must launch the guest and mount them. */
   if (mps != NULL) {
-    launch ();
+    if (launch () == -1) exit (1);
     mount_mps (mps);
   }
 
@@ -232,7 +255,8 @@ script (int prompt)
   int len, i;
 
   if (prompt)
-    printf ("Welcome to guestfish, the libguestfs filesystem interactive shell for\n"
+    printf ("\n"
+           "Welcome to guestfish, the libguestfs filesystem interactive shell for\n"
            "editing virtual machine filesystems.\n"
            "\n"
            "Type: 'help' for help with commands\n"
@@ -263,7 +287,7 @@ script (int prompt)
       exit (1);
     }
 
-    issue_command (cmd, argv);
+    (void) issue_command (cmd, argv);
   }
   if (prompt) printf ("\n");
 }
@@ -287,27 +311,99 @@ cmdline (char *argv[], int optind, int argc)
   while (optind < argc && strcmp (argv[optind], ":") != 0)
     optind++;
 
-  if (optind == argc)
-    issue_command (cmd, params);
-  else {
+  if (optind == argc) {
+    if (issue_command (cmd, params) == -1) exit (1);
+  else {
     argv[optind] = NULL;
-    issue_command (cmd, params);
+    if (issue_command (cmd, params) == -1) exit (1);
     cmdline (argv, optind+1, argc);
   }
 }
 
-static void
+static int
 issue_command (const char *cmd, char *argv[])
 {
-  int i;
-
-  fprintf (stderr, "cmd = %s", cmd);
-  for (i = 0; argv[i] != NULL; ++i)
-    fprintf (stderr, ", arg[%d]=%s", i, argv[i]);
-  fprintf (stderr, "\n");
-
+  int argc;
 
+  for (argc = 0; argv[argc] != NULL; ++argc)
+    ;
 
+  if (strcasecmp (cmd, "help") == 0) {
+    if (argc == 0)
+      list_commands ();
+    else
+      display_command (argv[0]);
+    return 0;
+  }
+  else if (strcasecmp (cmd, "quit") == 0 ||
+          strcasecmp (cmd, "exit") == 0 ||
+          strcasecmp (cmd, "q") == 0)
+    exit (0);
+  else if (strcasecmp (cmd, "add") == 0 ||
+          strcasecmp (cmd, "drive") == 0 ||
+          strcasecmp (cmd, "add_drive") == 0) {
+    if (argc != 1) {
+      fprintf (stderr, "use 'add image' to add a guest image\n");
+      return -1;
+    }
+    else
+      return guestfs_add_drive (g, argv[0]);
+  }
+  else if (strcasecmp (cmd, "cdrom") == 0) {
+    if (argc != 1) {
+      fprintf (stderr, "use 'cdrom image' to add a CD-ROM image\n");
+      return -1;
+    }
+    else
+      return guestfs_add_cdrom (g, argv[0]);
+  }
+  else if (strcasecmp (cmd, "launch") == 0) {
+    if (argc != 0) {
+      fprintf (stderr, "'launch' command takes no parameters\n");
+      return -1;
+    }
+    else
+      return launch ();
+  }
+  else
+    return run_action (cmd, argc, argv);
+}
 
+void
+list_builtin_commands (void)
+{
+  printf ("%-20s %s\n",
+         "help", "display a list of commands or help on a command");
+  printf ("%-20s %s\n",
+         "quit", "quit guestfish");
+  printf ("%-20s %s\n",
+         "add", "add a guest image to be examined or modified");
+  printf ("%-20s %s\n",
+         "cdrom", "add a CD-ROM image to be examined");
+  printf ("%-20s %s\n",
+         "launch", "launch the subprocess");
+}
 
+void
+display_builtin_command (const char *cmd)
+{
+  if (strcasecmp (cmd, "add") == 0)
+    printf ("add - add a guest image to be examined or modified\n"
+           "     add <image>\n");
+  else if (strcasecmp (cmd, "cdrom") == 0)
+    printf ("cdrom - add a CD-ROM image to be examined\n"
+           "     cdrom <iso-file>\n");
+  else if (strcasecmp (cmd, "help") == 0)
+    printf ("help - display a list of commands or help on a command\n"
+           "     help cmd\n"
+           "     help\n");
+  else if (strcasecmp (cmd, "quit") == 0)
+    printf ("quit - quit guestfish\n"
+           "     quit\n");
+  else if (strcasecmp (cmd, "launch") == 0)
+    printf ("launch - launch the subprocess\n"
+           "     launch\n");
+  else
+    fprintf (stderr, "%s: command not known, use -h to list all commands\n",
+            cmd);
 }