source: util.c @ e3cbd65

release-1.5release-1.6release-1.7release-1.8release-1.9
Last change on this file since e3cbd65 was 1b9d3cc, checked in by Geoffrey Thomas <geofft@mit.edu>, 12 years ago
atoi("puce") == 0 Signed-off-by: Geoffrey Thomas <geofft@mit.edu>
  • Property mode set to 100644
File size: 17.3 KB
Line 
1#include "owl.h"
2#include <stdlib.h>
3#include <string.h>
4#include <unistd.h>
5#include <ctype.h>
6#include <pwd.h>
7#include <sys/stat.h>
8#include <sys/types.h>
9
10void sepbar(const char *in)
11{
12  char buff[1024];
13  WINDOW *sepwin;
14  const owl_messagelist *ml;
15  const owl_view *v;
16  int x, y, i;
17  const char *foo, *appendtosepbar;
18
19  sepwin=owl_global_get_curs_sepwin(&g);
20  ml=owl_global_get_msglist(&g);
21  v=owl_global_get_current_view(&g);
22
23  werase(sepwin);
24  wattron(sepwin, A_REVERSE);
25  if (owl_global_is_fancylines(&g)) {
26    whline(sepwin, ACS_HLINE, owl_global_get_cols(&g));
27  } else {
28    whline(sepwin, '-', owl_global_get_cols(&g));
29  }
30
31  if (owl_global_is_sepbar_disable(&g)) {
32    getyx(sepwin, y, x);
33    wmove(sepwin, y, owl_global_get_cols(&g)-1);
34    return;
35  }
36
37  wmove(sepwin, 0, 2); 
38
39  if (owl_messagelist_get_size(ml)==0) {
40    strcpy(buff, " (-/-) ");
41  } else {
42    snprintf(buff, 1024, " (%i/%i/%i) ", owl_global_get_curmsg(&g)+1,
43            owl_view_get_size(v),
44            owl_messagelist_get_size(ml));
45  }
46  waddstr(sepwin, buff);
47
48  foo=owl_view_get_filtname(v);
49  if (strcmp(foo, owl_global_get_view_home(&g))) wattroff(sepwin, A_REVERSE);
50  waddstr(sepwin, " ");
51  waddstr(sepwin, owl_view_get_filtname(v));
52  waddstr(sepwin, " ");
53  if (strcmp(foo, owl_global_get_view_home(&g))) wattron(sepwin, A_REVERSE);
54
55  if (owl_mainwin_is_curmsg_truncated(owl_global_get_mainwin(&g))) {
56    getyx(sepwin, y, x);
57    wmove(sepwin, y, x+2);
58    wattron(sepwin, A_BOLD);
59    waddstr(sepwin, " <truncated> ");
60    wattroff(sepwin, A_BOLD);
61  }
62
63  i=owl_mainwin_get_last_msg(owl_global_get_mainwin(&g));
64  if ((i != -1) &&
65      (i < owl_view_get_size(v)-1)) {
66    getyx(sepwin, y, x);
67    wmove(sepwin, y, x+2);
68    wattron(sepwin, A_BOLD);
69    waddstr(sepwin, " <more> ");
70    wattroff(sepwin, A_BOLD);
71  }
72
73  if (owl_global_get_rightshift(&g)>0) {
74    getyx(sepwin, y, x);
75    wmove(sepwin, y, x+2);
76    snprintf(buff, 1024, " right: %i ", owl_global_get_rightshift(&g));
77    waddstr(sepwin, buff);
78  }
79
80  if (owl_global_is_zaway(&g) || owl_global_is_aaway(&g)) {
81    getyx(sepwin, y, x);
82    wmove(sepwin, y, x+2);
83    wattron(sepwin, A_BOLD);
84    wattroff(sepwin, A_REVERSE);
85    if (owl_global_is_zaway(&g) && owl_global_is_aaway(&g)) {
86      waddstr(sepwin, " AWAY ");
87    } else if (owl_global_is_zaway(&g)) {
88      waddstr(sepwin, " Z-AWAY ");
89    } else if (owl_global_is_aaway(&g)) {
90      waddstr(sepwin, " A-AWAY ");
91    }
92    wattron(sepwin, A_REVERSE);
93    wattroff(sepwin, A_BOLD);
94  }
95
96  if (owl_global_get_curmsg_vert_offset(&g)) {
97    getyx(sepwin, y, x);
98    wmove(sepwin, y, x+2);
99    wattron(sepwin, A_BOLD);
100    wattroff(sepwin, A_REVERSE);
101    waddstr(sepwin, " SCROLL ");
102    wattron(sepwin, A_REVERSE);
103    wattroff(sepwin, A_BOLD);
104  }
105 
106  if (in) {
107    getyx(sepwin, y, x);
108    wmove(sepwin, y, x+2);
109    waddstr(sepwin, in);
110  }
111
112  appendtosepbar = owl_global_get_appendtosepbar(&g);
113  if (appendtosepbar && *appendtosepbar) {
114    getyx(sepwin, y, x);
115    wmove(sepwin, y, x+2);
116    waddstr(sepwin, " ");
117    waddstr(sepwin, owl_global_get_appendtosepbar(&g));
118    waddstr(sepwin, " ");
119  }
120
121  getyx(sepwin, y, x);
122  wmove(sepwin, y, owl_global_get_cols(&g)-1);
123   
124  wattroff(sepwin, A_BOLD);
125  wattroff(sepwin, A_REVERSE);
126  wnoutrefresh(sepwin);
127}
128
129char **atokenize(const char *buffer, const char *sep, int *i)
130{
131  /* each element of return must be freed by user */
132  char **args;
133  char *workbuff, *foo;
134  int done=0, first=1, count=0;
135
136  workbuff=owl_malloc(strlen(buffer)+1);
137  memcpy(workbuff, buffer, strlen(buffer)+1);
138
139  args=NULL;
140  while (!done) {
141    if (first) {
142      first=0;
143      foo=strtok(workbuff, sep);
144    } else {
145      foo=strtok(NULL, sep);
146    }
147    if (foo==NULL) {
148      done=1;
149    } else {
150      args=owl_realloc(args, sizeof(char *) * (count+1));
151      args[count] = owl_strdup(foo);
152      count++;
153    }
154  }
155  *i=count;
156  owl_free(workbuff);
157  return(args);
158}
159
160const char *skiptokens(const char *buff, int n) {
161  /* skips n tokens and returns where that would be. */
162  char quote = 0;
163  while (*buff && n>0) {
164      while (*buff == ' ') buff++;
165      while (*buff && (quote || *buff != ' ')) {
166        if(quote) {
167          if(*buff == quote) quote = 0;
168        } else if(*buff == '"' || *buff == '\'') {
169          quote = *buff;
170        }
171        buff++;
172      }
173      while (*buff == ' ') buff++;
174      n--;
175  }
176  return buff;
177}
178
179/* Return a "nice" version of the path.  Tilde expansion is done, and
180 * duplicate slashes are removed.  Caller must free the return.
181 */
182char *owl_util_makepath(const char *in)
183{
184  int i, j, x;
185  char *out, user[MAXPATHLEN];
186  struct passwd *pw;
187
188  out=owl_malloc(MAXPATHLEN+1);
189  out[0]='\0';
190  j=strlen(in);
191  x=0;
192  for (i=0; i<j; i++) {
193    if (in[i]=='~') {
194      if ( (i==(j-1)) ||          /* last character */
195           (in[i+1]=='/') ) {     /* ~/ */
196        /* use my homedir */
197        pw=getpwuid(getuid());
198        if (!pw) {
199          out[x]=in[i];
200        } else {
201          out[x]='\0';
202          strcat(out, pw->pw_dir);
203          x+=strlen(pw->pw_dir);
204        }
205      } else {
206        /* another user homedir */
207        int a, b;
208        b=0;
209        for (a=i+1; i<j; a++) {
210          if (in[a]==' ' || in[a]=='/') {
211            break;
212          } else {
213            user[b]=in[a];
214            i++;
215            b++;
216          }
217        }
218        user[b]='\0';
219        pw=getpwnam(user);
220        if (!pw) {
221          out[x]=in[i];
222        } else {
223          out[x]='\0';
224          strcat(out, pw->pw_dir);
225          x+=strlen(pw->pw_dir);
226        }
227      }
228    } else if (in[i]=='/') {
229      /* check for a double / */
230      if (i<(j-1) && (in[i+1]=='/')) {
231        /* do nothing */
232      } else {
233        out[x]=in[i];
234        x++;
235      }
236    } else {
237      out[x]=in[i];
238      x++;
239    }
240  }
241  out[x]='\0';
242  return(out);
243}
244
245void atokenize_free(char **tok, int nels)
246{
247  int i;
248  for (i=0; i<nels; i++) {
249    owl_free(tok[i]);
250  }
251  owl_free(tok);
252}
253
254
255void owl_parsefree(char **argv, int argc)
256{
257  int i;
258
259  if (!argv) return;
260 
261  for (i=0; i<argc; i++) {
262    if (argv[i]) owl_free(argv[i]);
263  }
264  owl_free(argv);
265}
266
267char **owl_parseline(const char *line, int *argc)
268{
269  /* break a command line up into argv, argc.  The caller must free
270     the returned values.  If there is an error argc will be set to
271     -1, argv will be NULL and the caller does not need to free
272     anything */
273
274  char **argv;
275  int i, len, between=1;
276  char *curarg;
277  char quote;
278
279  argv=owl_malloc(sizeof(char *));
280  len=strlen(line);
281  curarg=owl_malloc(len+10);
282  strcpy(curarg, "");
283  quote='\0';
284  *argc=0;
285  for (i=0; i<len+1; i++) {
286    /* find the first real character */
287    if (between) {
288      if (line[i]==' ' || line[i]=='\t' || line[i]=='\0') {
289        continue;
290      } else {
291        between=0;
292        i--;
293        continue;
294      }
295    }
296
297    /* deal with a quote character */
298    if (line[i]=='"' || line[i]=="'"[0]) {
299      /* if this type of quote is open, close it */
300      if (quote==line[i]) {
301        quote='\0';
302        continue;
303      }
304
305      /* if no quoting is open then open with this */
306      if (quote=='\0') {
307        quote=line[i];
308        continue;
309      }
310
311      /* if another type of quote is open then treat this as a literal */
312      curarg[strlen(curarg)+1]='\0';
313      curarg[strlen(curarg)]=line[i];
314      continue;
315    }
316
317    /* if it's not a space or end of command, then use it */
318    if (line[i]!=' ' && line[i]!='\t' && line[i]!='\n' && line[i]!='\0') {
319      curarg[strlen(curarg)+1]='\0';
320      curarg[strlen(curarg)]=line[i];
321      continue;
322    }
323
324    /* otherwise, if we're not in quotes, add the whole argument */
325    if (quote=='\0') {
326      /* add the argument */
327      argv=owl_realloc(argv, sizeof(char *)*((*argc)+1));
328      argv[*argc] = owl_strdup(curarg);
329      *argc=*argc+1;
330      strcpy(curarg, "");
331      between=1;
332      continue;
333    }
334
335    /* if it is a space and we're in quotes, then use it */
336    curarg[strlen(curarg)+1]='\0';
337    curarg[strlen(curarg)]=line[i];
338  }
339
340  owl_free(curarg);
341
342  /* check for unbalanced quotes */
343  if (quote!='\0') {
344    owl_parsefree(argv, *argc);
345    *argc=-1;
346    return(NULL);
347  }
348
349  return(argv);
350}
351
352/* caller must free the return */
353char *owl_util_minutes_to_timestr(int in)
354{
355  int days, hours;
356  long run;
357  char *out;
358
359  run=in;
360
361  days=run/1440;
362  run-=days*1440;
363  hours=run/60;
364  run-=hours*60;
365
366  if (days>0) {
367    out=owl_sprintf("%i d %2.2i:%2.2li", days, hours, run);
368  } else {
369    out=owl_sprintf("    %2.2i:%2.2li", hours, run);
370  }
371  return(out);
372}
373
374/* hooks for doing memory allocation et. al. in owl */
375
376void *owl_malloc(size_t size)
377{
378  return(g_malloc(size));
379}
380
381void owl_free(void *ptr)
382{
383  g_free(ptr);
384}
385
386char *owl_strdup(const char *s1)
387{
388  return(g_strdup(s1));
389}
390
391void *owl_realloc(void *ptr, size_t size)
392{
393  return(g_realloc(ptr, size));
394}
395
396/* allocates memory and returns the string or null.
397 * caller must free the string.
398 */
399char *owl_sprintf(const char *fmt, ...)
400{
401  va_list ap;
402  char *ret = NULL;
403  va_start(ap, fmt);
404  ret = g_strdup_vprintf(fmt, ap);
405  va_end(ap);
406  return ret;
407}
408
409/* These are in order of their value in owl.h */
410static const struct {
411  int number;
412  const char *name;
413} color_map[] = {
414  {OWL_COLOR_INVALID, "invalid"},
415  {OWL_COLOR_DEFAULT, "default"},
416  {OWL_COLOR_BLACK, "black"},
417  {OWL_COLOR_RED, "red"},
418  {OWL_COLOR_GREEN, "green"},
419  {OWL_COLOR_YELLOW,"yellow"},
420  {OWL_COLOR_BLUE, "blue"},
421  {OWL_COLOR_MAGENTA, "magenta"},
422  {OWL_COLOR_CYAN, "cyan"},
423  {OWL_COLOR_WHITE, "white"},
424};
425
426/* Return the owl color associated with the named color.  Return -1
427 * if the named color is not available
428 */
429int owl_util_string_to_color(const char *color)
430{
431  int c, i;
432  char *p;
433
434  for (i = 0; i < (sizeof(color_map)/sizeof(color_map[0])); i++)
435    if (strcasecmp(color, color_map[i].name) == 0)
436      return color_map[i].number;
437
438  c = strtol(color, &p, 10);
439  if (p != color && c >= -1 && c < COLORS) {
440    return(c);
441  }
442  return(OWL_COLOR_INVALID);
443}
444
445/* Return a string name of the given owl color */
446const char *owl_util_color_to_string(int color)
447{
448  if (color >= OWL_COLOR_INVALID && color <= OWL_COLOR_WHITE)
449    return color_map[color - OWL_COLOR_INVALID].name;
450  return("Unknown color");
451}
452
453/* Get the default tty name.  Caller must free the return */
454char *owl_util_get_default_tty(void)
455{
456  const char *tmp;
457  char *out;
458
459  if (getenv("DISPLAY")) {
460    out=owl_strdup(getenv("DISPLAY"));
461  } else if ((tmp=ttyname(fileno(stdout)))!=NULL) {
462    out=owl_strdup(tmp);
463    if (!strncmp(out, "/dev/", 5)) {
464      owl_free(out);
465      out=owl_strdup(tmp+5);
466    }
467  } else {
468    out=owl_strdup("unknown");
469  }
470  return(out);
471}
472
473/* strip leading and trailing new lines.  Caller must free the
474 * return.
475 */
476char *owl_util_stripnewlines(const char *in)
477{
478 
479  char  *tmp, *ptr1, *ptr2, *out;
480
481  ptr1=tmp=owl_strdup(in);
482  while (ptr1[0]=='\n') {
483    ptr1++;
484  }
485  ptr2=ptr1+strlen(ptr1)-1;
486  while (ptr2>ptr1 && ptr2[0]=='\n') {
487    ptr2[0]='\0';
488    ptr2--;
489  }
490
491  out=owl_strdup(ptr1);
492  owl_free(tmp);
493  return(out);
494}
495
496/* Delete all lines matching "line" from the named file.  If no such
497 * line is found the file is left intact.  If backup==1 then leave a
498 * backup file containing the original contents.  The match is
499 * case-insensitive.
500 *
501 * Returns the number of lines removed
502 */
503int owl_util_file_deleteline(const char *filename, const char *line, int backup)
504{
505  char *backupfile, *newfile, *buf = NULL;
506  FILE *old, *new;
507  struct stat st;
508  int numremoved = 0;
509
510  if ((old = fopen(filename, "r")) == NULL) {
511    owl_function_error("Cannot open %s (for reading): %s",
512                       filename, strerror(errno));
513    return 0;
514  }
515
516  if (fstat(fileno(old), &st) != 0) {
517    owl_function_error("Cannot stat %s: %s", filename, strerror(errno));
518    return 0;
519  }
520
521  newfile = owl_sprintf("%s.new", filename);
522  if ((new = fopen(newfile, "w")) == NULL) {
523    owl_function_error("Cannot open %s (for writing): %s",
524                       filename, strerror(errno));
525    free(newfile);
526    fclose(old);
527    return 0;
528  }
529
530  if (fchmod(fileno(new), st.st_mode & 0777) != 0) {
531    owl_function_error("Cannot set permissions on %s: %s",
532                       filename, strerror(errno));
533    unlink(newfile);
534    fclose(new);
535    free(newfile);
536    fclose(old);
537    return 0;
538  }
539
540  while (owl_getline_chomp(&buf, old))
541    if (strcasecmp(buf, line) != 0)
542      fprintf(new, "%s\n", buf);
543    else
544      numremoved++;
545
546  fclose(new);
547  fclose(old);
548
549  if (backup) {
550    backupfile = owl_sprintf("%s.backup", filename);
551    unlink(backupfile);
552    if (link(filename, backupfile) != 0) {
553      owl_function_error("Cannot link %s: %s", backupfile, strerror(errno));
554      owl_free(backupfile);
555      unlink(newfile);
556      owl_free(newfile);
557      return 0;
558    }
559    owl_free(backupfile);
560  }
561
562  if (rename(newfile, filename) != 0) {
563    owl_function_error("Cannot move %s to %s: %s",
564                       newfile, filename, strerror(errno));
565    numremoved = 0;
566  }
567
568  unlink(newfile);
569  owl_free(newfile);
570
571  return numremoved;
572}
573
574int owl_util_max(int a, int b)
575{
576  if (a>b) return(a);
577  return(b);
578}
579
580int owl_util_min(int a, int b)
581{
582  if (a<b) return(a);
583  return(b);
584}
585
586/* Return the base class or instance from a zephyr class, by removing
587   leading `un' or trailing `.d'.
588   The caller is responsible for freeing the allocated string.
589*/
590char * owl_util_baseclass(const char * class)
591{
592  char *start, *end;
593
594  while(!strncmp(class, "un", 2)) {
595    class += 2;
596  }
597
598  start = owl_strdup(class);
599  end = start + strlen(start) - 1;
600  while(end > start && *end == 'd' && *(end-1) == '.') {
601    end -= 2;
602  }
603  *(end + 1) = 0;
604
605  return start;
606}
607
608const char * owl_get_datadir(void)
609{
610  const char * datadir = getenv("BARNOWL_DATA_DIR");
611  if(datadir != NULL)
612    return datadir;
613  return DATADIR;
614}
615
616/* Strips format characters from a valid utf-8 string. Returns the
617   empty string if 'in' does not validate. */
618char * owl_strip_format_chars(const char *in)
619{
620  char *r;
621  if (g_utf8_validate(in, -1, NULL)) {
622    const char *s, *p;
623    r = owl_malloc(strlen(in)+1);
624    r[0] = '\0';
625    s = in;
626    p = strchr(s, OWL_FMTEXT_UC_STARTBYTE_UTF8);
627    while(p) {
628      /* If it's a format character, copy up to it, and skip all
629         immediately following format characters. */
630      if (owl_fmtext_is_format_char(g_utf8_get_char(p))) {
631        strncat(r, s, p-s);
632        p = g_utf8_next_char(p);
633        while (owl_fmtext_is_format_char(g_utf8_get_char(p))) {
634          p = g_utf8_next_char(p);
635        }
636        s = p;
637        p = strchr(s, OWL_FMTEXT_UC_STARTBYTE_UTF8);
638      }
639      else {
640        p = strchr(p+1, OWL_FMTEXT_UC_STARTBYTE_UTF8);
641      }
642    }
643    if (s) strcat(r,s);
644  }
645  else {
646    r = owl_strdup("");
647  }
648  return r;
649}
650
651/* If in is not UTF-8, convert from ISO-8859-1. We may want to allow
652 * the caller to specify an alternative in the future. We also strip
653 * out characters in Unicode Plane 16, as we use that plane internally
654 * for formatting.
655 */
656char * owl_validate_or_convert(const char *in)
657{
658  if (g_utf8_validate(in, -1, NULL)) {
659    return owl_strip_format_chars(in);
660  }
661  else {
662    return g_convert(in, -1,
663                     "UTF-8", "ISO-8859-1",
664                     NULL, NULL, NULL);
665  }
666}
667/*
668 * Validate 'in' as UTF-8, and either return a copy of it, or an empty
669 * string if it is invalid utf-8.
670 */
671char * owl_validate_utf8(const char *in)
672{
673  char *out;
674  if (g_utf8_validate(in, -1, NULL)) {
675    out = owl_strdup(in);
676  } else {
677    out = owl_strdup("");
678  }
679  return out;
680}
681
682/* This is based on _extract() and _isCJ() from perl's Text::WrapI18N */
683int owl_util_can_break_after(gunichar c)
684{
685 
686  if (c == ' ') return 1;
687  if (c >= 0x3000 && c <= 0x312f) {
688    /* CJK punctuations, Hiragana, Katakana, Bopomofo */
689    if (c == 0x300a || c == 0x300c || c == 0x300e ||
690        c == 0x3010 || c == 0x3014 || c == 0x3016 ||
691        c == 0x3018 || c == 0x301a)
692      return 0;
693    return 1;
694  }
695  if (c >= 0x31a0 && c <= 0x31bf) {return 1;}  /* Bopomofo */
696  if (c >= 0x31f0 && c <= 0x31ff) {return 1;}  /* Katakana extension */
697  if (c >= 0x3400 && c <= 0x9fff) {return 1;}  /* Han Ideogram */
698  if (c >= 0xf900 && c <= 0xfaff) {return 1;}  /* Han Ideogram */
699  if (c >= 0x20000 && c <= 0x2ffff) {return 1;}  /* Han Ideogram */
700  return 0;
701}
702
703char *owl_escape_highbit(const char *str)
704{
705  GString *out = g_string_new("");
706  unsigned char c;
707  while((c = (*str++))) {
708    if(c == '\\') {
709      g_string_append(out, "\\\\");
710    } else if(c & 0x80) {
711      g_string_append_printf(out, "\\x%02x", (int)c);
712    } else {
713      g_string_append_c(out, c);
714    }
715  }
716  return g_string_free(out, 0);
717}
718
719/* innards of owl_getline{,_chomp} below */
720static int owl_getline_internal(char **s, FILE *fp, int newline)
721{
722  int size = 0;
723  int target = 0;
724  int count = 0;
725  int c;
726
727  while (1) {
728    c = getc(fp);
729    if ((target + 1) > size) {
730      size += BUFSIZ;
731      *s = owl_realloc(*s, size);
732    }
733    if (c == EOF)
734      break;
735    count++;
736    if (c != '\n' || newline)
737        (*s)[target++] = c;
738    if (c == '\n')
739      break;
740  }
741  (*s)[target] = 0;
742
743  return count;
744}
745
746/* Read a line from fp, allocating memory to hold it, returning the number of
747 * byte read.  *s should either be NULL or a pointer to memory allocated with
748 * owl_malloc; it will be owl_realloc'd as appropriate.  The caller must
749 * eventually free it.  (This is roughly the interface of getline in the gnu
750 * libc).
751 *
752 * The final newline will be included if it's there.
753 */
754int owl_getline(char **s, FILE *fp)
755{
756  return owl_getline_internal(s, fp, 1);
757}
758
759/* As above, but omitting the final newline */
760int owl_getline_chomp(char **s, FILE *fp)
761{
762  return owl_getline_internal(s, fp, 0);
763}
764
765/* Read the rest of the input available in fp into a string. */
766char *owl_slurp(FILE *fp)
767{
768  char *buf = NULL;
769  char *p;
770  int size = 0;
771  int count;
772
773  while (1) {
774    buf = owl_realloc(buf, size + BUFSIZ);
775    p = &buf[size];
776    size += BUFSIZ;
777
778    if ((count = fread(p, 1, BUFSIZ, fp)) < BUFSIZ)
779      break;
780  }
781  p[count] = 0;
782
783  return buf;
784}
Note: See TracBrowser for help on using the repository browser.