Added some ecpg stuff.

1 files changed, 114 insertions, 0 deletions

diff --git a/pg.pgc b/pg.pgc
new file mode 100644
index 0000000..e004dc0
--- /dev/null
+++ b/pg.pgc
@@ -0,0 +1,114 @@
+EXEC SQL WHENEVER SQLERROR CONTINUE;
+
+static inline int iserror() {
+	return !(SQLSTATE[0] == '0' && SQLSTATE[1] == '0');
+}
+
+static inline void printerror() {
+	printf("%c%c%c%c%c\n", SQLSTATE[0], SQLSTATE[1], SQLSTATE[2], SQLSTATE[3], SQLSTATE[4]);
+}
+
+static inline int testerror(const char *code) {
+	for(int i = 0; i < 5; i++) {
+		if(code[i] != SQLSTATE[i])
+			return 0;
+	}
+	return 1;
+}
+
+int pg_connect(const char *string) {
+	EXEC SQL BEGIN DECLARE SECTION;
+	const char *target = string;
+	EXEC SQL END DECLARE SECTION;
+
+	EXEC SQL CONNECT TO :target;
+
+	if(iserror()) {
+		sqlprint();
+		printerror();
+	}
+
+	return !iserror();
+}
+
+void pg_disconnect() {
+	EXEC SQL DISCONNECT;
+}
+
+int pg_init() {
+	EXEC SQL WHENEVER SQLERROR GOTO pg_init_error;
+	EXEC SQL CREATE TABLE schema (version INTEGER);
+	EXEC SQL INSERT INTO schema (version) VALUES (1);
+
+	EXEC SQL CREATE TABLE channel (
+		id serial,
+		name varchar,
+		PRIMARY KEY (id)
+	);
+	EXEC SQL CREATE TABLE channel_hours (
+		channel_id integer,
+		time_index integer,
+		lines integer,
+		PRIMARY KEY (channel_id, time_index),
+		FOREIGN KEY (channel_id) REFERENCES channel (id)
+	);
+
+	/* "user" is a reserved word */
+	EXEC SQL CREATE TABLE users (
+		id serial,
+		nick varchar,
+		words integer,
+		characters integer,
+		kicks integer,
+		kicked integer,
+		monolog_lines integer,
+		monologs integer,
+		seen_first integer,
+		seen_last integer,
+		PRIMARY KEY (id)
+	);
+	EXEC SQL CREATE TABLE user_hours (
+		user_id integer,
+		time_index integer,
+		lines integer,
+		PRIMARY KEY (user_id, time_index),
+		FOREIGN KEY (user_id) REFERENCES users (id)
+	);
+
+	EXEC SQL CREATE TABLE word (
+		id serial PRIMARY KEY,
+		name varchar,
+		count integer
+	);
+
+	EXEC SQL COMMIT;
+
+	return 1;
+
+pg_init_error:
+
+	sqlprint();
+	return 0;
+}
+
+EXEC SQL WHENEVER SQLERROR CONTINUE;
+int pg_upgrade() {
+	EXEC SQL BEGIN DECLARE SECTION;
+	int version = 0;
+	EXEC SQL END DECLARE SECTION;
+
+	EXEC SQL SELECT version INTO :version FROM schema;
+	if(testerror("02000")) {
+		EXEC SQL INSERT INTO schema (version) VALUES (1);
+	} else if(iserror()) {
+		sqlprint();
+		return 0;
+	}
+
+	printf("current schema version: %d\n", version);
+
+	EXEC SQL COMMIT;
+
+	return !iserror();
+}
+/* vim: set syn=c: */