wip

Author: aymanbagabas

pkg/db/migrate/0004_create_orgs_teams.go

@@ -0,0 +1,23 @@
+package migrate
+
+import (
+	"context"
+
+	"github.com/charmbracelet/soft-serve/pkg/db"
+)
+
+const (
+	createOrgsTeamsName    = "create_orgs_teams"
+	createOrgsTeamsVersion = 4
+)
+
+var createOrgsTeams = Migration{
+	Name:    createOrgsTeamsName,
+	Version: createOrgsTeamsVersion,
+	Migrate: func(ctx context.Context, tx *db.Tx) error {
+		return migrateUp(ctx, tx, createOrgsTeamsVersion, createOrgsTeamsName)
+	},
+	Rollback: func(ctx context.Context, tx *db.Tx) error {
+		return migrateDown(ctx, tx, createOrgsTeamsVersion, createOrgsTeamsName)
+	},
+}

pkg/db/migrate/0004_create_orgs_teams_postgres.down.sql

@@ -0,0 +1,95 @@
+CREATE TABLE IF NOT EXISTS handles (
+  id SERIAL PRIMARY KEY,
+  handle TEXT NOT NULL UNIQUE,
+  created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  updated_at TIMESTAMP NOT NULL
+);
+
+CREATE TABLE IF NOT EXISTS organizations (
+  id SERIAL PRIMARY KEY,
+  email TEXT NOT NULL,
+  handle_id INTEGER NOT NULL,
+  created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  updated_at TIMESTAMP NOT NULL,
+  CONSTRAINT handle_id_fk
+  FOREIGN KEY(handle_id) REFERENCES handles(id)
+  ON DELETE CASCADE
+  ON UPDATE CASCADE
+);
+
+CREATE TABLE IF NOT EXISTS organization_members (
+  id SERIAL PRIMARY KEY,
+  org_id INTEGER NOT NULL,
+  user_id INTEGER NOT NULL,
+  created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  updated_at TIMESTAMP NOT NULL,
+  UNIQUE (org_id, user_id),
+  CONSTRAINT org_id_fk
+  FOREIGN KEY(org_id) REFERENCES organizations(id)
+  ON DELETE CASCADE
+  ON UPDATE CASCADE,
+  CONSTRAINT user_id_fk
+  FOREIGN KEY(user_id) REFERENCES users(id)
+  ON DELETE CASCADE
+  ON UPDATE CASCADE
+);
+
+CREATE TABLE IF NOT EXISTS teams (
+  id SERIAL PRIMARY KEY,
+  name TEXT NOT NULL,
+  org_id INTEGER NOT NULL,
+  created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  updated_at TIMESTAMP NOT NULL,
+  UNIQUE (name, org_id),
+  CONSTRAINT org_id_fk
+  FOREIGN KEY(org_id) REFERENCES organizations(id)
+  ON DELETE CASCADE
+  ON UPDATE CASCADE
+);
+
+CREATE TABLE IF NOT EXISTS team_members (
+  id SERIAL PRIMARY KEY,
+  team_id INTEGER NOT NULL,
+  user_id INTEGER NOT NULL,
+  created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  updated_at TIMESTAMP NOT NULL,
+  UNIQUE (team_id, user_id),
+  CONSTRAINT team_id_fk
+  FOREIGN KEY(team_id) REFERENCES teams(id)
+  ON DELETE CASCADE
+  ON UPDATE CASCADE,
+  CONSTRAINT user_id_fk
+  FOREIGN KEY(user_id) REFERENCES users(id)
+  ON DELETE CASCADE
+  ON UPDATE CASCADE
+);
+
+-- Add email to users table
+ALTER TABLE users ADD COLUMN email TEXT;
+
+-- Add handle_id to users table
+ALTER TABLE users ADD COLUMN handle_id INTEGER;
+ALTER TABLE users ADD CONSTRAINT handle_id_fk
+  FOREIGN KEY(handle_id) REFERENCES handles(id)
+  ON DELETE CASCADE
+  ON UPDATE CASCADE;
+
+-- Migrate user username to handles
+INSERT INTO handles (handle, updated_at) SELECT username, updated_at FROM users;
+
+-- Update handle_id for users
+UPDATE users SET handle_id = handles.id FROM handles WHERE handles.handle = users.username;
+
+-- Make handle_id not null and unique
+ALTER TABLE users ALTER COLUMN handle_id SET NOT NULL;
+ALTER TABLE users ADD CONSTRAINT handle_id_unique UNIQUE (handle_id);
+
+-- Drop username from users
+ALTER TABLE users DROP COLUMN username;
+
+-- Add org_id to repos table
+ALTER TABLE repos ADD COLUMN org_id INTEGER;
+
+-- Alter user_id nullness in repos table
+ALTER TABLE repos ALTER COLUMN user_id DROP NOT NULL;
+

pkg/db/migrate/0004_create_orgs_teams_postgres.up.sql

@@ -18,6 +18,7 @@ var migrations = []Migration{
 	createTables,
 	webhooks,
 	migrateLfsObjects,
+	createOrgsTeams,
 }
 
 func execMigration(ctx context.Context, tx *db.Tx, version int, name string, down bool) error {