diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..dc756d7 --- /dev/null +++ b/.gitignore @@ -0,0 +1,39 @@ +HELP.md +.gradle +build/ +!gradle/wrapper/gradle-wrapper.jar +!**/src/main/**/build/ +!**/src/test/**/build/ +/src/main/resources/static/ +/temp-filestore/ + +### STS ### +.apt_generated +.classpath +.factorypath +.project +.settings +.springBeans +.sts4-cache +bin/ +!**/src/main/**/bin/ +!**/src/test/**/bin/ + +### IntelliJ IDEA ### +.idea +*.iws +*.iml +*.ipr +out/ +!**/src/main/**/out/ +!**/src/test/**/out/ + +### NetBeans ### +/nbproject/private/ +/nbbuild/ +/dist/ +/nbdist/ +/.nb-gradle/ + +### VS Code ### +.vscode/ diff --git a/build.gradle b/build.gradle new file mode 100644 index 0000000..6a61ff9 --- /dev/null +++ b/build.gradle @@ -0,0 +1,39 @@ +plugins { + id 'java' + id 'org.springframework.boot' version '3.2.2' + id 'io.spring.dependency-management' version '1.1.4' +} + +group = 'de.w665' +version = '0.0.2' + +java { + sourceCompatibility = '21' +} + +configurations { + compileOnly { + extendsFrom annotationProcessor + } +} + +repositories { + mavenCentral() +} + +dependencies { + developmentOnly 'org.springframework.boot:spring-boot-starter-actuator' + implementation 'org.springframework.boot:spring-boot-starter-mail' + implementation 'org.springframework.boot:spring-boot-starter-web' + compileOnly 'org.projectlombok:lombok' + developmentOnly 'org.springframework.boot:spring-boot-devtools' + annotationProcessor 'org.projectlombok:lombok' + testImplementation 'org.springframework.boot:spring-boot-starter-test' + implementation 'com.rethinkdb:rethinkdb-driver:2.4.4' + // https://mvnrepository.com/artifact/com.google.code.gson/gson + implementation group: 'com.google.code.gson', name: 'gson', version: '2.10.1' +} + +tasks.named('test') { + useJUnitPlatform() +} diff --git a/frontend/.vscode/extensions.json b/frontend/.vscode/extensions.json deleted file mode 100644 index 77b3745..0000000 --- a/frontend/.vscode/extensions.json +++ /dev/null @@ -1,4 +0,0 @@ -{ - // For more information, visit: https://go.microsoft.com/fwlink/?linkid=827846 - "recommendations": ["angular.ng-template"] -} diff --git a/frontend/.vscode/launch.json b/frontend/.vscode/launch.json deleted file mode 100644 index 925af83..0000000 --- a/frontend/.vscode/launch.json +++ /dev/null @@ -1,20 +0,0 @@ -{ - // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387 - "version": "0.2.0", - "configurations": [ - { - "name": "ng serve", - "type": "chrome", - "request": "launch", - "preLaunchTask": "npm: start", - "url": "http://localhost:4200/" - }, - { - "name": "ng test", - "type": "chrome", - "request": "launch", - "preLaunchTask": "npm: test", - "url": "http://localhost:9876/debug.html" - } - ] -} diff --git a/frontend/.vscode/tasks.json b/frontend/.vscode/tasks.json deleted file mode 100644 index a298b5b..0000000 --- a/frontend/.vscode/tasks.json +++ /dev/null @@ -1,42 +0,0 @@ -{ - // For more information, visit: https://go.microsoft.com/fwlink/?LinkId=733558 - "version": "2.0.0", - "tasks": [ - { - "type": "npm", - "script": "start", - "isBackground": true, - "problemMatcher": { - "owner": "typescript", - "pattern": "$tsc", - "background": { - "activeOnStart": true, - "beginsPattern": { - "regexp": "(.*?)" - }, - "endsPattern": { - "regexp": "bundle generation complete" - } - } - } - }, - { - "type": "npm", - "script": "test", - "isBackground": true, - "problemMatcher": { - "owner": "typescript", - "pattern": "$tsc", - "background": { - "activeOnStart": true, - "beginsPattern": { - "regexp": "(.*?)" - }, - "endsPattern": { - "regexp": "bundle generation complete" - } - } - } - } - ] -} diff --git a/frontend/angular.json b/frontend/angular.json index 3af4900..8fbc1d7 100644 --- a/frontend/angular.json +++ b/frontend/angular.json @@ -17,7 +17,7 @@ "build": { "builder": "@angular-devkit/build-angular:application", "options": { - "outputPath": "dist/frontend", + "outputPath": "../src/main/resources/static", "index": "src/index.html", "browser": "src/main.ts", "polyfills": [ diff --git a/gradle/wrapper/gradle-wrapper.jar b/gradle/wrapper/gradle-wrapper.jar new file mode 100644 index 0000000..d64cd49 Binary files /dev/null and b/gradle/wrapper/gradle-wrapper.jar differ diff --git a/gradle/wrapper/gradle-wrapper.properties b/gradle/wrapper/gradle-wrapper.properties new file mode 100644 index 0000000..1af9e09 --- /dev/null +++ b/gradle/wrapper/gradle-wrapper.properties @@ -0,0 +1,7 @@ +distributionBase=GRADLE_USER_HOME +distributionPath=wrapper/dists +distributionUrl=https\://services.gradle.org/distributions/gradle-8.5-bin.zip +networkTimeout=10000 +validateDistributionUrl=true +zipStoreBase=GRADLE_USER_HOME +zipStorePath=wrapper/dists diff --git a/gradlew b/gradlew new file mode 100644 index 0000000..1aa94a4 --- /dev/null +++ b/gradlew @@ -0,0 +1,249 @@ +#!/bin/sh + +# +# Copyright © 2015-2021 the original authors. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# https://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# + +############################################################################## +# +# Gradle start up script for POSIX generated by Gradle. +# +# Important for running: +# +# (1) You need a POSIX-compliant shell to run this script. If your /bin/sh is +# noncompliant, but you have some other compliant shell such as ksh or +# bash, then to run this script, type that shell name before the whole +# command line, like: +# +# ksh Gradle +# +# Busybox and similar reduced shells will NOT work, because this script +# requires all of these POSIX shell features: +# * functions; +# * expansions «$var», «${var}», «${var:-default}», «${var+SET}», +# «${var#prefix}», «${var%suffix}», and «$( cmd )»; +# * compound commands having a testable exit status, especially «case»; +# * various built-in commands including «command», «set», and «ulimit». +# +# Important for patching: +# +# (2) This script targets any POSIX shell, so it avoids extensions provided +# by Bash, Ksh, etc; in particular arrays are avoided. +# +# The "traditional" practice of packing multiple parameters into a +# space-separated string is a well documented source of bugs and security +# problems, so this is (mostly) avoided, by progressively accumulating +# options in "$@", and eventually passing that to Java. +# +# Where the inherited environment variables (DEFAULT_JVM_OPTS, JAVA_OPTS, +# and GRADLE_OPTS) rely on word-splitting, this is performed explicitly; +# see the in-line comments for details. +# +# There are tweaks for specific operating systems such as AIX, CygWin, +# Darwin, MinGW, and NonStop. +# +# (3) This script is generated from the Groovy template +# https://github.com/gradle/gradle/blob/HEAD/subprojects/plugins/src/main/resources/org/gradle/api/internal/plugins/unixStartScript.txt +# within the Gradle project. +# +# You can find Gradle at https://github.com/gradle/gradle/. +# +############################################################################## + +# Attempt to set APP_HOME + +# Resolve links: $0 may be a link +app_path=$0 + +# Need this for daisy-chained symlinks. +while + APP_HOME=${app_path%"${app_path##*/}"} # leaves a trailing /; empty if no leading path + [ -h "$app_path" ] +do + ls=$( ls -ld "$app_path" ) + link=${ls#*' -> '} + case $link in #( + /*) app_path=$link ;; #( + *) app_path=$APP_HOME$link ;; + esac +done + +# This is normally unused +# shellcheck disable=SC2034 +APP_BASE_NAME=${0##*/} +# Discard cd standard output in case $CDPATH is set (https://github.com/gradle/gradle/issues/25036) +APP_HOME=$( cd "${APP_HOME:-./}" > /dev/null && pwd -P ) || exit + +# Use the maximum available, or set MAX_FD != -1 to use that value. +MAX_FD=maximum + +warn () { + echo "$*" +} >&2 + +die () { + echo + echo "$*" + echo + exit 1 +} >&2 + +# OS specific support (must be 'true' or 'false'). +cygwin=false +msys=false +darwin=false +nonstop=false +case "$( uname )" in #( + CYGWIN* ) cygwin=true ;; #( + Darwin* ) darwin=true ;; #( + MSYS* | MINGW* ) msys=true ;; #( + NONSTOP* ) nonstop=true ;; +esac + +CLASSPATH=$APP_HOME/gradle/wrapper/gradle-wrapper.jar + + +# Determine the Java command to use to start the JVM. +if [ -n "$JAVA_HOME" ] ; then + if [ -x "$JAVA_HOME/jre/sh/java" ] ; then + # IBM's JDK on AIX uses strange locations for the executables + JAVACMD=$JAVA_HOME/jre/sh/java + else + JAVACMD=$JAVA_HOME/bin/java + fi + if [ ! -x "$JAVACMD" ] ; then + die "ERROR: JAVA_HOME is set to an invalid directory: $JAVA_HOME + +Please set the JAVA_HOME variable in your environment to match the +location of your Java installation." + fi +else + JAVACMD=java + if ! command -v java >/dev/null 2>&1 + then + die "ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH. + +Please set the JAVA_HOME variable in your environment to match the +location of your Java installation." + fi +fi + +# Increase the maximum file descriptors if we can. +if ! "$cygwin" && ! "$darwin" && ! "$nonstop" ; then + case $MAX_FD in #( + max*) + # In POSIX sh, ulimit -H is undefined. That's why the result is checked to see if it worked. + # shellcheck disable=SC2039,SC3045 + MAX_FD=$( ulimit -H -n ) || + warn "Could not query maximum file descriptor limit" + esac + case $MAX_FD in #( + '' | soft) :;; #( + *) + # In POSIX sh, ulimit -n is undefined. That's why the result is checked to see if it worked. + # shellcheck disable=SC2039,SC3045 + ulimit -n "$MAX_FD" || + warn "Could not set maximum file descriptor limit to $MAX_FD" + esac +fi + +# Collect all arguments for the java command, stacking in reverse order: +# * args from the command line +# * the main class name +# * -classpath +# * -D...appname settings +# * --module-path (only if needed) +# * DEFAULT_JVM_OPTS, JAVA_OPTS, and GRADLE_OPTS environment variables. + +# For Cygwin or MSYS, switch paths to Windows format before running java +if "$cygwin" || "$msys" ; then + APP_HOME=$( cygpath --path --mixed "$APP_HOME" ) + CLASSPATH=$( cygpath --path --mixed "$CLASSPATH" ) + + JAVACMD=$( cygpath --unix "$JAVACMD" ) + + # Now convert the arguments - kludge to limit ourselves to /bin/sh + for arg do + if + case $arg in #( + -*) false ;; # don't mess with options #( + /?*) t=${arg#/} t=/${t%%/*} # looks like a POSIX filepath + [ -e "$t" ] ;; #( + *) false ;; + esac + then + arg=$( cygpath --path --ignore --mixed "$arg" ) + fi + # Roll the args list around exactly as many times as the number of + # args, so each arg winds up back in the position where it started, but + # possibly modified. + # + # NB: a `for` loop captures its iteration list before it begins, so + # changing the positional parameters here affects neither the number of + # iterations, nor the values presented in `arg`. + shift # remove old arg + set -- "$@" "$arg" # push replacement arg + done +fi + + +# Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script. +DEFAULT_JVM_OPTS='"-Xmx64m" "-Xms64m"' + +# Collect all arguments for the java command: +# * DEFAULT_JVM_OPTS, JAVA_OPTS, JAVA_OPTS, and optsEnvironmentVar are not allowed to contain shell fragments, +# and any embedded shellness will be escaped. +# * For example: A user cannot expect ${Hostname} to be expanded, as it is an environment variable and will be +# treated as '${Hostname}' itself on the command line. + +set -- \ + "-Dorg.gradle.appname=$APP_BASE_NAME" \ + -classpath "$CLASSPATH" \ + org.gradle.wrapper.GradleWrapperMain \ + "$@" + +# Stop when "xargs" is not available. +if ! command -v xargs >/dev/null 2>&1 +then + die "xargs is not available" +fi + +# Use "xargs" to parse quoted args. +# +# With -n1 it outputs one arg per line, with the quotes and backslashes removed. +# +# In Bash we could simply go: +# +# readarray ARGS < <( xargs -n1 <<<"$var" ) && +# set -- "${ARGS[@]}" "$@" +# +# but POSIX shell has neither arrays nor command substitution, so instead we +# post-process each arg (as a line of input to sed) to backslash-escape any +# character that might be a shell metacharacter, then use eval to reverse +# that process (while maintaining the separation between arguments), and wrap +# the whole thing up as a single "set" statement. +# +# This will of course break if any of these variables contains a newline or +# an unmatched quote. +# + +eval "set -- $( + printf '%s\n' "$DEFAULT_JVM_OPTS $JAVA_OPTS $GRADLE_OPTS" | + xargs -n1 | + sed ' s~[^-[:alnum:]+,./:=@_]~\\&~g; ' | + tr '\n' ' ' + )" '"$@"' + +exec "$JAVACMD" "$@" diff --git a/gradlew.bat b/gradlew.bat new file mode 100644 index 0000000..93e3f59 --- /dev/null +++ b/gradlew.bat @@ -0,0 +1,92 @@ +@rem +@rem Copyright 2015 the original author or authors. +@rem +@rem Licensed under the Apache License, Version 2.0 (the "License"); +@rem you may not use this file except in compliance with the License. +@rem You may obtain a copy of the License at +@rem +@rem https://www.apache.org/licenses/LICENSE-2.0 +@rem +@rem Unless required by applicable law or agreed to in writing, software +@rem distributed under the License is distributed on an "AS IS" BASIS, +@rem WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +@rem See the License for the specific language governing permissions and +@rem limitations under the License. +@rem + +@if "%DEBUG%"=="" @echo off +@rem ########################################################################## +@rem +@rem Gradle startup script for Windows +@rem +@rem ########################################################################## + +@rem Set local scope for the variables with windows NT shell +if "%OS%"=="Windows_NT" setlocal + +set DIRNAME=%~dp0 +if "%DIRNAME%"=="" set DIRNAME=. +@rem This is normally unused +set APP_BASE_NAME=%~n0 +set APP_HOME=%DIRNAME% + +@rem Resolve any "." and ".." in APP_HOME to make it shorter. +for %%i in ("%APP_HOME%") do set APP_HOME=%%~fi + +@rem Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script. +set DEFAULT_JVM_OPTS="-Xmx64m" "-Xms64m" + +@rem Find java.exe +if defined JAVA_HOME goto findJavaFromJavaHome + +set JAVA_EXE=java.exe +%JAVA_EXE% -version >NUL 2>&1 +if %ERRORLEVEL% equ 0 goto execute + +echo. +echo ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH. +echo. +echo Please set the JAVA_HOME variable in your environment to match the +echo location of your Java installation. + +goto fail + +:findJavaFromJavaHome +set JAVA_HOME=%JAVA_HOME:"=% +set JAVA_EXE=%JAVA_HOME%/bin/java.exe + +if exist "%JAVA_EXE%" goto execute + +echo. +echo ERROR: JAVA_HOME is set to an invalid directory: %JAVA_HOME% +echo. +echo Please set the JAVA_HOME variable in your environment to match the +echo location of your Java installation. + +goto fail + +:execute +@rem Setup the command line + +set CLASSPATH=%APP_HOME%\gradle\wrapper\gradle-wrapper.jar + + +@rem Execute Gradle +"%JAVA_EXE%" %DEFAULT_JVM_OPTS% %JAVA_OPTS% %GRADLE_OPTS% "-Dorg.gradle.appname=%APP_BASE_NAME%" -classpath "%CLASSPATH%" org.gradle.wrapper.GradleWrapperMain %* + +:end +@rem End local scope for the variables with windows NT shell +if %ERRORLEVEL% equ 0 goto mainEnd + +:fail +rem Set variable GRADLE_EXIT_CONSOLE if you need the _script_ return code instead of +rem the _cmd.exe /c_ return code! +set EXIT_CODE=%ERRORLEVEL% +if %EXIT_CODE% equ 0 set EXIT_CODE=1 +if not ""=="%GRADLE_EXIT_CONSOLE%" exit %EXIT_CODE% +exit /b %EXIT_CODE% + +:mainEnd +if "%OS%"=="Windows_NT" endlocal + +:omega diff --git a/settings.gradle b/settings.gradle new file mode 100644 index 0000000..c0307ac --- /dev/null +++ b/settings.gradle @@ -0,0 +1 @@ +rootProject.name = 'sharepulse' diff --git a/src/main/java/de/w665/sharepulse/SharepulseApplication.java b/src/main/java/de/w665/sharepulse/SharepulseApplication.java new file mode 100644 index 0000000..2c4ac58 --- /dev/null +++ b/src/main/java/de/w665/sharepulse/SharepulseApplication.java @@ -0,0 +1,13 @@ +package de.w665.sharepulse; + +import org.springframework.boot.SpringApplication; +import org.springframework.boot.autoconfigure.SpringBootApplication; + +@SpringBootApplication +public class SharepulseApplication { + + public static void main(String[] args) { + SpringApplication.run(SharepulseApplication.class, args); + } + +} diff --git a/src/main/java/de/w665/sharepulse/config/MvcConfig.java b/src/main/java/de/w665/sharepulse/config/MvcConfig.java new file mode 100644 index 0000000..c1ef35c --- /dev/null +++ b/src/main/java/de/w665/sharepulse/config/MvcConfig.java @@ -0,0 +1,18 @@ +package de.w665.sharepulse.config; + +import org.springframework.context.annotation.Configuration; +import org.springframework.web.servlet.config.annotation.ViewControllerRegistry; +import org.springframework.web.servlet.config.annotation.WebMvcConfigurer; + +@Configuration +public class MvcConfig implements WebMvcConfigurer { + + @Override + public void addViewControllers(ViewControllerRegistry registry) { + registry.addViewController("/home").setViewName("forward:/index.html"); + registry.addViewController("/download").setViewName("forward:/index.html"); + registry.addViewController("/upload").setViewName("forward:/index.html"); + registry.addViewController("/credits").setViewName("forward:/index.html"); + + } +} diff --git a/src/main/java/de/w665/sharepulse/db/RethinkDBConfig.java b/src/main/java/de/w665/sharepulse/db/RethinkDBConfig.java new file mode 100644 index 0000000..0ad5016 --- /dev/null +++ b/src/main/java/de/w665/sharepulse/db/RethinkDBConfig.java @@ -0,0 +1,18 @@ +package de.w665.sharepulse.db; + +import lombok.Getter; +import org.springframework.beans.factory.annotation.Value; +import org.springframework.context.annotation.Configuration; + +@Getter +@Configuration +public class RethinkDBConfig { + @Value("${rethinkdb.host}") + private String host; + + @Value("${rethinkdb.port}") + private int port; + + @Value("${rethinkdb.database}") + private String database; +} diff --git a/src/main/java/de/w665/sharepulse/db/RethinkDBConnector.java b/src/main/java/de/w665/sharepulse/db/RethinkDBConnector.java new file mode 100644 index 0000000..ec58d8d --- /dev/null +++ b/src/main/java/de/w665/sharepulse/db/RethinkDBConnector.java @@ -0,0 +1,27 @@ +package de.w665.sharepulse.db; + +import com.rethinkdb.RethinkDB; +import com.rethinkdb.net.Connection; +import jakarta.annotation.PostConstruct; +import lombok.Getter; +import lombok.RequiredArgsConstructor; +import lombok.extern.slf4j.Slf4j; +import org.springframework.stereotype.Component; + +@Slf4j +@Component +@RequiredArgsConstructor +public class RethinkDBConnector { + + private final RethinkDBConfig config; + @Getter + private final RethinkDB r = RethinkDB.r; + @Getter + private Connection connection; + + @PostConstruct + public void connectToDatabae() { + connection = r.connection().hostname(config.getHost()).port(config.getPort()).connect(); + log.info("Connected to RethinkDB at " + config.getHost() + ":" + config.getPort() + " on database " + config.getDatabase()); + } +} diff --git a/src/main/java/de/w665/sharepulse/db/RethinkDBService.java b/src/main/java/de/w665/sharepulse/db/RethinkDBService.java new file mode 100644 index 0000000..879e9c8 --- /dev/null +++ b/src/main/java/de/w665/sharepulse/db/RethinkDBService.java @@ -0,0 +1,90 @@ +package de.w665.sharepulse.db; + +import com.rethinkdb.RethinkDB; +import com.rethinkdb.gen.exc.ReqlOpFailedError; +import com.rethinkdb.net.Connection; +import com.rethinkdb.net.Result; +import jakarta.annotation.PostConstruct; +import jakarta.annotation.PreDestroy; +import lombok.RequiredArgsConstructor; +import lombok.extern.slf4j.Slf4j; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Service; + +import java.util.ArrayList; +import java.util.List; +import java.util.concurrent.atomic.AtomicReference; + +@Slf4j +@Service +public class RethinkDBService { + + private final RethinkDBConfig config; + + private final RethinkDB r; + private final Connection connection; + + @Autowired + public RethinkDBService(RethinkDBConfig config, RethinkDBConnector connector) { + this.config = config; + + + // mapping to private vars for easier access + this.r = connector.getR(); + this.connection = connector.getConnection(); + } + + @PostConstruct + public void initialize() { + + //rethinkdb check if database exists + + /*Result result = r.dbList().run(connection); + List databases = new ArrayList<>(); + + Object db = result.first(); + ArrayList dbList = (ArrayList) db; + databases.addAll(dbList); + + if (!databases.contains("sharepulse")) { + r.dbCreate("sharepulse").run(connection); + log.info("Database 'sharepulse' created successfully."); + } else { + log.info("Database 'sharepulse' already exists. No action needed."); + }*/ + + // rethinkdb check if database exists + try { + r.dbCreate(config.getDatabase()).run(connection).stream(); + log.debug("Database " + config.getDatabase() + " created"); + } catch (ReqlOpFailedError e) { + log.debug("Database " + config.getDatabase() + " already exists. Error: " + e.getClass().getSimpleName()); + } + + // rethinkdb check if table file_uploads exists + try { + r.db(config.getDatabase()).tableCreate("file_uploads").run(connection).stream(); + log.debug("Table 'file_uploads' created successfully."); + } catch (ReqlOpFailedError e) { + log.debug("Table 'file_uploads' already exists. Clearing content..."); + r.db(config.getDatabase()).table("file_uploads").delete().run(connection); + log.debug("Table 'file_uploads' cleared successfully."); + } + + // rethinkdb check if table id_store exists + try { + r.db(config.getDatabase()).tableCreate("id_store").run(connection).stream(); + log.debug("Table 'id_store' created successfully."); + } catch (ReqlOpFailedError e) { + log.debug("Table 'id_store' already exists. No action needed."); + } + log.info("Database ready for operation!"); + } + + @PreDestroy + public void close() { + if (connection != null) { + connection.close(); + } + } +} diff --git a/src/main/java/de/w665/sharepulse/db/repo/FileUploadRepository.java b/src/main/java/de/w665/sharepulse/db/repo/FileUploadRepository.java new file mode 100644 index 0000000..e82c99a --- /dev/null +++ b/src/main/java/de/w665/sharepulse/db/repo/FileUploadRepository.java @@ -0,0 +1,67 @@ +package de.w665.sharepulse.db.repo; + +import com.google.gson.Gson; +import com.google.gson.GsonBuilder; +import com.google.gson.reflect.TypeToken; +import com.rethinkdb.RethinkDB; +import com.rethinkdb.net.Connection; +import de.w665.sharepulse.db.RethinkDBConnector; +import de.w665.sharepulse.db.RethinkDBService; +import de.w665.sharepulse.model.FileUpload; +import lombok.RequiredArgsConstructor; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Repository; + +import java.lang.reflect.Type; +import java.util.Map; + +@Repository +@RequiredArgsConstructor +public class FileUploadRepository { + + private final RethinkDBService rethinkDBService; + private final RethinkDB r; + private final Connection connection; + private final Gson gson; + + @Autowired + public FileUploadRepository(RethinkDBService rethinkDBService, RethinkDBConnector connector) { + this.rethinkDBService = rethinkDBService; + this.r = connector.getR(); + this.connection = connector.getConnection(); + + this.gson = new GsonBuilder() + .setDateFormat("dd-MM-yyyy HH:mm:ss") // date field formatting + .create(); + } + + public void insertFileUpload(FileUpload fileUpload) { + String json = gson.toJson(fileUpload); + + Type type = new TypeToken>(){}.getType(); + Map map = gson.fromJson(json, type); + + r.db("sharepulse").table("file_uploads").insert(map).run(connection); + } + + public FileUpload retrieveFileUploadByFileId(String fileId) { + return r.db("sharepulse").table("file_uploads") + .filter(r.hashMap("fileId", fileId)) + .run(connection, FileUpload.class) + .next(); + } + + public void updateFileUpload(FileUpload updatedFileUpload) { + String json = gson.toJson(updatedFileUpload); + + Type type = new TypeToken>(){}.getType(); + Map map = gson.fromJson(json, type); + + String fileId = updatedFileUpload.getFileId(); + + r.db("sharepulse").table("file_uploads") + .filter(r.hashMap("fileId", fileId)) + .update(map) + .run(connection); + } +} diff --git a/src/main/java/de/w665/sharepulse/exception/NoDownloadPermissionException.java b/src/main/java/de/w665/sharepulse/exception/NoDownloadPermissionException.java new file mode 100644 index 0000000..d2e0645 --- /dev/null +++ b/src/main/java/de/w665/sharepulse/exception/NoDownloadPermissionException.java @@ -0,0 +1,7 @@ +package de.w665.sharepulse.exception; + +public class NoDownloadPermissionException extends RuntimeException { + public NoDownloadPermissionException(String message) { + super(message); + } +} diff --git a/src/main/java/de/w665/sharepulse/model/FileUpload.java b/src/main/java/de/w665/sharepulse/model/FileUpload.java new file mode 100644 index 0000000..b6ef691 --- /dev/null +++ b/src/main/java/de/w665/sharepulse/model/FileUpload.java @@ -0,0 +1,28 @@ +package de.w665.sharepulse.model; + +import com.fasterxml.jackson.annotation.JsonFormat; +import com.google.gson.annotations.SerializedName; +import lombok.*; + +import java.util.Date; + +@Getter +@Setter +@Builder +@AllArgsConstructor +@NoArgsConstructor +@ToString +public class FileUpload { + private String id; + private String fileId; + private String fileName; + private long fileSize; + private boolean singleDownload; + @JsonFormat(shape = JsonFormat.Shape.STRING, pattern = "dd-MM-yyyy HH:mm:ss", timezone = "CET") + private Date uploadDate; + private String uploadedByIpAddress; + private long downloadCount; + private String fileDescription; + private boolean passwordProtected; + private String downloadPassword; +} diff --git a/src/main/java/de/w665/sharepulse/rest/ApiRestController.java b/src/main/java/de/w665/sharepulse/rest/ApiRestController.java new file mode 100644 index 0000000..828a67e --- /dev/null +++ b/src/main/java/de/w665/sharepulse/rest/ApiRestController.java @@ -0,0 +1,9 @@ +package de.w665.sharepulse.rest; + +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; + +@RestController +@RequestMapping("/api/v1") +public abstract class ApiRestController { +} diff --git a/src/main/java/de/w665/sharepulse/rest/TestRestResource.java b/src/main/java/de/w665/sharepulse/rest/TestRestResource.java new file mode 100644 index 0000000..ce94c50 --- /dev/null +++ b/src/main/java/de/w665/sharepulse/rest/TestRestResource.java @@ -0,0 +1,57 @@ +package de.w665.sharepulse.rest; + +import de.w665.sharepulse.db.repo.FileUploadRepository; +import de.w665.sharepulse.model.FileUpload; +import de.w665.sharepulse.service.FileService; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.core.io.InputStreamResource; +import org.springframework.core.io.Resource; +import org.springframework.http.HttpHeaders; +import org.springframework.http.MediaType; +import org.springframework.http.ResponseEntity; +import org.springframework.web.bind.annotation.*; + +import java.io.File; +import java.io.FileInputStream; +import java.io.IOException; + +@RestController +public class TestRestResource extends ApiRestController { + + private final FileUploadRepository fileUploadRepository; + private final FileService fileService; + + @Autowired + public TestRestResource(FileUploadRepository fileUploadRepository, FileService fileService) { + this.fileUploadRepository = fileUploadRepository; + this.fileService = fileService; + } + + @GetMapping("test") + public String test(@RequestParam String id) { + FileUpload fileUpload = fileUploadRepository.retrieveFileUploadByFileId(id); + return fileUpload != null ? fileUpload.toString() : "FileUpload not found for id: " + id; + } + + @GetMapping("test/download") + public ResponseEntity download(@RequestParam String id) throws IOException { + + File file = fileService.getFileById(id); + + + HttpHeaders headers = new HttpHeaders(); + headers.add(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=\"edmbass.flac\""); + //headers.add(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=\"" + file.getName() + "\""); + headers.add(HttpHeaders.CACHE_CONTROL, "no-cache, no-store, must-revalidate"); + headers.add(HttpHeaders.PRAGMA, "no-cache"); + headers.add(HttpHeaders.EXPIRES, "0"); + + InputStreamResource resource = new InputStreamResource(new FileInputStream(file)); + + return ResponseEntity.ok() + .headers(headers) + .contentLength(file.length()) + .contentType(MediaType.APPLICATION_OCTET_STREAM) + .body(resource); + } +} diff --git a/src/main/java/de/w665/sharepulse/rest/mappings/Download.java b/src/main/java/de/w665/sharepulse/rest/mappings/Download.java new file mode 100644 index 0000000..3733f2c --- /dev/null +++ b/src/main/java/de/w665/sharepulse/rest/mappings/Download.java @@ -0,0 +1,74 @@ +package de.w665.sharepulse.rest.mappings; + +import de.w665.sharepulse.exception.NoDownloadPermissionException; +import de.w665.sharepulse.model.FileUpload; +import de.w665.sharepulse.rest.ApiRestController; +import de.w665.sharepulse.service.FileSecurityService; +import de.w665.sharepulse.service.FileService; +import lombok.extern.slf4j.Slf4j; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.core.io.InputStreamResource; +import org.springframework.core.io.Resource; +import org.springframework.http.HttpHeaders; +import org.springframework.http.HttpStatus; +import org.springframework.http.MediaType; +import org.springframework.http.ResponseEntity; +import org.springframework.web.ErrorResponse; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.RequestParam; +import org.springframework.web.bind.annotation.RestController; + +import java.io.File; +import java.io.FileInputStream; +import java.io.IOException; +import java.util.HashMap; +import java.util.Map; + +@Slf4j +@RestController +public class Download extends ApiRestController { + + private final FileService fileService; + private final FileSecurityService fileSecurityService; + + @Autowired + public Download(FileService fileService, FileSecurityService fileSecurityService) { + this.fileService = fileService; + this.fileSecurityService = fileSecurityService; + } + + @GetMapping("/download") + public ResponseEntity download(@RequestParam String fileId, + @RequestParam(value = "password", required = false) String password) throws IOException { + + + FileUpload fileUpload = fileService.getFileUploadByFileId(fileId); + + try { + fileSecurityService.verifyDownloadPermission(fileUpload, password); + } catch (NoDownloadPermissionException e) { + log.debug("No download permission for file: " + fileId); + ErrorResponse errorResponse = ErrorResponse.builder(e, HttpStatus.FORBIDDEN, e.getMessage()) + .build(); + return new ResponseEntity<>(errorResponse, HttpStatus.FORBIDDEN); + } + + fileService.updateDownloadCount(fileUpload); + + File file = fileService.getFileById(fileId); + InputStreamResource resource = new InputStreamResource(new FileInputStream(file)); + + + HttpHeaders headers = new HttpHeaders(); + headers.add(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=\"" + fileUpload.getFileName() + "\""); + headers.add(HttpHeaders.CACHE_CONTROL, "no-cache, no-store, must-revalidate"); + headers.add(HttpHeaders.PRAGMA, "no-cache"); + headers.add(HttpHeaders.EXPIRES, "0"); + + return ResponseEntity.ok() + .headers(headers) + .contentLength(file.length()) + .contentType(MediaType.APPLICATION_OCTET_STREAM) + .body(resource); + } +} diff --git a/src/main/java/de/w665/sharepulse/rest/mappings/SpeedTest.java b/src/main/java/de/w665/sharepulse/rest/mappings/SpeedTest.java new file mode 100644 index 0000000..8a3a71c --- /dev/null +++ b/src/main/java/de/w665/sharepulse/rest/mappings/SpeedTest.java @@ -0,0 +1,33 @@ +package de.w665.sharepulse.rest.mappings; + +import de.w665.sharepulse.rest.ApiRestController; +import jakarta.servlet.http.HttpServletResponse; +import lombok.extern.slf4j.Slf4j; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.RestController; + +import java.io.IOException; + +@Slf4j +@RestController +public class SpeedTest extends ApiRestController { + + /** + * This method is used to test the download speed of the client. 1MB of dummy data is sent to the client. + * @param response + */ + @GetMapping("/speed-test") + public void speedTest(HttpServletResponse response) { + byte[] dummyData = new byte[1024 * 1024]; + response.setContentType("application/octet-stream"); + response.setContentLength(dummyData.length); + response.setStatus(HttpServletResponse.SC_OK); + try { + response.getOutputStream().write(dummyData); + response.getOutputStream().flush(); + } catch (IOException e) { + log.error("Speed test failed."); + return; + } + } +} diff --git a/src/main/java/de/w665/sharepulse/rest/mappings/Upload.java b/src/main/java/de/w665/sharepulse/rest/mappings/Upload.java new file mode 100644 index 0000000..ee14611 --- /dev/null +++ b/src/main/java/de/w665/sharepulse/rest/mappings/Upload.java @@ -0,0 +1,42 @@ +package de.w665.sharepulse.rest.mappings; + +import de.w665.sharepulse.rest.ApiRestController; +import de.w665.sharepulse.service.FileService; +import jakarta.servlet.http.HttpServletRequest; +import lombok.extern.slf4j.Slf4j; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.http.HttpStatus; +import org.springframework.http.ResponseEntity; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.RequestParam; +import org.springframework.web.bind.annotation.RestController; +import org.springframework.web.multipart.MultipartFile; + +@Slf4j +@RestController +public class Upload extends ApiRestController { + + private final FileService fileService; + + @Autowired + public Upload(FileService fileService) { + this.fileService = fileService; + } + + // Currently testing + @PostMapping("/upload") + public ResponseEntity getUpload(@RequestParam("file") MultipartFile file, HttpServletRequest request, + @RequestParam(value = "password", required = false) String password, + @RequestParam(value = "singleDownload", defaultValue = "false") boolean singleDownload, + @RequestParam(value = "fileDescription", required = false) String fileDescription) { + if (file.isEmpty()) { + log.debug("User tried to upload an empty file. IP: " + request.getRemoteAddr()); + return new ResponseEntity<>("Please select a file to upload.", HttpStatus.NOT_ACCEPTABLE); + } + + String fileId = fileService.processUploadedFile(file, request.getRemoteAddr(), password, singleDownload, fileDescription); + log.debug("User uploaded file " + file.getOriginalFilename() + " from IP " + request.getRemoteAddr() + " successfully."); + return new ResponseEntity<>("File " + file.getOriginalFilename() + " uploaded successfully! ID(" + fileId + ")", HttpStatus.OK); + } +} diff --git a/src/main/java/de/w665/sharepulse/service/FileIdService.java b/src/main/java/de/w665/sharepulse/service/FileIdService.java new file mode 100644 index 0000000..457de4b --- /dev/null +++ b/src/main/java/de/w665/sharepulse/service/FileIdService.java @@ -0,0 +1,34 @@ +package de.w665.sharepulse.service; + +import org.springframework.stereotype.Service; + +import java.text.SimpleDateFormat; +import java.util.Date; + +@Service +public class FileIdService { + + private static int dailyCounter = 0; + private static String lastDate = ""; + + // Not safe to use + public static synchronized String generateId() { + String today = new SimpleDateFormat("yyMMdd").format(new Date()); + + if (!today.equals(lastDate)) { + dailyCounter = 0; + lastDate = today; + } + + String counterEncoded = Integer.toString(++dailyCounter, 36).toUpperCase(); + + return today + counterEncoded; + } + + public String generateNewUniqueId() { + return generateId(); + } + + + +} diff --git a/src/main/java/de/w665/sharepulse/service/FileSecurityService.java b/src/main/java/de/w665/sharepulse/service/FileSecurityService.java new file mode 100644 index 0000000..f55903e --- /dev/null +++ b/src/main/java/de/w665/sharepulse/service/FileSecurityService.java @@ -0,0 +1,37 @@ +package de.w665.sharepulse.service; + +import de.w665.sharepulse.exception.NoDownloadPermissionException; +import de.w665.sharepulse.model.FileUpload; +import lombok.extern.slf4j.Slf4j; +import org.springframework.stereotype.Service; + +import java.io.File; + +@Slf4j +@Service +public class FileSecurityService { + + public boolean verifyDownloadPermission(FileUpload file, String password) throws NoDownloadPermissionException { + + + if (!file.isPasswordProtected() && !file.isSingleDownload()) { + return true; + } + + if (file.isSingleDownload()) { + if(file.getDownloadCount() > 0) { + throw new NoDownloadPermissionException("File has already been downloaded."); + } + if(!file.isPasswordProtected()) { + return true; + } + } + + if(file.getDownloadPassword().equals(password)) { + return true; + } + + throw new NoDownloadPermissionException("Password protected file can only be downloaded with correct password."); + + } +} diff --git a/src/main/java/de/w665/sharepulse/service/FileService.java b/src/main/java/de/w665/sharepulse/service/FileService.java new file mode 100644 index 0000000..1edcd46 --- /dev/null +++ b/src/main/java/de/w665/sharepulse/service/FileService.java @@ -0,0 +1,110 @@ +package de.w665.sharepulse.service; + +import de.w665.sharepulse.db.repo.FileUploadRepository; +import de.w665.sharepulse.model.FileUpload; +import jakarta.annotation.PostConstruct; +import lombok.AllArgsConstructor; +import lombok.extern.slf4j.Slf4j; +import org.apache.tomcat.util.http.fileupload.FileUtils; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.beans.factory.annotation.Value; +import org.springframework.stereotype.Service; +import org.springframework.web.multipart.MultipartFile; + +import java.io.File; +import java.io.IOException; +import java.nio.file.Files; +import java.nio.file.Path; +import java.nio.file.Paths; +import java.util.Date; + +@Slf4j +@Service +public class FileService { + + private final FileIdService fileIdService; + + private final FileUploadRepository fileUploadRepository; + + @Value("${sharepulse.temp-filestore-path}") + private String tempDirPath; + + @Autowired + public FileService(FileIdService fileIdService, FileUploadRepository fileUploadRepository) { + this.fileIdService = fileIdService; + this.fileUploadRepository = fileUploadRepository; + } + + @PostConstruct + public void createFolder() { + Path path = Paths.get(getTempDirPath()); + log.debug("Checking temp file directory " + path.getFileName()); + try { + if (!Files.exists(path)) { + Files.createDirectory(path); + log.debug("Directory created"); + } else { + log.debug("Directory already exists. Clearing content."); + FileUtils.cleanDirectory(new File(getTempDirPath())); + log.debug("Directory content cleared."); + } + } catch (IOException e) { + e.printStackTrace(); + } + } + + public String processUploadedFile(MultipartFile file, String uploaderIp, String password, boolean singleDownload, String fileDescription) { + + String fileId = fileIdService.generateNewUniqueId(); + + FileUpload fileUpload = FileUpload.builder() + .fileId(fileId) + .fileName(file.getOriginalFilename()) + .fileSize(file.getSize()) + .singleDownload(singleDownload) + .uploadDate(new Date()) + .uploadedByIpAddress(uploaderIp) + .downloadCount(0) + .fileDescription(fileDescription) + .passwordProtected(password != null && !password.isEmpty()) + .downloadPassword(password) + .build(); + + System.out.println(fileUpload.toString()); + + fileUploadRepository.insertFileUpload(fileUpload); + + + // TODO: rename file to fileID + Path path = Paths.get(getTempDirPath() + File.separator + fileId); + try { + Files.write(path, file.getBytes()); + log.debug("File " + file.getOriginalFilename() + " written to " + path.getFileName()); + return fileId; + } catch (IOException e) { + e.printStackTrace(); + return null; + } + } + + public File getFileById(String fileId) { + return new File(getTempDirPath() + File.separator + fileId); + } + + public FileUpload getFileUploadByFileId(String fileId) { + return fileUploadRepository.retrieveFileUploadByFileId(fileId); + } + + public long updateDownloadCount(FileUpload fileUpload) { + fileUpload.setDownloadCount(fileUpload.getDownloadCount() + 1); + fileUploadRepository.updateFileUpload(fileUpload); + return fileUpload.getDownloadCount(); + } + + + private String getTempDirPath() { + return System.getProperty("user.dir") + tempDirPath; + } + + +} diff --git a/src/main/resources/application.properties b/src/main/resources/application.properties new file mode 100644 index 0000000..05b0892 --- /dev/null +++ b/src/main/resources/application.properties @@ -0,0 +1,24 @@ +# Application config +sharepulse.temp-filestore-path=/temp-filestore + +# Static path +spring.web.resources.static-locations=classpath:/static/browser/ + +# Rest API path prefix +spring.data.rest.base-path=/api/v1 + +# Upload size overwrite +spring.servlet.multipart.max-file-size=1GB +spring.servlet.multipart.max-request-size=1GB + +# Logging +logging.level.de.w665.sharepulse=DEBUG + +# Database +rethinkdb.host=localhost +rethinkdb.port=28015 +rethinkdb.database=sharepulse + +# Miscellaneous +server.port=80 +spring.application.name=sharepulse diff --git a/src/test/java/de/w665/sharepulse/SharepulseApplicationTests.java b/src/test/java/de/w665/sharepulse/SharepulseApplicationTests.java new file mode 100644 index 0000000..1bf6ffe --- /dev/null +++ b/src/test/java/de/w665/sharepulse/SharepulseApplicationTests.java @@ -0,0 +1,13 @@ +package de.w665.sharepulse; + +import org.junit.jupiter.api.Test; +import org.springframework.boot.test.context.SpringBootTest; + +@SpringBootTest +class SharepulseApplicationTests { + + @Test + void contextLoads() { + } + +}