]> git.armaanb.net Git - mmenu.git/blob - mmenu.c
Fix style
[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') i+=3;
112                 nl = (char *) realloc(nl, j);
113                 if (!nl) memfail();
114                 next[0] = outp[i+1];
115                 strncat(nl, next, 2);
116         }
117         free(outp);
118         return nl;
119 }
120
121 void
122 dayback(Widget unused, XtPointer client_data, XtPointer call_data)
123 {
124         (void)unused;
125         Widget disp = (Widget) client_data;
126         XmListCallbackStruct *list_cbs = (XmListCallbackStruct *) call_data;
127         char *nl = getmenu(days[list_cbs->item_position - 1]);
128         XmTextReplace((Widget) disp, 0, strlen(XmTextGetString(disp)), nl);
129 }
130
131 // Convert an array of string to an array of compound strings
132 XmStringTable ArgvToXmStringTable (int argc, char **argv)
133 {
134         XmStringTable new =
135                 (XmStringTable) XtMalloc ((argc+1) * sizeof (XmString));
136         if (!new)
137                 return (XmStringTable) 0;
138         new[argc] = (XmString) 0;
139         while (--argc >= 0)
140                 new[argc] = XmStringGenerate ((XtPointer) argv[argc], "tag2",
141                                                                                                                                         XmCHARSET_TEXT, NULL);
142         return new;
143 }
144
145 int
146 main(int argc, char *argv[])
147 {
148         // Initialize motif
149         Widget         toplevel, rowcol, daysel, disp;
150         XmStringTable  strs;
151         XtAppContext   app;
152         Arg            args[5];
153
154         XtSetLanguageProc(NULL, NULL, NULL);
155         toplevel = XtVaOpenApplication(&app, "Castle Menu", NULL, 0, &argc, argv,
156                                                                                                                                  NULL, sessionShellWidgetClass, NULL);
157
158         int n = 0;
159         XtSetArg (args[n], XmNorientation, XmHORIZONTAL); n++;
160         rowcol = XmCreateRowColumn(toplevel, "rowcol", args, n);
161
162         // Create text widget to display menu
163         n = 0;
164         char *nl = getmenu("");
165         XtSetArg(args[n], XmNvalue, nl); n++;
166         XtSetArg(args[n], XmNeditable, False); n++;
167         XtSetArg(args[n], XmNcolumns, 80); n++;
168         XtSetArg(args[n], XmNcursorPositionVisible, False); n++;
169         disp = XmCreateScrolledText(rowcol, "text", args, n);
170         XtManageChild(disp);
171
172         // Create day selector
173         strs = ArgvToXmStringTable (XtNumber(days), days);
174         daysel = XmCreateScrolledList(rowcol, "list", NULL, 0);
175         XtVaSetValues(daysel, XmNitems, strs, XmNitemCount, XtNumber(days), NULL);
176         XtAddCallback(daysel, XmNbrowseSelectionCallback, dayback, (XtPointer) disp);
177         XtManageChild(daysel);
178         XtManageChild(rowcol);
179
180         // Display everything
181         XtRealizeWidget(toplevel);
182         XtAppMainLoop(app);
183 }