add expiry option, add cron container with automatic deletion of expired content

Niko · Niko · commit 80a70a0ef7ef · 2020-03-31T14:21:59.000+04:00
diff --git a/build/index.php b/build/index.php
@@ -1,6 +1,6 @@
 <?php
 if(!isset($_FILES['file'])) {
-	?>
+?>
 <html><head><meta http-equiv="content-type" content="text/html; charset=UTF-8" /><title>file sharing</title>
 <style type="text/css">
 a {
@@ -24,6 +24,30 @@
 	$newdir = substr(md5(uniqid()), 0, 12);
 	exec('mkdir '.$basedir.'/'.$newdir);
 	exec('mv '.$_FILES['file']['tmp_name'].' '.$basedir.'/'.$newdir.'/'.$newFName);
+        if (isset($_REQUEST['exp'])) {
+          preg_match('/^(\d+)([hdwmy]?)$/', $_REQUEST['exp'], $matches);
+          if ($matches[1] && $matches[2]) {
+            switch ($matches[2]) {
+              case 'h':
+                $hours = $matches[1];
+                break;
+              case 'd':
+                $hours = 24 * $matches[1];
+                break;
+              case 'w':
+                $hours = 24 * 7 * $matches[1];
+                break;
+              case 'm':
+                $hours = 24 * 30.5 * $matches[1];
+                break;
+              case 'y':
+                $hours = 24 * 365 * $matches[1];
+                break;
+            }
+            $expiry = mktime() + ($hours * 60 * 60);
+            exec('touch '.$basedir.'/'.$newdir.'/expires-at-'.$expiry);
+          }
+        }
 	if (stristr($_SERVER['HTTP_USER_AGENT'], 'curl')) {
                 echo "https://".$_SERVER['HTTP_HOST']."/dl/".$newdir."/".$newFName;
 		echo "\n";
diff --git a/cron/Dockerfile b/cron/Dockerfile
@@ -0,0 +1,13 @@
+FROM alpine:3
+
+RUN apk add bash dcron
+RUN mkdir -p /var/logs
+
+# 20 minute interval
+ENV EXECUTION_CRON_EXPRESSION */20 * * * *
+
+COPY entrypoint.sh /
+COPY run-cron.sh /
+RUN chmod +x /entrypoint.sh
+RUN chmod +x /run-cron.sh
+ENTRYPOINT ["/entrypoint.sh"]
diff --git a/cron/entrypoint.sh b/cron/entrypoint.sh
@@ -0,0 +1,13 @@
+#!/bin/bash
+
+trap 'exit 0' SIGTERM | tee -a /var/logs/cron.log
+
+cat > crontab.tmp << EOF
+$EXECUTION_CRON_EXPRESSION /run-cron.sh $@ | tee -a /var/logs/cron.log
+# An empty line is required at the end of this file for a valid cron file.
+EOF
+
+crontab "crontab.tmp" 
+rm -f "crontab.tmp" 
+/usr/sbin/crond -L /var/logs/cron.log
+tail -n 5000 -f /var/logs/cron.log
diff --git a/cron/run-cron.sh b/cron/run-cron.sh
@@ -0,0 +1,23 @@
+#!/bin/bash -eux
+
+DL_DIR=/var/www/dl
+
+if [ ! -d ${DL_DIR} ]; then
+  exit 0
+fi
+
+CURRENT_TS=$(date '+%s')
+find "${DL_DIR}" -type f -name "expires-at-*" | while read expires_file; do
+  # /var/www/dl/c1984d98fcc6/expires-at-1585729013
+  TS=$(basename ${expires_file} | sed -E 's/^expires-at-(\d+)$/\1/g')
+  if [ $(echo ${TS} | grep -q '^\d\+$') -gt 0 ]; then
+    continue;
+  fi
+  if (( CURRENT_TS > TS )); then
+    echo -en $(date -d @${CURRENT_TS})
+    echo -en " > "
+    echo -en $(date -d @${TS})
+    echo " >> DELETING $(dirname ${expires_file})"
+    rm -r $(dirname ${expires_file})
+  fi
+done
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -11,3 +11,8 @@ services:
       - file_storage:/var/www/dl
     ports:
       - 127.0.0.1:8023:80
+  cron:
+    build: cron/
+    restart: unless-stopped
+    volumes_from:
+      - server
