Initial commit

app/server/routes/api_doc.py

@@ -0,0 +1,9 @@
+# ------- 3rd party imports -------
+from flask import Blueprint, render_template
+
+api_doc_blueprint = Blueprint('api_doc_blueprint', __name__, template_folder='templates')
+
+
+@api_doc_blueprint.route('/api_doc')
+def api_doc():
+    return render_template('api_doc.html')

app/server/routes/error.py

@@ -0,0 +1,9 @@
+# ------- 3rd party imports -------
+from flask import Blueprint, render_template
+
+error_blueprint = Blueprint('error_blueprint', __name__, template_folder='templates')
+
+
+@error_blueprint.route('/error')
+def error():
+    return render_template('error.html')

app/server/routes/get_token.py

@@ -0,0 +1,9 @@
+# ------- 3rd party imports -------
+from flask import Blueprint, render_template
+
+get_token_blueprint = Blueprint('get_token_blueprint', __name__, template_folder='templates')
+
+
+@get_token_blueprint.route('/get_token')
+def get_token():
+    return render_template('get_token.html')

app/server/routes/index.py

@@ -0,0 +1,9 @@
+# ------- 3rd party imports -------
+from flask import render_template, Blueprint
+
+index_blueprint = Blueprint('index_blueprint', __name__, template_folder='templates')
+
+
+@index_blueprint.route("/")
+def index():
+    return render_template('index.html')

app/server/routes/internal/favicon.py

@@ -0,0 +1,13 @@
+# ------- standard library imports -------
+import os
+
+# ------- 3rd party imports -------
+from flask import Blueprint, send_from_directory
+
+app_blueprint = Blueprint('app_blueprint', __name__)
+
+
+@app_blueprint.route('/favicon.ico')
+def favicon():
+    return send_from_directory(os.path.join(app_blueprint.root_path, 'static'),
+                               'favicon.ico', mimetype='image/vnd.microsoft.icon')

app/server/routes/internal/redirect_to_url.py

@@ -0,0 +1,24 @@
+# ------- 3rd party imports -------
+from flask import Blueprint, redirect, url_for
+
+# ------- local imports -------
+from app.server.db.models import Url
+from app.server.db.extensions import db
+
+redirect_to_url_blueprint = Blueprint('redirect_to_url_blueprint', __name__)
+
+
+@redirect_to_url_blueprint.route('/<short_url>')
+def redirect_to_url(short_url):
+    """
+    This function will query the database with the short_url and will
+    redirect to the original url if it exist in the database.
+    """
+    url = Url.query.filter_by(short_url=short_url).first()
+
+    if url:
+        url.visits = url.visits + 1
+        db.session.commit()
+        return redirect(url.original_url)
+
+    return redirect(url_for('page_not_found_blueprint.page_not_found'))

app/server/routes/internal/send_verification_code.py

@@ -0,0 +1,50 @@
+# ------- standard library imports -------
+import re
+
+# ------- 3rd party imports -------
+from flask_mail import Mail, Message
+from flask import Blueprint, request, redirect, url_for, session
+
+# ------- local imports -------
+from app import app
+from app.server.db.extensions import db
+from app.server.db.models import VerificationCode, Email
+
+send_otp_blueprint = Blueprint('send_otp_blueprint', __name__)
+
+
+@send_otp_blueprint.route('/email_validation', methods=['POST'])
+def email_validation():
+    input_email = request.form['email']
+    pattern = r'[^@]+@[^@]+\.[^@]+'
+    is_email_valid = re.match(pattern, input_email)
+
+    if is_email_valid:
+        email = Email.query.filter_by(email=input_email).first()
+
+        if email and email.is_verified:
+            auth_token = Email.query.filter_by(email=input_email).first().auth_token
+            return redirect(url_for('your_api_token_blueprint.your_api_token', auth_token=auth_token))
+
+        elif email and not email.is_verified:
+            return redirect(url_for('verify_code_blueprint.enter_verification_code', is_verified=False))
+
+        else:
+            if not email:
+                session['user_email'] = input_email
+
+                email = Email(email=input_email)
+                verification_code = VerificationCode(email=email)
+                db.session.add(verification_code, email)
+                db.session.commit()
+
+                # print('user added to db')
+                # print('verification_code:')
+                # print(verification_code)
+                # print('*' * 33)
+
+                mail = Mail(app.app)
+                msg = Message('ShortMe Verification Code', sender='shortme.biz@gmail.com', recipients=[email.email])
+                msg.body = str(f'Hi!\nThis is your verification code: {verification_code.verification_code}')
+                mail.send(msg)
+                return redirect(url_for('verify_code_blueprint.enter_verification_code'))

app/server/routes/internal/shorten_url.py

