summaryrefslogtreecommitdiff
path: root/src/hus.c
blob: 96049541e813b61960ff10be417a843cc14c90e5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
#include <assert.h>
#include <errno.h>
#include <getopt.h>
#include <pwd.h>
#include <stdlib.h>
#include <string.h>
#include <sys/random.h>
#include <sys/types.h>
#include <unistd.h>

#include "util/debug.h"
#include "util/error.h"
#include "util/versie.h"

static char *shufrange = NULL;

static char **lines = NULL;
static size_t nlines = 0;

static void shuf_lines() {
  size_t *js = calloc(nlines, sizeof(size_t));
  errno = 0;
  if (getrandom(js, nlines*sizeof(size_t), 0) == -1) exit(1);

  for (ssize_t i = nlines - 1; i > 0; i--) {
    size_t j = js[i] % (nlines-1);

    char *x = lines[i];
    lines[i] = lines[j];
    lines[j] = x;
  }

  free(js);
}

static void usage(FILE *f) {
  fprintf(f,
      "Gebruik: hus [-hV] [BESTAND]\n"
      "         hus -i LA-HO [-hV]\n"
      "\n"
      "Hussel de regels van het gegeven bestand.\n"
      "Als er geen bestand gegeven is, of bestand is -, hussel dan de regels op standaard invoer.\n"
      "\n"
      "  -i LA-HO    Genereer regels van laag naar hoog en hussel deze\n"
      "  -h          Toon deze hulptekst\n"
      "  -V          Toon versienummer\n");
}

static char** parse_options(int argc, char **argv) {
  int opt;
  while ((opt = getopt(argc, argv, "i:hV")) != -1) {
    switch (opt) {
      case 'i':
        shufrange = optarg;
        break;

      case 'h':
        usage(stdout);
        exit(0);

      case 'V':
        drukkedoos_print_versie(stdout);
        exit(0);

      case '?':
        fprintf(stderr, "hus: Ongeldige optie: -%c\n", optopt);
        usage(stderr);
        exit(1);
    }
  }

  return argv + optind;
}

static int exec_shufrange() {
  assert(shufrange != NULL);
  char *seploc = NULL;

  if ((seploc = strchr(shufrange, '-')) != NULL) {
    *seploc = '\0';
    const long long start = strtoll(shufrange, NULL, 10);
    const long long end = strtoll(seploc+1, NULL, 10);

    if (start > end) {
      fprintf(stderr, "hus: HO groter dan LA\n");
      return 1;
    }

    nlines = end - start + 1;
    lines = calloc(nlines, sizeof(char*));
    for (long long i = 0; i <= start+end; i++) {
      long long n = i + start;
      asprintf(lines + i, "%lld", n);
    }
  } else {
    fprintf(stderr, "hus: geen LA en HO waarde opgegeven\n");
    return 1;
  }

  return 0;
}

static int exec_shufstream(FILE *restrict stream) {
  size_t cap = 1 << 8;
  lines = malloc(sizeof(char*)*cap);

  char *line = NULL;
  size_t linen = 0;

  ssize_t nread = 0;
  while ((errno = 0, nread = getline(&line, &linen, stream)) != -1) {
    if (nread > 0 && line[nread-1] == '\n') line[--nread] = '\0';

    if (nlines == cap) {
      cap *= 2;
      lines = realloc(lines, sizeof(char*)*cap);
    }
    lines[nlines] = strdup(line);
    nlines++;
  }
  free(line);

  return 0;
}

int entry_hus(int argc, char **argv) {
  int res;
#define CHECK(f) if ((res = f) != 0) return res;

  char **args = parse_options(argc, argv);

  // count the amount of arguments provided
  size_t nargs;
  for (nargs = 0; args[nargs] != NULL; nargs++);

  if (shufrange) {
    CHECK(exec_shufrange());
  } else if (nargs == 0 || (nargs == 1 && strcmp(args[0], "-") == 0)) {
    CHECK(exec_shufstream(stdin));
  } else if (nargs == 1) {
    FILE *stream = fopen(args[0], "r");
    if (stream == NULL) {
      fprintf(stderr, "hus: %s: kon bestand niet openen\n", args[0]);
      return 1;
    }

    CHECK(exec_shufstream(stream));
    fclose(stream);
  } else {
    fprintf(stderr, "hus: teveel argumenten gegeven\n");
    return 1;
  }

  shuf_lines();
  for (size_t i = 0; i < nlines; i++) {
    printf("%s\n", lines[i]);
  }

  return 0;
}