initial commit

7 files changed, 311 insertions, 0 deletions

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..bd2f63d
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,41 @@
+target/
+!.mvn/wrapper/maven-wrapper.jar
+!**/src/main/**/target/
+!**/src/test/**/target/
+.kotlin
+
+### IntelliJ IDEA ###
+.idea/modules.xml
+.idea/jarRepositories.xml
+.idea/compiler.xml
+.idea/libraries/
+*.iws
+*.iml
+*.ipr
+
+### Eclipse ###
+.apt_generated
+.classpath
+.factorypath
+.project
+.settings
+.springBeans
+.sts4-cache
+
+### NetBeans ###
+/nbproject/private/
+/nbbuild/
+/dist/
+/nbdist/
+/.nb-gradle/
+build/
+!**/src/main/**/build/
+!**/src/test/**/build/
+
+### VS Code ###
+.vscode/
+
+### Mac OS ###
+.DS_Store
+/.env
+.idea
\ No newline at end of file
diff --git a/pom.xml b/pom.xml
new file mode 100644
index 0000000..6860f34
--- /dev/null
+++ b/pom.xml
@@ -0,0 +1,70 @@
+<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+  xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+  <modelVersion>4.0.0</modelVersion>
+
+  <groupId>org.example</groupId>
+  <artifactId>NewDiscordAsStorage</artifactId>
+  <version>1.0-SNAPSHOT</version>
+  <packaging>jar</packaging>
+
+  <name>NewDiscordAsStorage</name>
+  <url>http://maven.apache.org</url>
+
+  <properties>
+    <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
+  </properties>
+
+  <dependencies>
+      <dependency>
+          <groupId>net.dv8tion</groupId>
+          <artifactId>JDA</artifactId>
+          <version>6.2.0</version>
+          <!-- Optionally disable audio natives to reduce jar size by excluding `opus-java` and `tink` -->
+          <exclusions>
+              <!-- required for encoding audio into opus, not needed if audio is already provided in opus encoding
+              <exclusion>
+                  <groupId>club.minnced</groupId>
+                  <artifactId>opus-java</artifactId>
+              </exclusion> -->
+              <!-- required for encrypting and decrypting audio
+              <exclusion>
+                  <groupId>com.google.crypto.tink</groupId>
+                  <artifactId>tink</artifactId>
+              </exclusion> -->
+          </exclusions>
+      </dependency>
+      <dependency>
+          <groupId>io.github.cdimascio</groupId>
+          <artifactId>dotenv-java</artifactId>
+          <version>3.0.0</version>
+      </dependency>
+      <dependency>
+          <groupId>org.postgresql</groupId>
+          <artifactId>postgresql</artifactId>
+          <version>42.7.7</version>
+      </dependency>
+      <dependency>
+          <groupId>com.sparkjava</groupId>
+          <artifactId>spark-core</artifactId>
+          <version>2.9.4</version>
+      </dependency>
+    <dependency>
+      <groupId>junit</groupId>
+      <artifactId>junit</artifactId>
+      <version>3.8.1</version>
+      <scope>test</scope>
+    </dependency>
+  </dependencies>
+    <build>
+        <plugins>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-compiler-plugin</artifactId>
+                <configuration>
+                    <source>15</source>
+                    <target>15</target>
+                </configuration>
+            </plugin>
+        </plugins>
+    </build>
+</project>
diff --git a/schema.sql b/schema.sql
new file mode 100644
index 0000000..aa5ee69
--- /dev/null
+++ b/schema.sql
@@ -0,0 +1,43 @@
+-- =========================
+-- Directories table
+-- =========================
+CREATE TABLE IF NOT EXISTS directories (
+    directory_id BIGSERIAL PRIMARY KEY,
+    path TEXT NOT NULL,
+    created_at TIMESTAMPTZ NOT NULL DEFAULT now(),
+    CONSTRAINT directories_path_unique UNIQUE (path)
+);
+
+-- =========================
+-- Files table
+-- =========================
+CREATE TABLE IF NOT EXISTS files (
+    file_id BIGSERIAL PRIMARY KEY,
+    disc_channel_id VARCHAR(255) NOT NULL,
+    disc_message_id VARCHAR(255) NOT NULL,
+    directory_id BIGINT NOT NULL
+        REFERENCES directories(directory_id)
+        ON DELETE RESTRICT,
+    file_name TEXT NOT NULL,
+    file_description TEXT,
+    size BIGINT,
+    mime_type TEXT,
+    created_at TIMESTAMPTZ NOT NULL DEFAULT now(),
+    CONSTRAINT files_unique_name_per_directory
+        UNIQUE (directory_id, file_name)
+);
+
+-- =========================
+-- Indexes
+-- =========================
+
+CREATE INDEX IF NOT EXISTS idx_directories_path
+ON directories (path);
+
+CREATE INDEX IF NOT EXISTS idx_files_directory
+ON files (directory_id);
+
+
+INSERT INTO directories (path)
+VALUES ('')
+ON CONFLICT (path) DO NOTHING;
\ No newline at end of file
diff --git a/src/main/java/com/pinapelz/Database.java b/src/main/java/com/pinapelz/Database.java
new file mode 100644
index 0000000..41d5257
--- /dev/null
+++ b/src/main/java/com/pinapelz/Database.java
@@ -0,0 +1,60 @@
+/*
+Postgres will serve as the index for managing all the files. Iteration through all messages is too slow
+ */
+package com.pinapelz;
+
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.sql.*;
+import java.util.Properties;
+
+public class Database {
+
+    private Connection conn;
+
+    public Database(String host, String user, String password, String db){
+        try {
+            conn = createDBConnection(host, user, password, db);
+            System.out.println("[Database] Running schema.sql as necessary");
+            String schemaSQL = Files.readString(Path.of("schema.sql"));
+            Statement statement = conn.createStatement();
+            statement.execute(schemaSQL);
+        } catch (IOException | SQLException e) {
+            throw new RuntimeException(e);
+        }
+
+    }
+
+    public static Connection createDBConnection(String host, String user, String password, String db) throws IOException, SQLException {
+        String url = "jdbc:postgresql://"+host+"/"+db+"?sslmode=require&channel_binding=require";
+        Properties props = new Properties();
+        props.setProperty("user", user);
+        props.setProperty("password", password);
+        return DriverManager.getConnection(url, props);
+    }
+
+    public void recordFileMetadata(String channelId, String messageId, int rootDirId, String fileName, String description, int size, String mimeType) throws SQLException {
+        PreparedStatement ps = conn.prepareStatement("""
+    INSERT INTO files (
+        disc_channel_id,
+        disc_message_id,
+        directory_id,
+        file_name,
+        file_description,
+        size,
+        mime_type
+    )
+    VALUES (?, ?, ?, ?, ?, ?, ?)
+""");
+        ps.setString(1, channelId);
+        ps.setString(2, messageId);
+        ps.setLong(3, rootDirId);
+        ps.setString(4, fileName);
+        ps.setString(5, description);
+        ps.setLong(6, size);
+        ps.setString(7, mimeType);
+        ps.executeUpdate();
+
+    }
+}
diff --git a/src/main/java/com/pinapelz/FileSystem.java b/src/main/java/com/pinapelz/FileSystem.java
new file mode 100644
index 0000000..61b1bd6
--- /dev/null
+++ b/src/main/java/com/pinapelz/FileSystem.java
@@ -0,0 +1,23 @@
+package com.pinapelz;
+
+import net.dv8tion.jda.api.entities.Message;
+
+import java.sql.SQLException;
+
+public class FileSystem {
+    private Database database;
+    public FileSystem(String dbHost, String dbUser, String dbPass, String dbName){
+        database = new Database(dbHost, dbUser, dbPass, dbName);
+    }
+
+    public void createNewFile(String messageId, String channelId, String description, Message.Attachment attachment){
+        int fileSize = attachment.getSize();
+        String filename = attachment.getFileName();
+        String mimeType = attachment.getContentType();
+        try {
+            database.recordFileMetadata(channelId, messageId, 1, filename, description, fileSize, mimeType );
+        } catch (SQLException e) {
+            throw new RuntimeException(e);
+        }
+    }
+}
diff --git a/src/main/java/com/pinapelz/Main.java b/src/main/java/com/pinapelz/Main.java
new file mode 100644
index 0000000..22ead2c
--- /dev/null
+++ b/src/main/java/com/pinapelz/Main.java
@@ -0,0 +1,35 @@
+package com.pinapelz;
+
+import io.github.cdimascio.dotenv.Dotenv;
+import net.dv8tion.jda.api.JDABuilder;
+import net.dv8tion.jda.api.requests.GatewayIntent;
+import static spark.Spark.*;
+
+public class Main
+{
+    private static final Dotenv dotenv = Dotenv.load();
+
+    public static String readSetting(String parameter) {
+        String value = System.getenv(parameter);
+        if (value != null) return value;
+        return dotenv.get(parameter);
+    }
+
+    public static void startBot(){
+        String dbHost = readSetting("PGHOST");
+        String dbUser = readSetting("PGUSER");
+        String dbPass = readSetting("PGPASSWORD");
+        String dbName = readSetting("PGDATABASE");
+        JDABuilder.createDefault(readSetting("BOT_TOKEN"))
+                .addEventListeners(new MessageListener(dbHost, dbUser, dbPass, dbName))
+                .enableIntents(GatewayIntent.MESSAGE_CONTENT)
+                .build();
+    }
+
+    public static void main(String[] args) throws Exception{
+        startBot();
+        get("/hello", (req, res) -> "Hello World");
+    }
+
+
+}
diff --git a/src/main/java/com/pinapelz/MessageListener.java b/src/main/java/com/pinapelz/MessageListener.java
new file mode 100644
index 0000000..642d484
--- /dev/null
+++ b/src/main/java/com/pinapelz/MessageListener.java
@@ -0,0 +1,39 @@
+package com.pinapelz;
+
+import net.dv8tion.jda.api.entities.Message;
+import net.dv8tion.jda.api.entities.channel.middleman.MessageChannel;
+import net.dv8tion.jda.api.events.message.MessageReceivedEvent;
+import net.dv8tion.jda.api.hooks.ListenerAdapter;
+
+public class MessageListener extends ListenerAdapter {
+
+    private FileSystem fileSystem;
+
+    public MessageListener(String dbHost, String dbUser, String dbPass, String dbName){
+        fileSystem = new FileSystem(dbHost, dbUser, dbPass, dbName);
+
+    }
+
+    @Override
+    public void onMessageReceived(MessageReceivedEvent event)
+    {
+        if (event.getAuthor().isBot()) return;
+        Message message = event.getMessage();
+        String content = message.getContentRaw();
+
+        System.out.println(message.getAttachments().get(0).getUrl());
+        if(!message.getAttachments().isEmpty()){
+            System.out.println("Attachment Received! Filing this away now...");
+            for(Message.Attachment attachment : message.getAttachments()){
+                fileSystem.createNewFile(message.getChannelId(), message.getChannelId(), content, attachment);
+            }
+        }
+
+        if (content.equals("!ping"))
+        {
+            MessageChannel channel = event.getChannel();
+            channel.sendMessage("Pong!").queue(); // Important to call .queue() on the RestAction returned by sendMessage(...)
+        }
+    }
+
+}