diff --git a/src/main/java/dev/salmonllama/fsbot/database/models/WelcomeMessage.java b/src/main/java/dev/salmonllama/fsbot/database/models/WelcomeMessage.java index 501e72e..218a904 100644 --- a/src/main/java/dev/salmonllama/fsbot/database/models/WelcomeMessage.java +++ b/src/main/java/dev/salmonllama/fsbot/database/models/WelcomeMessage.java @@ -11,16 +11,22 @@ import java.sql.Timestamp; public class WelcomeMessage extends DatabaseModel { // Need the message, the last time it was updated, and the id of the last person who updated it. + private String serverId; private String message; private Timestamp updated; private String editor; private WelcomeMessage(WelcomeMessageBuilder builder) { + serverId = builder.serverId; message = builder.message; updated = builder.updated; editor = builder.editor; } + public String getServerId() { + return serverId; + } + public String getMessage() { return message; } @@ -33,21 +39,37 @@ public class WelcomeMessage extends DatabaseModel { return editor; } + public static String schema() { + return "CREATE TABLE IF NOT EXISTS welcome_messages (" + + "server_id TEXT PRIMARY KEY," + + "message TEXT," + + "updated TEXT," + + "editor TEXT)"; + } + public static class WelcomeMessageBuilder { - private String message; + private String serverId; + private String message = null; private Timestamp updated = null; private String editor = null; - public WelcomeMessageBuilder(String message) { + public WelcomeMessageBuilder(String serverId) { + this.serverId = serverId; + } + + public WelcomeMessageBuilder setMessage(String message) { this.message = message; + return this; } - public void setUpdated(Timestamp updated) { + public WelcomeMessageBuilder setUpdated(Timestamp updated) { this.updated = updated; + return this; } - public void setEditor(String editor) { + public WelcomeMessageBuilder setEditor(String editor) { this.editor = editor; + return this; } public WelcomeMessage build() {