From deeb3ad3a44b44465ae47453963255ce1944b5a7 Mon Sep 17 00:00:00 2001
From: rladenson <78043712+rladenson@users.noreply.github.com>
Date: Thu, 10 Aug 2023 18:06:23 +1200
Subject: [PATCH] fix: add constraints for system name and avatar privacy
 (#579)

Merges PluralKit/PluralKit#579
---
 PluralKit.Core/Database/Migrations/39.sql         | 7 +++++++
 PluralKit.Core/Database/Utils/DatabaseMigrator.cs | 2 +-
 2 files changed, 8 insertions(+), 1 deletion(-)
 create mode 100644 PluralKit.Core/Database/Migrations/39.sql

diff --git a/PluralKit.Core/Database/Migrations/39.sql b/PluralKit.Core/Database/Migrations/39.sql
new file mode 100644
index 00000000..37ea0eec
--- /dev/null
+++ b/PluralKit.Core/Database/Migrations/39.sql
@@ -0,0 +1,7 @@
+-- database version 39
+-- add missing privacy constraints
+
+alter table systems add constraint systems_name_privacy_check check (name_privacy = ANY (ARRAY[1,2]));
+alter table systems add constraint systems_avatar_privacy_check check (avatar_privacy = ANY (ARRAY[1,2]));
+
+update info set schema_version = 39;
\ No newline at end of file
diff --git a/PluralKit.Core/Database/Utils/DatabaseMigrator.cs b/PluralKit.Core/Database/Utils/DatabaseMigrator.cs
index 9b0a39c0..d0f8a4e3 100644
--- a/PluralKit.Core/Database/Utils/DatabaseMigrator.cs
+++ b/PluralKit.Core/Database/Utils/DatabaseMigrator.cs
@@ -9,7 +9,7 @@ namespace PluralKit.Core;
 internal class DatabaseMigrator
 {
     private const string RootPath = "PluralKit.Core.Database"; // "resource path" root for SQL files
-    private const int TargetSchemaVersion = 38;
+    private const int TargetSchemaVersion = 39;
     private readonly ILogger _logger;
 
     public DatabaseMigrator(ILogger logger)