]> git.armaanb.net Git - mmenu.git/blob - mmenu.c
Check for double spaces in addition to newlines
[mmenu.git] / mmenu.c
1 #include <stdlib.h>
2 #include <stdio.h>
3 #include <stdbool.h>
4
5 #include <Xm/Xm.h>
6 #include <Xm/Text.h>
7 #include <Xm/RowColumn.h>
8 #include <Xm/List.h>
9
10 #include <curl/curl.h>
11
12 struct memstruct {
13         char *memory;
14         size_t size;
15 };
16
17 char *days[] = {"Monday", "Tuesday", "Wednesday", "Thursday", "Friday"};
18
19 int
20 memfail(void)
21 {
22         printf("ERROR: Out of memory\n");
23         exit(4);
24 }
25
26 static size_t
27 memback(void *contents, size_t size, size_t nmemb, void *userp)
28 {
29         size_t realsize = size * nmemb;
30         struct memstruct *mem = (struct memstruct *)userp;
31
32         char *ptr = realloc(mem->memory, mem->size + realsize + 1);
33         if (!ptr) memfail();
34
35         mem->memory = ptr;
36         memcpy(&(mem->memory[mem->size]), contents, realsize);
37         mem->size += realsize;
38         mem->memory[mem->size] = 0;
39
40         return realsize;
41 }
42
43 char *
44 getmenu(char *date)
45 {
46         CURL *curl = curl_easy_init();
47         CURLcode res;
48
49         if (!curl) {
50                 fprintf(stderr, "ERROR: Failed to initialize cURL\n");
51                 exit(1);
52         }
53
54         struct memstruct chunk;
55         chunk.memory = malloc(1); // Grown as needed by the realloc in memback()
56         chunk.size = 0; // No data yet
57
58         const char *baseurl =
59                 "https://nobilis.nobles.edu/skyworld/castlemenu.php?Date=";
60         int len = strlen(baseurl) + strlen(date) + 1;
61         char *url = (char *)malloc(len);
62         snprintf(url, len, "%s%s", baseurl, date);
63         printf("INFO: Using URL %s\n", url);
64         curl_easy_setopt(curl, CURLOPT_URL, url);
65         curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, memback);
66         curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, 0);
67         curl_easy_setopt(curl, CURLOPT_WRITEDATA, (void *)&chunk);
68
69         printf("INFO: Fetching webpage\n");
70         res = curl_easy_perform(curl);
71         free(url);
72
73         if (res != CURLE_OK) {
74                 fprintf(stderr,"ERROR: Could not fetch webpage\n%s\n",
75                                                 curl_easy_strerror(res));
76                 exit(2);
77         }
78
79         curl_easy_cleanup(curl);
80
81         // Parse HTML
82         printf("INFO: Parsing HTML\n");
83         bool intag = false;
84         char *outp = (char *) calloc(1, sizeof(char));
85         if (!outp) memfail();
86
87         // Extract text from between HTML tags
88         int j = 1;
89         char next[2];
90         next[1] = '\0';
91         for (int i = 345; chunk.memory[i]; i++) {
92                 char c = chunk.memory[i];
93                 if (c == '<') intag = true;
94                 if (!intag) {
95                         j++;
96                         outp = (char *) realloc(outp, j);
97                         if (!outp) memfail();
98                         next[0] = c;
99                         strncat(outp, next, 2);
100                 }
101                 if (c == '>') intag = false;
102         }
103
104         // Strip empty newlines
105         char *nl = (char *) calloc(1, sizeof(char));
106         if (!nl) memfail();
107
108         j = 1;
109         for (int i = 0; outp[i]; i++) {
110                 j++;
111                 if (!((outp[i] == '\n' && outp[i+1] == '\n') ||
112                                 (outp[i] == ' ' && outp[i+1] == ' '))) {
113                         nl = (char *) realloc(nl, j);
114                         if (!nl) memfail();
115                         next[0] = outp[i+1];
116                         strncat(nl, next, 2);
117                 }
118         }
119         free(outp);
120
121         return nl;
122 }
123
124 void
125 dayback(Widget unused, XtPointer client_data, XtPointer call_data)
126 {
127         (void)unused;
128         Widget disp = (Widget) client_data;
129         XmListCallbackStruct *list_cbs = (XmListCallbackStruct *) call_data;
130         char *nl = getmenu(days[list_cbs->item_position - 1]);
131         XmTextReplace((Widget) disp, 0, strlen(XmTextGetString(disp)), nl);
132 }
133
134 // Convert an array of string to an array of compound strings
135 XmStringTable ArgvToXmStringTable (int argc, char **argv)
136 {
137         XmStringTable new =
138                 (XmStringTable) XtMalloc ((argc+1) * sizeof (XmString));
139         if (!new)
140                 return (XmStringTable) 0;
141         new[argc] = (XmString) 0;
142         while (--argc >= 0)
143                 new[argc] = XmStringGenerate ((XtPointer) argv[argc], "tag2",
144                                                                                                                                         XmCHARSET_TEXT, NULL);
145         return new;
146 }
147
148 int
149 main(int argc, char *argv[])
150 {
151         // Initialize motif
152         Widget         toplevel, rowcol, daysel, disp;
153         XmStringTable  strs;
154         XtAppContext   app;
155         Arg            args[5];
156
157         XtSetLanguageProc(NULL, NULL, NULL);
158         toplevel = XtVaOpenApplication(&app, "Castle Menu", NULL, 0, &argc, argv,
159                                                                                                                                  NULL, sessionShellWidgetClass, NULL);
160
161         int n = 0;
162         XtSetArg (args[n], XmNorientation, XmHORIZONTAL); n++;
163         rowcol = XmCreateRowColumn(toplevel, "rowcol", args, n);
164         XtManageChild(rowcol);
165
166         // Create text widget to display menu
167         n = 0;
168         XtSetArg(args[n], XmNvalue, getmenu("")); n++;
169         XtSetArg(args[n], XmNeditable, False); n++;
170         XtSetArg(args[n], XmNcolumns, 80); n++;
171         XtSetArg(args[n], XmNrows, 20); n++;
172         XtSetArg (args[n], XmNeditMode, XmMULTI_LINE_EDIT); n++;
173         XtSetArg(args[n], XmNcursorPositionVisible, False); n++;
174         disp = XmCreateScrolledText(rowcol, "text", args, n);
175         XtManageChild(disp);
176
177         // Create day selector
178         strs = ArgvToXmStringTable (XtNumber(days), days);
179         daysel = XmCreateScrolledList(rowcol, "list", NULL, 0);
180         XtVaSetValues(daysel, XmNitems, strs, XmNitemCount, XtNumber(days), NULL);
181         XtAddCallback(daysel, XmNbrowseSelectionCallback, dayback, (XtPointer) disp);
182         XtManageChild(daysel);
183
184         // Display everything
185         XtRealizeWidget(toplevel);
186         XtAppMainLoop(app);
187 }