summaryrefslogtreecommitdiff
path: root/import_booking.c
blob: 4fbb0467f2bcef70682ef9ece15ea9462dc6438f (plain) (blame)
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
#include <assert.h>
#include <err.h>
#include <stdio.h>

#include <libical/ical.h>

#include "sqlite/sqlite3.h"

static const char *DATABASE = "booking.db";

static char *SCHEMA = "CREATE TABLE event ("
		      "    start   TIMESTAMP,"
		      "    end     TIMESTAMP,"
		      "    summary TEXT"
		      ")";

static char *
read_stream(char *s, size_t size, void *d)
{
	return fgets(s, (int)size, (FILE *)d);
}

static void
component(sqlite3 *db, icalcomponent *c)
{
	struct icaltimetype dts = icalcomponent_get_dtstart(c);
	struct icaltimetype dte = icalcomponent_get_dtend(c);

	time_t start = icaltime_as_timet(dts);
	time_t end = icaltime_as_timet(dte);
	const char *summary = icalcomponent_get_summary(c);

	sqlite3_stmt *stmt = NULL;
	const char *sql = "INSERT INTO event (start, end, summary) VALUES ("
			  "datetime(?, 'unixepoch'),"
			  "datetime(?, 'unixepoch')"
			  ", ?)";
	if (sqlite3_prepare_v2(db, sql, -1, &stmt, NULL))
		errx(1, "sqlite3_prepare_v2");

	if (sqlite3_bind_int64(stmt, 1, start))
		errx(1, "sqlite3_bind_int64");
	if (sqlite3_bind_int64(stmt, 2, end))
		errx(1, "sqlite3_bind_int64");
	if (sqlite3_bind_text(stmt, 3, summary, -1, SQLITE_STATIC))
		errx(1, "sqlite3_bind_text");
	if (SQLITE_DONE != sqlite3_step(stmt))
		errx(1, "sqlite3_step");
}

int
main(void)
{
	sqlite3 *db;
	char *errmsg = NULL;
	if (sqlite3_open(DATABASE, &db))
		errx(1, "sqlite3_open: %s", errmsg);

	if (sqlite3_exec(db, SCHEMA, NULL, NULL, &errmsg))
		errx(1, "sqlite3_exec: %s", errmsg);

	icalparser *parser = icalparser_new();
	assert(parser);

	FILE *stream = stdin;
	assert(stream);

	if (sqlite3_exec(db, "BEGIN", NULL, NULL, &errmsg))
		errx(1, "sqlite3_exec: %s", errmsg);
	icalparser_set_gen_data(parser, stream);
	for (;;) {
		char *line = icalparser_get_line(parser, read_stream);
		if (!line) {
			break;
		}
		icalcomponent *c = icalparser_add_line(parser, line);
		if (!c) {
			continue;
		}
		for (icalcompiter i = icalcomponent_begin_component(
			 c, ICAL_VEVENT_COMPONENT);
		     icalcompiter_deref(&i); icalcompiter_next(&i)) {
			icalcomponent *ic = icalcompiter_deref(&i);
			component(db, ic);
		}
		icalcomponent_free(c);
	}
	if (sqlite3_exec(db, "COMMIT", NULL, NULL, &errmsg))
		errx(1, "sqlite3_exec: %s", errmsg);

	icalparser_free(parser);
	sqlite3_close(db);
}