From fe8f705f3c89e8776318158c7fc88b2a359ec2b2 Mon Sep 17 00:00:00 2001
From: Jon Sundin <sundinjon@gmail.com>
Date: Thu, 25 Jan 2024 23:43:01 +0000
Subject: [PATCH] db setup

---
 src/main/java/Main.java | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/src/main/java/Main.java b/src/main/java/Main.java
index de2c86b..80111b6 100644
--- a/src/main/java/Main.java
+++ b/src/main/java/Main.java
@@ -1,14 +1,46 @@
 import Controller.SocialMediaController;
 import io.javalin.Javalin;
 
+import Util.ConnectionUtil;
+
+import java.sql.Connection;
+import java.sql.PreparedStatement;
+import java.sql.SQLException;
+// import java.util.Scanner;
+
 /**
  * This class is provided with a main method to allow you to manually run and test your application. This class will not
  * affect your program in any way and you may write whatever code you like here.
  */
 public class Main {
     public static void main(String[] args) {
+        databaseSetup();
         SocialMediaController controller = new SocialMediaController();
         Javalin app = controller.startAPI();
         app.start(8080);
     }
+
+    public static void databaseSetup(){
+        try {
+            Connection conn = ConnectionUtil.getConnection();
+            String createAccountTable = "CREATE TABLE Account ("
+                                        + "account_id INTEGER PRIMARY KEY AUTO_INCREMENT"
+                                        + "username VARCHAR(255) UNIQUE,"
+                                        + "password VARCHAR(255)";
+            PreparedStatement ps1 = conn.prepareStatement(createAccountTable);
+            ps1.executeUpdate();
+
+            String createMessageTable = "CREATE TABLE Message ("
+                                        + "message_id INTEGER PRIMARY KEY AUTO_INCREMENT,"
+                                        + "posted_by INTEGER,"
+                                        + "message_text VARCHAR(255),"
+                                        + "time_posted_epoch LONG,"
+                                        + "FOREIGN KEY (posted_by) REFERENCES Account(account_id)"
+                                        + ")";
+            PreparedStatement ps2 = conn.prepareStatement(createMessageTable);
+            ps2.executeUpdate();
+        }catch(SQLException e){
+            e.printStackTrace();
+        }
+    }
 }