jim-clock.c

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