diff --git a/suchwowx/helpers.py b/suchwowx/helpers.py index 41e6526..a6c4105 100644 --- a/suchwowx/helpers.py +++ b/suchwowx/helpers.py @@ -1,6 +1,9 @@ +import ipfsApi from eth_account.messages import encode_defunct -from suchwowx.factory import w3 +from suchwowx.models import Meme +from suchwowx.factory import w3, db +from suchwowx import config def verify_signature(message, signature, public_address): @@ -10,3 +13,28 @@ def verify_signature(message, signature, public_address): return True else: return False + +def upload_to_ipfs(meme_id: str): + meme = Meme.query.get(meme_id) + if not meme: + return False + try: + full_path = f'{config.DATA_FOLDER}/uploads/{meme.file_name}' + client = ipfsApi.Client('127.0.0.1', 5001) + artwork_hashes = client.add(full_path) + artwork_hash = artwork_hashes[0]['Hash'] + print(f'[+] Uploaded artwork to IPFS: {artwork_hash}') + meta = { + 'name': meme.title, + 'description': meme.description, + 'image': f'ipfs://{artwork_hash}', + 'by': meme.creator_handle, + 'properties': { + 'creator': meme.creator_handle + } + } + meta_hash = client.add_json(meta) + print(f'[+] Uploaded metadata to IPFS: {meta_hash}') + return (meta_hash, artwork_hash) + except Exception as e: + return False diff --git a/suchwowx/models.py b/suchwowx/models.py index 0fe5bbc..e1e99fc 100644 --- a/suchwowx/models.py +++ b/suchwowx/models.py @@ -99,3 +99,6 @@ class Meme(db.Model): def __repr__(self): return str(f'meme-{self.id}') + + def get_fs_path(self): + return f'{config.DATA_FOLDER}/uploads/{self.file_name}' diff --git a/suchwowx/routes/meme.py b/suchwowx/routes/meme.py index 1b498cf..2c487a8 100644 --- a/suchwowx/routes/meme.py +++ b/suchwowx/routes/meme.py @@ -1,4 +1,4 @@ -from os import path +from os import path, remove from secrets import token_urlsafe from json import loads, dumps @@ -10,6 +10,7 @@ from requests.exceptions import HTTPError from web3 import Web3 from suchwowx.models import Meme, User +from suchwowx.helpers import upload_to_ipfs from suchwowx.factory import db from suchwowx import config @@ -36,8 +37,8 @@ def mod(): return redirect(url_for('meme.index')) memes = Meme.query.filter( Meme.meta_ipfs_hash == None - ).order_by(Meme.create_date.desc()) - return render_template('review.html', memes=memes) + ).order_by(Meme.create_date.asc()) + return render_template('mod.html', memes=memes) @bp.route('/publish', methods=['GET', 'POST']) def publish(): @@ -55,7 +56,7 @@ def publish(): flash(msg, 'error') if "file" in request.files: return '' - return redirect(url_for('meme.index')) + return redirect(url_for('meme.index') + '?ipfs_error=1') if "file" in request.files: if form_err: return '' @@ -70,38 +71,22 @@ def publish(): full_path = f'{config.DATA_FOLDER}/uploads/{filename}' file.save(full_path) try: - meta_hash = None - artwork_hash = None - if current_user.verified or current_user.is_moderator(): - client = ipfsApi.Client('127.0.0.1', 5001) - artwork_hashes = client.add(full_path) - print(artwork_hashes) - artwork_hash = artwork_hashes[0]['Hash'] - print(artwork_hash) - print(f'[+] Uploaded artwork to IPFS: {artwork_hash}') - meta = { - 'name': title, - 'description': description, - 'image': f'ipfs://{artwork_hash}', - 'by': creator, - 'properties': { - 'creator': creator - } - } - meta_hash = client.add_json(meta) - print(f'[+] Uploaded metadata to IPFS: {meta_hash}') - flash('Published new meme to IPFS', 'success') meme = Meme( file_name=filename, - meta_ipfs_hash=meta_hash, - meme_ipfs_hash=artwork_hash, title=title, description=description, creator_handle=creator ) db.session.add(meme) db.session.commit() - flash('Published new meme to database', 'success') + if current_user.verified or current_user.is_moderator(): + res = upload_to_ipfs(meme.id) + meme.meta_ipfs_hash = res[0] + meme.meme_ipfs_hash = res[1] + db.session.commit() + flash('Published new meme to local database and IPFS.', 'success') + else: + flash('Published new meme to database for review by moderators.', 'success') return redirect(url_for('meme.index')) except ConnectionError: flash('[!] Unable to connect to local ipfs', 'error') @@ -112,16 +97,56 @@ def publish(): meme=meme ) - @bp.route('/meme/') -def meme(meme_id): +def show(meme_id): meme = Meme.query.filter(Meme.id == meme_id).first() if not meme: return redirect('/') if not meme.meta_ipfs_hash and not current_user.is_authenticated: - flash('You need to be a moderator to view that meme', 'warning') + flash('You need to be a moderator to view that meme.', 'warning') return redirect(url_for('meme.index')) elif not meme.meta_ipfs_hash and not current_user.is_moderator(): - flash('You need to be a moderator to view that meme', 'warning') + flash('You need to be a moderator to view that meme.', 'warning') return redirect(url_for('meme.index')) return render_template('meme.html', meme=meme) + +@bp.route('/meme//') +def approve(meme_id, action): + if not current_user.is_authenticated: + flash('You need to be logged in to reach this page.', 'warning') + return redirect(url_for('meme.index')) + if not current_user.is_moderator(): + flash('You need to be a moderator to reach this page.', 'warning') + return redirect(url_for('meme.index')) + meme = Meme.query.get(meme_id) + if not meme: + flash('That meme does not exist.', 'warning') + return redirect(url_for('meme.index')) + if meme.meta_ipfs_hash != None: + flash('That meme already has been published to IPFS.', 'warning') + return redirect(url_for('meme.show', meme_id=meme.id)) + if action == 'approve': + res = upload_to_ipfs(meme.id) + if not res: + flash('Unable to post to IPFS, daemon may be offline.', 'error') + return redirect(url_for('meme.show', meme_id=meme.id)) + existing_meta_ipfs = Meme.query.filter(Meme.meta_ipfs_hash == res[0]).first() + existing_meme_ipfs = Meme.query.filter(Meme.meme_ipfs_hash == res[1]).first() + if existing_meta_ipfs or existing_meme_ipfs: + flash('Cannot use an existing IPFS hash for either metadata or memes.', 'warning') + return redirect(url_for('meme.show', meme_id=meme.id)) + meme.meta_ipfs_hash = res[0] + meme.meme_ipfs_hash = res[1] + db.session.commit() + flash('Published new meme to IPFS.', 'success') + elif action == 'deny': + # delete image + # delete from database + if path.exists(meme.get_fs_path()): + remove(meme.get_fs_path()) + db.session.delete(meme) + db.session.commit() + flash('Deleted image and removed meme from database.', 'success') + else: + flash('Unknown action.', 'warning') + return redirect(url_for('meme.mod')) diff --git a/suchwowx/templates/index.html b/suchwowx/templates/index.html index 1e47581..2369dbe 100644 --- a/suchwowx/templates/index.html +++ b/suchwowx/templates/index.html @@ -6,16 +6,14 @@
{% include 'includes/navbar.html' %} - {% if get_flashed_messages(with_categories=true) %} + {% if request.args.get('ipfs_error') %}

You can run your own local IPFS instance and maintain metadata and artwork, -
try installing the software and running the following:

$ ipfs daemon
-

{% endif %} {% if memes %} @@ -26,7 +24,7 @@
- + {% if meme.file_name.endswith('mp4') %}

Meme IPFS: {{ meme.meme_ipfs_hash }}

{% endif %}

Meme ID: {{ meme }}

+ {% if not meme.meta_ipfs_hash %} +
+
+
+ Approve +
+
+ Deny +
+
+ {% endif %}
{% endif %}
- + {% include 'includes/footer.html' %} diff --git a/suchwowx/templates/review.html b/suchwowx/templates/mod.html similarity index 96% rename from suchwowx/templates/review.html rename to suchwowx/templates/mod.html index ac9bdca..88e45ac 100644 --- a/suchwowx/templates/review.html +++ b/suchwowx/templates/mod.html @@ -14,7 +14,7 @@
- + {% if meme.file_name.endswith('mp4') %}