From df604ced267d92a2997690434a1f8c07b9d3aca5 Mon Sep 17 00:00:00 2001
From: Nicholas Molnar <65710+neekolas@users.noreply.github.com>
Date: Tue, 28 May 2024 11:04:00 -0700
Subject: [PATCH] Add initial migration

---
 ...240528175324_wipe-and-init-schema.down.sql | 77 +++++++++++++++++++
 ...20240528175324_wipe-and-init-schema.up.sql |  9 +++
 2 files changed, 86 insertions(+)
 create mode 100644 pkg/migrations/mls/20240528175324_wipe-and-init-schema.down.sql
 create mode 100644 pkg/migrations/mls/20240528175324_wipe-and-init-schema.up.sql

diff --git a/pkg/migrations/mls/20240528175324_wipe-and-init-schema.down.sql b/pkg/migrations/mls/20240528175324_wipe-and-init-schema.down.sql
new file mode 100644
index 00000000..713402ba
--- /dev/null
+++ b/pkg/migrations/mls/20240528175324_wipe-and-init-schema.down.sql
@@ -0,0 +1,77 @@
+SET statement_timeout = 0;
+
+--bun:split
+DROP TABLE IF EXISTS installations, group_messages, welcome_messages, inbox_log, address_log;
+
+--bun:split
+CREATE TABLE installations(
+	id BYTEA PRIMARY KEY,
+	created_at BIGINT NOT NULL,
+	updated_at BIGINT NOT NULL,
+	credential_identity BYTEA NOT NULL,
+	key_package BYTEA NOT NULL,
+	expiration BIGINT NOT NULL
+);
+
+--bun:split
+CREATE INDEX idx_installations_created_at ON installations(created_at);
+
+--bun:split
+CREATE TABLE group_messages(
+	id BIGSERIAL PRIMARY KEY,
+	created_at TIMESTAMP NOT NULL DEFAULT NOW(),
+	group_id BYTEA NOT NULL,
+	data BYTEA NOT NULL,
+	group_id_data_hash BYTEA NOT NULL
+);
+
+--bun:split
+CREATE INDEX idx_group_messages_group_id_created_at ON group_messages(group_id, created_at);
+
+--bun:split
+CREATE UNIQUE INDEX idx_group_messages_group_id_data_hash ON group_messages(group_id_data_hash);
+
+--bun:split
+CREATE TABLE welcome_messages(
+	id BIGSERIAL PRIMARY KEY,
+	created_at TIMESTAMP NOT NULL DEFAULT NOW(),
+	installation_key BYTEA NOT NULL,
+	data BYTEA NOT NULL,
+	hpke_public_key BYTEA NOT NULL,
+	installation_key_data_hash BYTEA NOT NULL
+);
+
+--bun:split
+CREATE INDEX idx_welcome_messages_installation_key_created_at ON welcome_messages(installation_key, created_at);
+
+--bun:split
+CREATE UNIQUE INDEX idx_welcome_messages_group_key_data_hash ON welcome_messages(installation_key_data_hash);
+
+--bun:split
+CREATE TABLE inbox_log(
+	sequence_id BIGSERIAL PRIMARY KEY,
+	inbox_id BYTEA NOT NULL,
+	server_timestamp_ns BIGINT NOT NULL,
+	identity_update_proto BYTEA NOT NULL
+);
+
+--bun:split
+CREATE INDEX idx_inbox_log_inbox_id_sequence_id ON inbox_log(inbox_id, sequence_id);
+
+--bun:split
+CREATE TABLE address_log(
+	address TEXT NOT NULL,
+	inbox_id BYTEA NOT NULL,
+	association_sequence_id BIGINT,
+	revocation_sequence_id BIGINT
+);
+
+--bun:split
+CREATE INDEX idx_address_log_address_inbox_id ON address_log(address, inbox_id);
+
+--bun:split
+CREATE TYPE inbox_filter AS (
+	inbox_id BYTEA,
+	sequence_id BIGINT
+);
+
diff --git a/pkg/migrations/mls/20240528175324_wipe-and-init-schema.up.sql b/pkg/migrations/mls/20240528175324_wipe-and-init-schema.up.sql
new file mode 100644
index 00000000..87d60f3e
--- /dev/null
+++ b/pkg/migrations/mls/20240528175324_wipe-and-init-schema.up.sql
@@ -0,0 +1,9 @@
+SET statement_timeout = 0;
+
+--bun:split
+
+SELECT 1
+
+--bun:split
+
+SELECT 2