fix: add constraint

DanielleMaywood · DanielleMaywood · commit 9f3a036e5c72 · 2024-09-25T16:10:01.000Z
diff --git a/coderd/database/dump.sql b/coderd/database/dump.sql
diff --git a/coderd/database/migrations/000258_add_otp_reset_to_user.down.sql b/coderd/database/migrations/000258_add_otp_reset_to_user.down.sql
@@ -1,3 +1,5 @@
+ALTER TABLE users DROP CONSTRAINT one_time_passcode_set;
+
 ALTER TABLE users DROP COLUMN hashed_one_time_passcode;
 ALTER TABLE users DROP COLUMN one_time_passcode_expires_at;
 ALTER TABLE users DROP COLUMN must_reset_password;
diff --git a/coderd/database/migrations/000258_add_otp_reset_to_user.up.sql b/coderd/database/migrations/000258_add_otp_reset_to_user.up.sql
@@ -4,5 +4,10 @@ COMMENT ON COLUMN users.hashed_one_time_passcode IS 'A hash of the one-time-pass
 ALTER TABLE users ADD COLUMN one_time_passcode_expires_at timestamp with time zone;
 COMMENT ON COLUMN users.one_time_passcode_expires_at IS 'The time when the one-time-passcode expires.';
 
+ALTER TABLE users ADD CONSTRAINT one_time_passcode_set CHECK (
+    (hashed_one_time_passcode IS NULL AND one_time_passcode_expires_at IS NULL)
+    OR (hashed_one_time_passcode IS NOT NULL AND one_time_passcode_expires_at IS NOT NULL)
+);
+
 ALTER TABLE users ADD COLUMN must_reset_password bool NOT NULL DEFAULT false;
 COMMENT ON COLUMN users.must_reset_password IS 'Determines if the user should be forced to change their password.';
