Refactor main and add print mode

[lackey] / src / main.c

diff --git a/src/main.c b/src/main.c
index ce355394baec00373c7ec3ba6d89896f6f10b347..bf518ffdd7201ce9219be3ecc1868365788ad677 100644 (file)
--- a/src/main.c
+++ b/src/main.c
@@ -15,15 +15,17 @@
  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
 
+#include <stdio.h>
 #include <stdlib.h>
 #include <signal.h>
 
-#include "args.h"
 #include "util.h"
+#include "args.h"
 #include "conf.h"
 #include "date.h"
 #include "cal.h"
 #include "view.h"
+#include "print.h"
 
 /* Config parser */
 static void on_config(const char *group, const char *name, const char *key, const char *value)
@@ -36,7 +38,10 @@ static void on_config(const char *group, const char *name, const char *key, cons
 /* Control-C handler, so we don't hose the therminal */
 static void on_sigint(int signum)
 {
-       view_exit();
+       if (PRINT)
+               print_exit();
+       else
+               view_exit();
        exit(0);
 }
 
@@ -51,19 +56,25 @@ int main(int argc, char **argv)
        conf_setup(".lackeyrc", on_config);
 
        /* Initialize */
-       args_init();
        util_init();
-       conf_init();
        date_init();
        cal_init();
 
-       /* Run args main */
-       args_main();
+       /* Common main */
+       args_start();
+       conf_start();
 
-       /* Run view main */
-       view_init();
-       view_main();
-       view_exit();
+       /* Mode main */
+       if (PRINT) {
+               print_init();
+               print_main();
+               print_exit();
+       }
+       else {
+               view_init();
+               view_main();
+               view_exit();
+       }
 
        return 0;
 }