most of the commnad line stuff

Signed-off-by: Mattias Andrée <maandree@operamail.com>

1 files changed, 143 insertions, 5 deletions

diff --git a/src/scrotty.c b/src/scrotty.c
index 37e3c1b..8cb3dc7 100644
--- a/src/scrotty.c
+++ b/src/scrotty.c
@@ -15,6 +15,14 @@
  * You should have received a copy of the GNU General Public License
  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
+#ifndef PROGRAM_NAME
+# define PROGRAM_NAME  "scrotty"
+#endif
+#ifndef PROGRAM_VERSION
+# define PROGRAM_VERSION  "1.0"
+#endif
+
+
 #include <stdio.h>
 #include <unistd.h>
 #include <errno.h>
@@ -67,6 +75,12 @@ static const char* inttable[] =
  */
 static const char* execname;
 
+/**
+ * Arguments for `convert`,
+ * the output file should be printed into `convert_args[2]`.
+ */
+static char** convert_args;
+
 
 /**
  * Create an PNM-file that is sent to `convert` for convertion to a compressed format
@@ -178,7 +192,8 @@ static int save(const char* fbpath, const char* imgpath, int fbno, long width, l
 	  close(pipe_rw[0]);
 	}
       /* Exec. `convert` to convert the PNM-image we create to a compressed image. */
-      execlp("convert", "convert", DEVDIR "/stdin", imgpath, NULL);
+      sprintf(convert_args[2], "%s", imgpath);
+      execvp("convert", convert_args);
       perror(execname);
       exit(1);
     }
@@ -282,8 +297,96 @@ static int save_fb(int fbno)
 }
 
 
+#define p(...)  if (printf(__VA_ARGS__) < 0)  return -1;
+
+
+/**
+ * Print usage information
+ * 
+ * @return  Zero on success, -1 on error
+ */
+static int print_help(void)
+{
+  p("SYNOPSIS\n");
+  p("\t%s [options...] [filename-pattern] [-- options-for-convert...]\n", execname);
+  p("\n");
+  p("OPTIONS\n");
+  p("\t--help         Print usage information.\n");
+  p("\t--version      Print program name and version.\n");
+  p("\t--copyright    Print copyright information.\n");
+  p("\t--exec CMD     Command to run for each saved image.\n");
+  p("\n");
+  p("SPECIAL STRINGS\n");
+  p("\tBoth the --exec and filename-pattern parameters can take format specifiers\n");
+  p("\tthat are expanded by scrotty when encountered. There are two types of format\n");
+  p("\tspecifier. Characters preceded by a '%%' are interpretted by strftime(2).\n");
+  p("\tSee `man strftime` for examples. These options may be used to refer to the\n");
+  p("\tcurrent date and time. The second kind are internal to scrotty and are prefixed\n");
+  p("\tby '$'. The following specifiers are recognised:\n");
+  p("\n");
+  p("\t\n");
+  p("\t$i  framebuffer index\n");
+  p("\t$f  image filename/pathname (ignored when used in filename-pattern)\n");
+  p("\t$n  image filename          (ignored when used in filename-pattern)\n");
+  p("\t$p  image width multiplied by image height\n");
+  p("\t$w  image width\n");
+  p("\t$h  image height\n");
+  p("\t$$  expands to a literal '$'\n");
+  p("\t\\n  expands to a new line\n");
+  p("\t\\\\  expands to a literal '\\'\n");
+  p("\t\\   expands to a literal ' ' (the string is a backslash followed by a space)\n");
+  p("\n");
+  p("\tA space that is not prefixed by a backslash in --exec is interpreted as an\n");
+  p("\targument delimiter. This is the case even at the beginning and end of the\n");
+  p("\tstring and if the a space was the previous character in the string.\n");
+  p("\n");
+  return 0;
+}
+
+
+/**
+ * Print program name and version
+ * 
+ * @return  Zero on success, -1 on error
+ */
+static int print_version(void)
+{
+  p("%s %s\n", PROGRAM_NAME, PROGRAM_VERSION);
+  return 0;
+}
+
+
+/**
+ * Print copyright information
+ * 
+ * @return  Zero on success, -1 on error
+ */
+static int print_copyright(void)
+{
+  p("scrotty -- Screenshot program for Linux's TTY\n");
+  p("Copyright (C) 2014  Mattias Andrée (maandree@member.fsf.org)\n");
+  p("\n");
+  p("This program is free software: you can redistribute it and/or modify\n");
+  p("it under the terms of the GNU General Public License as published by\n");
+  p("the Free Software Foundation, either version 3 of the License, or\n");
+  p("(at your option) any later version.\n");
+  p("\n");
+  p("This program is distributed in the hope that it will be useful,\n");
+  p("but WITHOUT ANY WARRANTY; without even the implied warranty of\n");
+  p("MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\n");
+  p("GNU General Public License for more details.\n");
+  p("\n");
+  p("You should have received a copy of the GNU General Public License\n");
+  p("along with this program.  If not, see <http://www.gnu.org/licenses/>.\n");
+  return 0;
+}
+
+
+#undef p
+
+
 /**
- * Take a screenshow of all framebuffers
+ * Take a screenshot of all framebuffers
  * 
  * @param   argc  The number of elements in `argv`
  * @param   argv  Command line arguments
@@ -291,12 +394,47 @@ static int save_fb(int fbno)
  */
 int main(int argc, char* argv[])
 {
-  int fbno, r;
-  
-  (void) argc;
+  int fbno, r, i, dash = argc, exec = -1, help = 0, version = 0, copyright = 0, filepattern = -1;
+  static char convert_args_2[PATH_MAX];
   
   execname = *argv;
   
+  /* Parse command line. */
+  for (i = 1; i < argc; i++)
+    {
+      if      (!strcmp(argv[i], "--help"))       help = 1;
+      else if (!strcmp(argv[i], "--version"))    version = 1;
+      else if (!strcmp(argv[i], "--copyright"))  copyright = 1;
+      else if (!strcmp(argv[i], "--exec"))       exec = ++i; /* TODO use this */
+      else if (!strcmp(argv[i], "--"))
+	{
+	  dash = i + 1;
+	  break;
+	}
+      else if ((argv[i][0] == '-') || (filepattern != -1))
+	return fprintf(stderr, "Unrecognised option. Type `%s --help` for help.\n", execname), 1;
+      else
+	filepattern = i; /* TODO use this */
+    }
+  
+  /* Check that --exec is valid. */
+  if (exec == argc)
+    return fprintf(stderr, "--exec has no argument. Type `%s --help` for help.\n", execname), 1;
+  
+  /* Was --help, --version or --copyright used? */
+  if (help)       return -(print_help());
+  if (version)    return -(print_version());
+  if (copyright)  return -(print_copyright());
+  
+  /* Create arguments for `convert`. */
+  convert_args = alloca((4 + (argc - dash)) * sizeof(char*));
+  convert_args[0] = "convert";
+  convert_args[1] = DEVDIR "/stdin";
+  convert_args[2] = convert_args_2;
+  for (i = dash; i < argc; i++)
+    convert_args[i - dash + 2] = argv[i];
+  convert_args[3 + (argc - dash)] = NULL;
+  
   /* Take a screenshot of each framebuffer. */
   for (fbno = 0;; fbno++)
     {