timetracker

commit d93042afe023d288f614cbd04b17c5fce214bcd1
parent 933fb5798ab838b539a9af4416430a624458cfa4
Author: tongong <tongong@gmx.net>
Date:   Tue,  8 Jun 2021 16:53:10 +0200

added timetracker-save

Diffstat:
A
timetracker-save.c
 | 
148
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

1 file changed, 148 insertions(+), 0 deletions(-)
diff --git a/timetracker-save.c b/timetracker-save.c
@@ -0,0 +1,148 @@
+#include <signal.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/stat.h>
+#include <sys/types.h>
+#include <time.h>
+#include <unistd.h>
+
+#define MAX_PROGRAM_NAME  30   /* longer program names get cut off */
+#define WRITE_INTERVAL    600  /* 10 minutes */
+
+#define FILENAME   "/timetracker/data" /* relative to $XDG_DATA_HOME */
+#define FILENAME2  "/.local/share/timetracker/data" /* relative to $HOME */
+
+/* TODO */
+/* exit handler -> should write to data file
+ *                 should not interfere with normal write */
+
+struct program {
+    char name[MAX_PROGRAM_NAME];
+    int time;
+};
+
+
+void exithandler(int dummy) {
+    printf("stop");
+}
+
+/* returns path to data file; should be free()d */
+char * filelocation() {
+
+    char *loc = NULL;
+    char *xdg_data_home = getenv("XDG_DATA_HOME");
+    if (xdg_data_home) {
+        loc = malloc(strlen(xdg_data_home) + strlen(FILENAME) + 1);
+        strcpy(loc, xdg_data_home);
+        strcpy(loc + strlen(xdg_data_home), FILENAME);
+    } else {
+        char *xdg_home = getenv("HOME");
+        loc = malloc(strlen(xdg_home) + strlen(FILENAME2) + 1);
+        strcpy(loc, xdg_home);
+        strcpy(loc + strlen(xdg_home), FILENAME2);
+    }
+
+    return loc;
+}
+
+/* creates .local/share/timetracker (or $XDG_DATA_HOME/timetracker) */
+/* assumes .local/share is already there */
+void createdatadir() {
+    char * loc = filelocation();
+    for (int i = strlen(loc) - 1; i >= 0; i--) {
+        if (loc[i] == '/') {
+            loc[i] = '\0';
+            break;
+        }
+    }
+    /* fails if directory exists already but that is no problem */
+    mkdir(loc, 0777);
+    free(loc);
+}
+
+
+int main() {
+    char lastline[MAX_PROGRAM_NAME];
+    time_t lasttime = 0;
+
+    time_t writetime = time(NULL) + WRITE_INTERVAL;
+
+    struct program *programs = NULL;
+    int programnum = 0; /* number of programs stored in the programs array */
+
+    createdatadir();
+
+    signal(SIGINT, exithandler);
+
+    /* ensures that no two timestamps in output file are identical */
+    sleep(1);
+
+    while (1) {
+        size_t size = 0; /* i do not need this value but getline does */
+        char *line = NULL; /* current line / program name */
+        if (getline(&line, &size, stdin) == -1) exit(1);
+        /* remove \n at the end of the line */
+        line[strlen(line) - 1] = '\0';
+        /* remove ":" and "," from program name because they would mess up
+         * the output file */
+        for (int i = 0; i < strlen(line); i++) {
+            if (line[i] == ':' || line[i] == ',') line[i] = '-';
+        }
+
+        time_t newtime = time(NULL);
+
+        if (lasttime) {
+            /* is the program already in the list? (and where?) */
+            int index = -1;
+            for (int i = 0; i < programnum; i++) {
+                if (!strcmp(lastline, programs[i].name)) {
+                    index = i;
+                }
+            }
+            if (index == -1) {
+                /* extend/create list if necessary */
+                if (programnum % 10 == 0) {
+                    programs = realloc(programs, sizeof(struct program) *
+                            (10 + programnum));
+                }
+                /* add new program to list */
+                strncpy(programs[programnum].name, lastline, MAX_PROGRAM_NAME);
+                programs[programnum].time = newtime - lasttime;
+                programnum++;
+            } else {
+                /* update time for existing program */
+                programs[index].time += newtime - lasttime;
+            }
+
+            if (writetime <= newtime) {
+                /* printf("writing to data file\n"); */
+                /* write to data file */
+                char * loc = filelocation();
+                FILE *dataFile = fopen(loc, "a");;
+                free(loc);
+
+                fprintf(dataFile, "%d,", newtime);
+                for (int i = 0; i < programnum - 1; i++) {
+                    fprintf(dataFile, "%s:%d,", programs[i].name, programs[i].time);
+                }
+                fprintf(dataFile, "%s:%d\n", programs[programnum - 1].name, programs[programnum - 1].time);
+                fclose(dataFile);
+
+                /* reset everything */
+                free(programs);
+                programnum = 0;
+                writetime = newtime + WRITE_INTERVAL;
+            }
+        }
+
+        /* for the next interation */
+        lasttime = newtime;
+        strncpy(lastline, line, MAX_PROGRAM_NAME);
+        lastline[MAX_PROGRAM_NAME - 1] = '\0';
+        free(line);
+    }
+
+    /* will never be reached so can be deleted? */
+    return 0;
+}