forked from wrenn/wrenn
Rafeed M. Bhuiyan
d3e4812e46
Co-authored-by: Tasnim Kabir Sadik <tksadik92@gmail.com> Reviewed-on: wrenn/sandbox#8
83 lines
2.8 KiB
PL/PgSQL
83 lines
2.8 KiB
PL/PgSQL
-- +goose Up
|
|
|
|
-- 1. Add UUID id column to templates and make it the primary key.
|
|
ALTER TABLE templates ADD COLUMN id UUID DEFAULT gen_random_uuid();
|
|
UPDATE templates SET id = gen_random_uuid() WHERE id IS NULL;
|
|
ALTER TABLE templates ALTER COLUMN id SET NOT NULL;
|
|
ALTER TABLE templates DROP CONSTRAINT templates_pkey;
|
|
ALTER TABLE templates ADD PRIMARY KEY (id);
|
|
|
|
-- 2. Name becomes a display field with team-scoped uniqueness.
|
|
ALTER TABLE templates ADD CONSTRAINT uq_templates_team_name UNIQUE (team_id, name);
|
|
|
|
-- 3. Prevent team templates from using names that belong to global (platform) templates.
|
|
-- A team template insert/update with a name matching any platform template is rejected.
|
|
-- +goose StatementBegin
|
|
CREATE OR REPLACE FUNCTION check_global_template_name_collision()
|
|
RETURNS TRIGGER AS $$
|
|
BEGIN
|
|
IF NEW.team_id != '00000000-0000-0000-0000-000000000000' THEN
|
|
IF EXISTS (
|
|
SELECT 1 FROM templates
|
|
WHERE name = NEW.name
|
|
AND team_id = '00000000-0000-0000-0000-000000000000'
|
|
) THEN
|
|
RAISE EXCEPTION 'template name "%" is reserved by a global template', NEW.name
|
|
USING ERRCODE = 'unique_violation';
|
|
END IF;
|
|
END IF;
|
|
RETURN NEW;
|
|
END;
|
|
$$ LANGUAGE plpgsql;
|
|
-- +goose StatementEnd
|
|
|
|
CREATE TRIGGER trg_check_global_template_name
|
|
BEFORE INSERT OR UPDATE ON templates
|
|
FOR EACH ROW
|
|
EXECUTE FUNCTION check_global_template_name_collision();
|
|
|
|
-- 4. Seed the built-in "minimal" template so it appears in all listings.
|
|
-- Both id and team_id are the all-zeros UUID (platform sentinel).
|
|
INSERT INTO templates (id, name, type, vcpus, memory_mb, size_bytes, team_id)
|
|
VALUES (
|
|
'00000000-0000-0000-0000-000000000000',
|
|
'minimal',
|
|
'base',
|
|
1,
|
|
512,
|
|
0,
|
|
'00000000-0000-0000-0000-000000000000'
|
|
) ON CONFLICT DO NOTHING;
|
|
|
|
-- 5. Add template UUID references to template_builds.
|
|
ALTER TABLE template_builds
|
|
ADD COLUMN template_id UUID,
|
|
ADD COLUMN team_id UUID;
|
|
|
|
-- 5. Add template UUID references to sandboxes.
|
|
ALTER TABLE sandboxes
|
|
ADD COLUMN template_id UUID,
|
|
ADD COLUMN template_team_id UUID;
|
|
|
|
-- +goose Down
|
|
|
|
ALTER TABLE sandboxes
|
|
DROP COLUMN IF EXISTS template_team_id,
|
|
DROP COLUMN IF EXISTS template_id;
|
|
|
|
ALTER TABLE template_builds
|
|
DROP COLUMN IF EXISTS team_id,
|
|
DROP COLUMN IF EXISTS template_id;
|
|
|
|
-- Remove the seeded minimal template.
|
|
DELETE FROM templates WHERE id = '00000000-0000-0000-0000-000000000000';
|
|
|
|
DROP TRIGGER IF EXISTS trg_check_global_template_name ON templates;
|
|
DROP FUNCTION IF EXISTS check_global_template_name_collision();
|
|
|
|
ALTER TABLE templates DROP CONSTRAINT IF EXISTS uq_templates_team_name;
|
|
|
|
ALTER TABLE templates DROP CONSTRAINT IF EXISTS templates_pkey;
|
|
ALTER TABLE templates ADD PRIMARY KEY (name);
|
|
ALTER TABLE templates DROP COLUMN IF EXISTS id;
|