Add sup-compatible prefixes and achieve sup-compatible print output
[notmuch] / g_mime_test.c
1 /*
2  * Copyright © 2009 Carl Worth
3  *
4  * This program is free software: you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation, either version 3 of the License, or
7  * (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program.  If not, see http://www.gnu.org/licenses/ .
16  *
17  * Author: Carl Worth <cworth@cworth.org>
18  */
19
20 #include <stdio.h>
21 #include <stdlib.h>
22 #include <string.h>
23 #include <errno.h>
24 #include <time.h>
25
26 #include <gmime/gmime.h>
27
28 #define ARRAY_SIZE(arr) (sizeof (arr) / sizeof (arr[0]))
29
30 /* These prefix values are specifically chosen to be compatible
31  * with sup, (http://sup.rubyforge.org), written by
32  * William Morgan <wmorgan-sup@masanjin.net>, and released
33  * under the GNU GPL v2.
34  */
35
36 typedef struct {
37     const char *name;
38     const char *prefix;
39 } prefix_t;
40
41 prefix_t NORMAL_PREFIX[] = {
42     { "subject", "S" },
43     { "body", "B" },
44     { "from_name", "FN" },
45     { "to_name", "TN" },
46     { "name", "N" },
47     { "attachment", "A" }
48 };
49
50 prefix_t BOOLEAN_PREFIX[] = {
51     { "type", "K" },
52     { "from_email", "FE" },
53     { "to_email", "TE" },
54     { "email", "E" },
55     { "date", "D" },
56     { "label", "L" },
57     { "source_id", "I" },
58     { "attachment_extension", "O" },
59     { "msgid", "Q" },
60     { "thread", "H" },
61     { "ref", "R" }
62 };
63
64 static const char *
65 find_prefix (const char *name)
66 {
67     int i;
68
69     for (i = 0; i < ARRAY_SIZE (NORMAL_PREFIX); i++)
70         if (strcmp (name, NORMAL_PREFIX[i].name) == 0)
71             return NORMAL_PREFIX[i].prefix;
72
73     for (i = 0; i < ARRAY_SIZE (BOOLEAN_PREFIX); i++)
74         if (strcmp (name, BOOLEAN_PREFIX[i].name) == 0)
75             return BOOLEAN_PREFIX[i].prefix;
76
77     return "";
78 }
79
80 int TERM_COMBINED = 0;
81
82 static void
83 print_term (const char *prefix_name, const char *value)
84 {
85     const char *prefix;
86
87     if (value == NULL)
88         return;
89
90     prefix = find_prefix (prefix_name);
91
92     if (TERM_COMBINED)
93         printf ("\"%s%s\", ", prefix, value);
94     else
95         printf ("[\"%s\", \"%s\"], ", value, prefix);
96 }
97
98 static void
99 add_address_name (InternetAddress *address, const char *prefix_name)
100 {
101     const char *name;
102
103     name = internet_address_get_name (address);
104
105     if (name)
106         print_term (prefix_name, name);
107 }
108
109 static void
110 add_address_names (InternetAddressList *addresses, const char *address_type)
111 {
112     int i;
113     InternetAddress *address;
114
115     for (i = 0; i < internet_address_list_length (addresses); i++) {
116         address = internet_address_list_get_address (addresses, i);
117         add_address_name (address, address_type);
118         add_address_name (address, "name");
119         add_address_name (address, "body");
120     }
121 }
122
123 static void
124 add_address_addr (InternetAddress *address, const char *prefix_name)
125 {
126     InternetAddressMailbox *mailbox = INTERNET_ADDRESS_MAILBOX (address);
127     const char *addr;
128
129     addr = internet_address_mailbox_get_addr (mailbox);
130
131     if (addr)
132         print_term (prefix_name, addr);
133 }
134
135 static void
136 add_address_addrs (InternetAddressList *addresses, const char *address_type)
137 {
138     int i;
139     InternetAddress *address;
140
141     for (i = 0; i < internet_address_list_length (addresses); i++) {
142         address = internet_address_list_get_address (addresses, i);
143         add_address_addr (address, address_type);
144         add_address_addr (address, "email");
145     }
146 }
147
148 int
149 main (int argc, char **argv)
150 {
151     GMimeStream *stream;
152     GMimeParser *parser;
153     GMimeMessage *message;
154     InternetAddressList *addresses;
155
156     const char *filename;
157     FILE *file;
158
159     const char *value, *from;
160     int i;
161
162     time_t time;
163     struct tm gm_time_tm;
164     char time_str[16]; /* YYYYMMDDHHMMSS + 1 for Y100k compatibility ;-) */
165
166     if (argc < 2) {
167         fprintf (stderr, "Usage: %s <mail-message>\n",
168                  argv[0]);
169         exit (1);
170     }
171
172     filename = argv[1];
173
174     file = fopen (filename, "r");
175     if (! file) {
176         fprintf (stderr, "Error opening %s: %s\n", filename, strerror (errno));
177         exit (1);
178     }
179     
180     g_mime_init (0);
181     
182     stream = g_mime_stream_file_new (file);
183
184     parser = g_mime_parser_new_with_stream (stream);
185
186     message = g_mime_parser_construct_message (parser);
187
188     printf ("text is:\n[");
189     from = g_mime_message_get_sender (message);
190     addresses = internet_address_list_parse_string (from);
191
192     add_address_names (addresses, "from_name");
193
194     add_address_names (g_mime_message_get_all_recipients (message),
195                        "to_name");
196
197     value = g_mime_message_get_subject (message);
198     print_term ("subject", value);
199     print_term ("body", value);
200
201     printf ("]\nterms is:\n[");
202
203     TERM_COMBINED = 1;
204
205     from = g_mime_message_get_sender (message);
206     addresses = internet_address_list_parse_string (from);
207
208     add_address_addrs (addresses, "from_email");
209
210     add_address_addrs (g_mime_message_get_all_recipients (message),
211                        "to_email");
212
213     g_mime_message_get_date (message, &time, NULL);
214
215     gmtime_r (&time, &gm_time_tm);
216
217     if (strftime (time_str, sizeof (time_str),
218                   "%Y%m%d%H%M%S", &gm_time_tm) == 0) {
219         fprintf (stderr, "Internal error formatting time\n");
220         exit (1);
221     }
222
223     print_term ("date", time_str);
224
225     print_term ("label", "inbox");
226     print_term ("label", "unread");
227     print_term ("type", "mail");
228
229     value = g_mime_message_get_message_id (message);
230     print_term ("msgid", value);
231
232     print_term ("source_id", "1");
233
234     value = g_mime_message_get_message_id (message);
235     print_term ("thread", value);
236
237     printf ("]\n");
238
239     g_object_unref (message);
240     g_object_unref (parser);
241     g_object_unref (stream);
242     
243     return 0;
244 }