@@ -0,0 +1,40 @@
+# ------- standard library imports -------
+import json
+import requests
+
+# ------- 3rd party imports -------
+import flask
+from flask import Blueprint, request, redirect, url_for
+
+# ------- local imports -------
+from app import app
+
+shorten_url_blueprint = Blueprint('shorten_url_blueprint', __name__, template_folder='templates')
+
+
+@shorten_url_blueprint.route('/shorten', methods=['POST'])
+def shorten_url():
+    base_url = flask.url_for("index_blueprint.index", _external=True)
+    original_url = request.form['original_url']
+    shorten_endpoint = base_url + 'api/shorten'
+
+    params = {
+        'url': original_url
+    }
+
+    headers = {
+        'Authorization': f'Bearer {app.app.secret_key}'
+    }
+
+    response = requests.post(shorten_endpoint, headers=headers, params=params)
+
+    if response.status_code == 200:
+        response = json.loads(response.text)
+        short_url = response['short_url']
+        original_url = response['original_url']
+
+        return redirect(url_for('your_short_url_blueprint.your_short_url',
+                                short_url=short_url,
+                                original_url=original_url))
+    else:
+        return redirect(url_for('error_blueprint.error'))

app/server/routes/page_not_found.py

@@ -0,0 +1,9 @@
+# ------- 3rd party imports -------
+from flask import Blueprint, render_template
+
+page_not_found_blueprint = Blueprint('page_not_found_blueprint', __name__, template_folder='templates')
+
+
+@page_not_found_blueprint.route('/page_not_found')
+def page_not_found():
+    return render_template('404.html')

app/server/routes/total_clicks.py

@@ -0,0 +1,24 @@
+# ------- standard library imports -------
+import json
+import requests
+
+# ------- 3rd party imports -------
+import flask
+from flask import Blueprint, render_template, request
+
+total_clicks_blueprint = Blueprint('total_clicks_blueprint', __name__, template_folder='templates')
+
+
+@total_clicks_blueprint.route('/total_clicks')
+def total_clicks():
+    short_url = request.args['short_url']
+    base_url = flask.url_for("index_blueprint.index", _external=True)
+    total_clicks_endpoint = base_url + 'api/total_clicks'
+
+    params = {
+        'url': short_url
+    }
+
+    response = requests.get(total_clicks_endpoint, params=params)
+    total_url_clicks = json.loads(response.text)['total']
+    return render_template('total-clicks.html', total_clicks=total_url_clicks)

app/server/routes/verify_code.py

@@ -0,0 +1,32 @@
+# ------- 3rd party imports -------
+from flask import Blueprint, render_template, request, redirect, url_for, session
+
+# ------- local imports -------
+from app.server.db.extensions import db
+from app.server.db.models import VerificationCode, AuthToken, Email
+
+verify_code_blueprint = Blueprint('verify_code_blueprint', __name__, template_folder='templates')
+
+
+@verify_code_blueprint.route('/verify', methods=['GET', 'POST'])
+def enter_verification_code():
+    is_verified = request.args.get('is_verified')
+    is_code_valid = request.args.get('is_code_valid')
+    return render_template('verify.html', is_code_valid=is_code_valid, is_verified=is_verified)
+
+
+@verify_code_blueprint.route('/validate_code', methods=['POST'])
+def validate_code():
+    input_code = request.form['verification']
+    code = VerificationCode.query.filter_by(verification_code=input_code).first()
+
+    if code:
+        email = Email.query.filter_by(email=session['user_email']).first()
+        email.is_verified = True
+        auth_token = AuthToken(email=email)
+        db.session.add(auth_token)
+        db.session.commit()
+        return redirect(url_for('your_api_token_blueprint.your_api_token', auth_token=auth_token))
+
+    else:
+        return redirect(url_for('verify_code_blueprint.enter_verification_code', is_code_valid=False))

app/server/routes/your_api_token.py

@@ -0,0 +1,15 @@
+# ------- 3rd party imports -------
+from flask import Blueprint, render_template, request, url_for
+
+your_api_token_blueprint = Blueprint('your_api_token_blueprint', __name__, template_folder='templates')
+
+
+@your_api_token_blueprint.route('/your_api_token')
+def your_api_token():
+    auth_token = request.args.get("auth_token")
+
+    if auth_token:
+        return render_template('your_api_token.html', auth_token=auth_token)
+
+    else:
+        return render_template(url_for('page_not_found_blueprint.page_not_found'))

app/server/routes/your_short_url.py

@@ -0,0 +1,19 @@
+# ------- 3rd party imports -------
+import flask
+from flask import Blueprint, render_template, request
+
+your_short_url_blueprint = Blueprint('your_short_url_blueprint', __name__, template_folder='templates')
+
+
+@your_short_url_blueprint.route('/your_short_url')
+def your_short_url():
+    original_url = request.args['original_url']
+    short_url = request.args['short_url']
+    base_url = flask.url_for("index_blueprint.index", _external=True)
+    full_short_url = f'{base_url}{short_url}'
+    full_short_url = full_short_url.replace('http://www.', '')
+
+    return render_template('your_short_url.html',
+                           original_url=original_url,
+                           short_url=short_url,
+                           full_short_url=full_short_url)