1 files changed, 9 insertions, 13 deletions
diff --git a/app.py b/app.py
index 19410c7..f94514a 100644
--- a/app.py
+++ b/app.py
@@ -5,10 +5,12 @@ from flask import Flask, request, jsonify, send_from_directory
 
 app = Flask(__name__)
 app.config['UPLOAD_FOLDER'] = os.environ.get('UPLOAD_FOLDER', './uploads')
-app.config['MAX_CONTENT_LENGTH'] = 5 * 1024 * 1024  # 5MB
+app.config['MAX_CONTENT_LENGTH'] = int(os.environ.get('MAX_CONTENT_LENGTH', '5')) * 1024 * 1024  # in MB
 
 VALID_FILENAME_REGEX = r'^[a-zA-Z0-9\-_\.]+$'
 
+UPLOAD_TOKEN = os.environ.get('UPLOAD_TOKEN')
+
 def is_valid_filename(filename):
     return bool(re.match(VALID_FILENAME_REGEX, filename))
 
@@ -17,6 +19,12 @@ def upload_file():
     if 'file' not in request.files:
         return jsonify({'error': 'No file part in the request'}), 400
 
+    if 'token' not in request.headers:
+        return jsonify({'error': 'No token supplied'}), 401
+
+    if request.headers['token'] != UPLOAD_TOKEN:
+        return jsonify({'error': 'Invalid token supplied'}), 401
+
     file = request.files['file']
     if file.filename == '':
         return jsonify({'error': 'No file selected for upload'}), 400
@@ -35,18 +43,6 @@ def download_file(filename):
     except FileNotFoundError:
         return jsonify({'error': 'File not found'}), 404
 
-@app.route('/list', methods=['GET'])
-def list_files():
-    files = []
-    total_size = 0
-    for filename in os.listdir(app.config['UPLOAD_FOLDER']):
-        path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
-        if os.path.isfile(path):
-            size = os.path.getsize(path)
-            files.append({'filename': filename, 'size': size})
-            total_size += size
-    return jsonify({'files': files, 'count': len(files), 'total_size': total_size})
-
 if __name__ == '__main__':
     app.run(debug=True, host='0.0.0.0', port=int(os.environ.get('PORT', 5000)))