forked from spitkov/sxbin
added download all to folder view which zips then downloads refactored the whole sotring mechanisms now it stores it in a db NOT IN A FUCKING HTML FILE THATS LIKE SO OPTIMAL also you can add /raw to any text and get it raw uwu :3 major refactor :333
This commit is contained in:
parent
25c0f89486
commit
c1201dd7a6
199
app.py
199
app.py
@ -4,15 +4,73 @@ import shortuuid
|
||||
import os
|
||||
from datetime import datetime
|
||||
import zipfile
|
||||
import sqlite3
|
||||
import threading
|
||||
import time
|
||||
import shutil
|
||||
from datetime import timedelta
|
||||
|
||||
app = Flask(__name__)
|
||||
UPLOAD_FOLDER = './uploads'
|
||||
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
|
||||
DATABASE = 'data.db'
|
||||
|
||||
if not os.path.exists(UPLOAD_FOLDER):
|
||||
os.makedirs(UPLOAD_FOLDER)
|
||||
|
||||
data_store = {}
|
||||
# Database setup and helper functions
|
||||
def get_db():
|
||||
db = getattr(threading.current_thread(), '_database', None)
|
||||
if db is None:
|
||||
db = threading.current_thread()._database = sqlite3.connect(DATABASE)
|
||||
return db
|
||||
|
||||
def init_db():
|
||||
with app.app_context():
|
||||
db = get_db()
|
||||
with app.open_resource('schema.sql', mode='r') as f:
|
||||
db.cursor().executescript(f.read())
|
||||
db.commit()
|
||||
|
||||
@app.teardown_appcontext
|
||||
def close_connection(exception):
|
||||
db = getattr(threading.current_thread(), '_database', None)
|
||||
if db is not None:
|
||||
db.close()
|
||||
|
||||
# Initialize database
|
||||
init_db()
|
||||
|
||||
# Add this function to delete old files
|
||||
def delete_old_files():
|
||||
while True:
|
||||
db = get_db()
|
||||
cursor = db.cursor()
|
||||
|
||||
# Delete files older than 30 days
|
||||
thirty_days_ago = datetime.now() - timedelta(days=30)
|
||||
cursor.execute("SELECT vanity, type, data FROM content WHERE created_at < ?", (thirty_days_ago,))
|
||||
old_files = cursor.fetchall()
|
||||
|
||||
for vanity, content_type, data in old_files:
|
||||
if content_type == 'file':
|
||||
file_path = os.path.join(app.config['UPLOAD_FOLDER'], f'{vanity}_{data}')
|
||||
if os.path.exists(file_path):
|
||||
os.remove(file_path)
|
||||
elif content_type == 'folder':
|
||||
folder_path = os.path.join(app.config['UPLOAD_FOLDER'], vanity)
|
||||
if os.path.exists(folder_path):
|
||||
shutil.rmtree(folder_path)
|
||||
|
||||
cursor.execute("DELETE FROM content WHERE created_at < ?", (thirty_days_ago,))
|
||||
db.commit()
|
||||
|
||||
time.sleep(86400) # Sleep for 24 hours
|
||||
|
||||
# Start the cleanup thread
|
||||
cleanup_thread = threading.Thread(target=delete_old_files)
|
||||
cleanup_thread.daemon = True
|
||||
cleanup_thread.start()
|
||||
|
||||
@app.route('/')
|
||||
def index():
|
||||
@ -20,28 +78,36 @@ def index():
|
||||
|
||||
@app.route('/content/<vanity>')
|
||||
def content(vanity):
|
||||
target = data_store.get(vanity)
|
||||
db = get_db()
|
||||
cursor = db.cursor()
|
||||
cursor.execute("SELECT * FROM content WHERE vanity = ?", (vanity,))
|
||||
target = cursor.fetchone()
|
||||
|
||||
if target:
|
||||
if target['type'] == 'pastebin':
|
||||
return render_template('content.html', content=target['content'], created_at=target['created_at'])
|
||||
elif target['type'] == 'file':
|
||||
file_path = os.path.join(app.config['UPLOAD_FOLDER'], f'{vanity}_{target["filename"]}')
|
||||
content_type, content_data = target[1], target[2]
|
||||
if content_type == 'pastebin':
|
||||
return render_template('content.html', content=content_data, created_at=target[3])
|
||||
elif content_type == 'file':
|
||||
file_path = os.path.join(app.config['UPLOAD_FOLDER'], f'{vanity}_{content_data}')
|
||||
file_info = {
|
||||
'name': target['filename'],
|
||||
'name': content_data,
|
||||
'size': os.path.getsize(file_path),
|
||||
'modified_at': datetime.fromtimestamp(os.path.getmtime(file_path)).strftime('%Y-%m-%d %H:%M:%S'),
|
||||
'url': url_for('download_file', vanity=vanity)
|
||||
}
|
||||
return render_template('file.html', **file_info)
|
||||
elif target['type'] == 'url':
|
||||
return render_template('content.html', url=target['url'])
|
||||
elif content_type == 'url':
|
||||
return render_template('content.html', url=content_data)
|
||||
return 'Not Found', 404
|
||||
|
||||
@app.route('/download/<vanity>', methods=['GET'])
|
||||
def download_file(vanity):
|
||||
target = data_store.get(vanity)
|
||||
if target and target['type'] == 'file':
|
||||
filename = f'{vanity}_{target["filename"]}'
|
||||
db = get_db()
|
||||
cursor = db.cursor()
|
||||
cursor.execute("SELECT * FROM content WHERE vanity = ? AND type = 'file'", (vanity,))
|
||||
target = cursor.fetchone()
|
||||
if target:
|
||||
filename = f'{vanity}_{target[2]}'
|
||||
return send_from_directory(app.config['UPLOAD_FOLDER'], filename, as_attachment=True)
|
||||
return 'Not Found', 404
|
||||
|
||||
@ -50,12 +116,12 @@ def upload_pastebin():
|
||||
content = request.form['content']
|
||||
vanity = shortuuid.uuid()[:6]
|
||||
created_at = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
|
||||
data_store[vanity] = {'type': 'pastebin', 'content': content, 'created_at': created_at}
|
||||
|
||||
html_content = render_template('content.html', content=content, created_at=created_at)
|
||||
html_file_path = os.path.join('templates', f'{vanity}.html')
|
||||
with open(html_file_path, 'w') as f:
|
||||
f.write(html_content)
|
||||
|
||||
db = get_db()
|
||||
cursor = db.cursor()
|
||||
cursor.execute("INSERT INTO content (vanity, type, data, created_at) VALUES (?, ?, ?, ?)",
|
||||
(vanity, 'pastebin', content, created_at))
|
||||
db.commit()
|
||||
|
||||
return jsonify({'vanity': vanity})
|
||||
|
||||
@ -71,19 +137,13 @@ def upload_file():
|
||||
filename = secure_filename(file.filename)
|
||||
filepath = os.path.join(app.config['UPLOAD_FOLDER'], f'{vanity}_{filename}')
|
||||
file.save(filepath)
|
||||
data_store[vanity] = {'type': 'file', 'filename': filename}
|
||||
|
||||
file_info = {
|
||||
'name': filename,
|
||||
'size': os.path.getsize(filepath),
|
||||
'modified_at': datetime.fromtimestamp(os.path.getmtime(filepath)).strftime('%Y-%m-%d %H:%M:%S'),
|
||||
'url': url_for('download_file', vanity=vanity)
|
||||
}
|
||||
html_content = render_template('file.html', **file_info)
|
||||
html_file_path = os.path.join('templates', f'{vanity}.html')
|
||||
with open(html_file_path, 'w') as f:
|
||||
f.write(html_content)
|
||||
|
||||
|
||||
db = get_db()
|
||||
cursor = db.cursor()
|
||||
cursor.execute("INSERT INTO content (vanity, type, data) VALUES (?, ?, ?)",
|
||||
(vanity, 'file', filename))
|
||||
db.commit()
|
||||
|
||||
return jsonify({'vanity': vanity})
|
||||
|
||||
def save_file(file, folder_path):
|
||||
@ -115,7 +175,11 @@ def upload_folder():
|
||||
|
||||
handle_uploaded_folder(files, folder_path)
|
||||
|
||||
data_store[vanity] = {'type': 'folder', 'files': [file.filename for file in files]}
|
||||
db = get_db()
|
||||
cursor = db.cursor()
|
||||
cursor.execute("INSERT INTO content (vanity, type, data) VALUES (?, ?, ?)",
|
||||
(vanity, 'folder', ','.join([file.filename for file in files])))
|
||||
db.commit()
|
||||
|
||||
return jsonify({'vanity': vanity})
|
||||
|
||||
@ -123,51 +187,59 @@ def upload_folder():
|
||||
def shorten_url():
|
||||
original_url = request.form['url']
|
||||
vanity = shortuuid.uuid()[:6]
|
||||
data_store[vanity] = {'type': 'url', 'url': original_url}
|
||||
|
||||
html_content = f'<html><body><script>window.location.href="{original_url}";</script></body></html>'
|
||||
html_file_path = os.path.join('templates', f'{vanity}.html')
|
||||
with open(html_file_path, 'w') as f:
|
||||
f.write(html_content)
|
||||
|
||||
db = get_db()
|
||||
cursor = db.cursor()
|
||||
cursor.execute("INSERT INTO content (vanity, type, data) VALUES (?, ?, ?)",
|
||||
(vanity, 'url', original_url))
|
||||
db.commit()
|
||||
|
||||
return jsonify({'vanity': vanity})
|
||||
|
||||
@app.route('/<vanity>', methods=['GET'])
|
||||
def redirect_vanity(vanity):
|
||||
target = data_store.get(vanity)
|
||||
db = get_db()
|
||||
cursor = db.cursor()
|
||||
cursor.execute("SELECT * FROM content WHERE vanity = ?", (vanity,))
|
||||
target = cursor.fetchone()
|
||||
|
||||
if target:
|
||||
if target['type'] == 'pastebin':
|
||||
return render_template(f'{vanity}.html')
|
||||
elif target['type'] == 'file':
|
||||
file_path = os.path.join(app.config['UPLOAD_FOLDER'], f'{vanity}_{target["filename"]}')
|
||||
content_type, content_data = target[1], target[2]
|
||||
if content_type == 'pastebin':
|
||||
return render_template('content.html', content=content_data, created_at=target[3])
|
||||
elif content_type == 'file':
|
||||
file_path = os.path.join(app.config['UPLOAD_FOLDER'], f'{vanity}_{content_data}')
|
||||
file_info = {
|
||||
'name': target['filename'],
|
||||
'name': content_data,
|
||||
'size': os.path.getsize(file_path),
|
||||
'modified_at': datetime.fromtimestamp(os.path.getmtime(file_path)).strftime('%Y-%m-%d %H:%M:%S'),
|
||||
'url': url_for('download_file', vanity=vanity)
|
||||
}
|
||||
return render_template('file.html', **file_info)
|
||||
elif target['type'] == 'folder':
|
||||
elif content_type == 'folder':
|
||||
return redirect(url_for('folder_content', vanity=vanity))
|
||||
elif target['type'] == 'url':
|
||||
return render_template('content.html', url=target['url'])
|
||||
elif content_type == 'url':
|
||||
return render_template('content.html', url=content_data)
|
||||
return render_template('404.html'), 404
|
||||
|
||||
@app.route('/<vanity>/raw', methods=['GET'])
|
||||
def raw_vanity(vanity):
|
||||
target = data_store.get(vanity)
|
||||
db = get_db()
|
||||
cursor = db.cursor()
|
||||
cursor.execute("SELECT * FROM content WHERE vanity = ? AND type = 'pastebin'", (vanity,))
|
||||
target = cursor.fetchone()
|
||||
|
||||
if target:
|
||||
if target['type'] == 'pastebin':
|
||||
return render_template(f'{vanity}raw.html')
|
||||
|
||||
return render_template('404.html'), 404
|
||||
return target[2], 200, {'Content-Type': 'text/plain; charset=utf-8'}
|
||||
return 'Not Found', 404
|
||||
|
||||
@app.route('/folder/<vanity>', methods=['GET'])
|
||||
def folder_content(vanity):
|
||||
target = data_store.get(vanity)
|
||||
if target and target['type'] == 'folder':
|
||||
db = get_db()
|
||||
cursor = db.cursor()
|
||||
cursor.execute("SELECT * FROM content WHERE vanity = ? AND type = 'folder'", (vanity,))
|
||||
target = cursor.fetchone()
|
||||
if target:
|
||||
folder_path = os.path.join(app.config['UPLOAD_FOLDER'], vanity)
|
||||
files = []
|
||||
for root, _, filenames in os.walk(folder_path):
|
||||
@ -188,14 +260,27 @@ def folder_content(vanity):
|
||||
prev_url = url_for('folder_content', vanity=vanity, page=page-1) if page > 1 else None
|
||||
next_url = url_for('folder_content', vanity=vanity, page=page+1) if end < total_files else None
|
||||
|
||||
return render_template('folder.html', files=files, prev_url=prev_url, next_url=next_url)
|
||||
download_all_url = url_for('download_folder_as_zip', vanity=vanity)
|
||||
|
||||
# Get the current folder name
|
||||
current_folder = os.path.basename(folder_path)
|
||||
|
||||
return render_template('folder.html',
|
||||
files=files,
|
||||
prev_url=prev_url,
|
||||
next_url=next_url,
|
||||
download_all_url=download_all_url,
|
||||
current_folder=current_folder)
|
||||
|
||||
return 'Not Found', 404
|
||||
|
||||
@app.route('/folder/<vanity>/download', methods=['GET'])
|
||||
def download_folder_as_zip(vanity):
|
||||
target = data_store.get(vanity)
|
||||
if target and target['type'] == 'folder':
|
||||
db = get_db()
|
||||
cursor = db.cursor()
|
||||
cursor.execute("SELECT * FROM content WHERE vanity = ? AND type = 'folder'", (vanity,))
|
||||
target = cursor.fetchone()
|
||||
if target:
|
||||
folder_path = os.path.join(app.config['UPLOAD_FOLDER'], vanity)
|
||||
zip_path = os.path.join(app.config['UPLOAD_FOLDER'], f'{vanity}.zip')
|
||||
|
||||
|
6
schema.sql
Normal file
6
schema.sql
Normal file
@ -0,0 +1,6 @@
|
||||
CREATE TABLE IF NOT EXISTS content (
|
||||
vanity TEXT PRIMARY KEY,
|
||||
type TEXT NOT NULL,
|
||||
data TEXT NOT NULL,
|
||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||||
);
|
@ -3,37 +3,43 @@
|
||||
<head>
|
||||
<meta charset="UTF-8">
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||
<title>Folder</title>
|
||||
<title>Folder Contents</title>
|
||||
<style>
|
||||
body {
|
||||
font-family: Arial, sans-serif;
|
||||
background-color: #121212;
|
||||
color: #e0e0e0;
|
||||
margin: 0;
|
||||
padding: 0;
|
||||
display: flex;
|
||||
justify-content: center;
|
||||
align-items: center;
|
||||
height: 100vh;
|
||||
padding: 20px;
|
||||
}
|
||||
.container {
|
||||
background: #1e1e1e;
|
||||
border-radius: 8px;
|
||||
box-shadow: 0 0 10px rgba(0, 0, 0, 0.3);
|
||||
padding: 20px;
|
||||
width: 400px;
|
||||
text-align: center;
|
||||
max-width: 800px;
|
||||
margin: 0 auto;
|
||||
}
|
||||
h1 {
|
||||
margin-bottom: 20px;
|
||||
color: #f5f5f5;
|
||||
text-align: center;
|
||||
}
|
||||
ul {
|
||||
.folder-structure {
|
||||
background: #2a2a2a;
|
||||
border-radius: 4px;
|
||||
padding: 10px;
|
||||
margin-bottom: 20px;
|
||||
}
|
||||
.file-list {
|
||||
list-style-type: none;
|
||||
padding: 0;
|
||||
}
|
||||
li {
|
||||
.file-list li {
|
||||
margin: 10px 0;
|
||||
padding: 5px;
|
||||
background: #333;
|
||||
border-radius: 4px;
|
||||
}
|
||||
a {
|
||||
color: #61afef;
|
||||
@ -42,16 +48,56 @@
|
||||
a:hover {
|
||||
text-decoration: underline;
|
||||
}
|
||||
.download-all-btn {
|
||||
display: block;
|
||||
width: 200px;
|
||||
margin: 20px auto;
|
||||
padding: 10px;
|
||||
background-color: #4CAF50;
|
||||
color: white;
|
||||
text-align: center;
|
||||
text-decoration: none;
|
||||
border-radius: 5px;
|
||||
font-weight: bold;
|
||||
}
|
||||
.download-all-btn:hover {
|
||||
background-color: #45a049;
|
||||
}
|
||||
.pagination {
|
||||
text-align: center;
|
||||
margin-top: 20px;
|
||||
}
|
||||
.pagination a {
|
||||
margin: 0 10px;
|
||||
}
|
||||
</style>
|
||||
</head>
|
||||
<body>
|
||||
<div class="container">
|
||||
<h1>Folder</h1>
|
||||
<ul>
|
||||
<h1>Folder Contents</h1>
|
||||
|
||||
<div class="folder-structure">
|
||||
<!-- You can add a tree-like structure here if needed -->
|
||||
<p>Current folder: /{{ current_folder }}</p>
|
||||
</div>
|
||||
|
||||
<ul class="file-list">
|
||||
{% for file in files %}
|
||||
<li><a href="{{ file.url }}">{{ file.name }}</a></li>
|
||||
<li><a href="{{ file.url }}">{{ file.name }}</a></li>
|
||||
{% endfor %}
|
||||
</ul>
|
||||
|
||||
<a href="{{ download_all_url }}" class="download-all-btn">Download All</a>
|
||||
|
||||
<div class="pagination">
|
||||
{% if prev_url %}
|
||||
<a href="{{ prev_url }}">Previous</a>
|
||||
{% endif %}
|
||||
|
||||
{% if next_url %}
|
||||
<a href="{{ next_url }}">Next</a>
|
||||
{% endif %}
|
||||
</div>
|
||||
</div>
|
||||
</body>
|
||||
</html>
|
||||
|
BIN
uploads/.DS_Store
vendored
Normal file
BIN
uploads/.DS_Store
vendored
Normal file
Binary file not shown.
Binary file not shown.
Before Width: | Height: | Size: 18 KiB |
Loading…
Reference in New Issue
Block a user