jim-clock.c

   1 /*
   2  * jim-clock.c
   3  *
   4  * Implements the clock command
   5  */
   6
   7 /* For strptime() */
   8 #ifndef _XOPEN_SOURCE
   9 #define _XOPEN_SOURCE 500
  10 #endif
  11
  12 #include <stdlib.h>
  13 #include <string.h>
  14 #include <stdio.h>
  15 #include <time.h>
  16
  17 #include "jimautoconf.h"
  18 #include <jim-subcmd.h>
  19
  20 #ifdef HAVE_SYS_TIME_H
  21 #include <sys/time.h>
  22 #endif
  23
  24 static int clock_cmd_format(Jim_Interp *interp, int argc, Jim_Obj *const *argv)
  25 {
  26     /* How big is big enough? */
  27     char buf[100];
  28     time_t t;
  29     long seconds;
  30
  31     const char *format = "%a %b %d %H:%M:%S %Z %Y";
  32
  33     if (argc == 2 || (argc == 3 && !Jim_CompareStringImmediate(interp, argv[1], "-format"))) {
  34         return -1;
  35     }
  36
  37     if (argc == 3) {
  38         format = Jim_String(argv[2]);
  39     }
  40
  41     if (Jim_GetLong(interp, argv[0], &seconds) != JIM_OK) {
  42         return JIM_ERR;
  43     }
  44     t = seconds;
  45
  46     if (strftime(buf, sizeof(buf), format, localtime(&t)) == 0) {
  47         Jim_SetResultString(interp, "format string too long", -1);
  48         return JIM_ERR;
  49     }
  50
  51     Jim_SetResultString(interp, buf, -1);
  52
  53     return JIM_OK;
  54 }
  55
  56 #ifdef HAVE_STRPTIME
  57 static int clock_cmd_scan(Jim_Interp *interp, int argc, Jim_Obj *const *argv)
  58 {
  59     char *pt;
  60     struct tm tm;
  61     time_t now = time(0);
  62
  63     if (!Jim_CompareStringImmediate(interp, argv[1], "-format")) {
  64         return -1;
  65     }
  66
  67     /* Initialise with the current date/time */
  68     localtime_r(&now, &tm);
  69
  70     pt = strptime(Jim_String(argv[0]), Jim_String(argv[2]), &tm);
  71     if (pt == 0 || *pt != 0) {
  72         Jim_SetResultString(interp, "Failed to parse time according to format", -1);
  73         return JIM_ERR;
  74     }
  75
  76     /* Now convert into a time_t */
  77     Jim_SetResultInt(interp, mktime(&tm));
  78
  79     return JIM_OK;
  80 }
  81 #endif
  82
  83 static int clock_cmd_seconds(Jim_Interp *interp, int argc, Jim_Obj *const *argv)
  84 {
  85     Jim_SetResultInt(interp, time(NULL));
  86
  87     return JIM_OK;
  88 }
  89
  90 static int clock_cmd_micros(Jim_Interp *interp, int argc, Jim_Obj *const *argv)
  91 {
  92     struct timeval tv;
  93
  94     gettimeofday(&tv, NULL);
  95
  96     Jim_SetResultInt(interp, (jim_wide) tv.tv_sec * 1000000 + tv.tv_usec);
  97
  98     return JIM_OK;
  99 }
 100
 101 static int clock_cmd_millis(Jim_Interp *interp, int argc, Jim_Obj *const *argv)
 102 {
 103     struct timeval tv;
 104
 105     gettimeofday(&tv, NULL);
 106
 107     Jim_SetResultInt(interp, (jim_wide) tv.tv_sec * 1000 + tv.tv_usec / 1000);
 108
 109     return JIM_OK;
 110 }
 111
 112 static const jim_subcmd_type clock_command_table[] = {
 113     {   "seconds",
 114         NULL,
 115         clock_cmd_seconds,
 116         0,
 117         0,
 118         /* Description: Returns the current time as seconds since the epoch */
 119     },
 120     {   "clicks",
 121         NULL,
 122         clock_cmd_micros,
 123         0,
 124         0,
 125         /* Description: Returns the current time in 'clicks' */
 126     },
 127     {   "microseconds",
 128         NULL,
 129         clock_cmd_micros,
 130         0,
 131         0,
 132         /* Description: Returns the current time in microseconds */
 133     },
 134     {   "milliseconds",
 135         NULL,
 136         clock_cmd_millis,
 137         0,
 138         0,
 139         /* Description: Returns the current time in milliseconds */
 140     },
 141     {   "format",
 142         "seconds ?-format format?",
 143         clock_cmd_format,
 144         1,
 145         3,
 146         /* Description: Format the given time */
 147     },
 148 #ifdef HAVE_STRPTIME
 149     {   "scan",
 150         "str -format format",
 151         clock_cmd_scan,
 152         3,
 153         3,
 154         /* Description: Determine the time according to the given format */
 155     },
 156 #endif
 157     { NULL }
 158 };
 159
 160 int Jim_clockInit(Jim_Interp *interp)
 161 {
 162     if (Jim_PackageProvide(interp, "clock", "1.0", JIM_ERRMSG))
 163         return JIM_ERR;
 164
 165     Jim_CreateCommand(interp, "clock", Jim_SubCmdProc, (void *)clock_command_table, NULL);
 166     return JIM_OK;
 167 }