app.py

from flask import Flask, render_template, request, redirect, url_for, jsonify, session, abort
from authlib.integrations.flask_client import OAuth
import config
import databases
from dotenv import load_dotenv

import os 

load_dotenv()

themes = ['3024-day', '3024-night', 'abbott', 'abcdef', 'ambiance-mobile', 'ambiance', 'ayu-dark', 
          'ayu-mirage', 'base16-dark', 'base16-light', 'bespin', 'blackboard', 'cobalt', 'colorforth', 
          'darcula', 'dracula', 'duotone-dark', 'duotone-light', 'eclipse', 'elegant', 'erlang-dark', 
          'gruvbox-dark', 'hopscotch', 'icecoder', 'idea', 'isotope', 'juejin', 'lesser-dark', 'liquibyte', 
          'lucario', 'material-darker', 'material-ocean', 'material-palenight', 'material', 'mbo', 'mdn-like', 
          'midnight', 'monokai', 'moxer', 'neat', 'neo', 'night', 'nord', 'oceanic-next', 'panda-syntax', 
          'paraiso-dark', 'paraiso-light', 'pastel-on-dark', 'railscasts', 'rubyblue', 'seti', 'shadowfox', 
          'solarized', 'ssms', 'the-matrix', 'tomorrow-night-bright', 'tomorrow-night-eighties', 'ttcn', 
          'twilight', 'vibrant-ink', 'xq-dark', 'xq-light', 'yeti', 'yonce', 'zenburn']

languages = ['apl', 'asciiarmor', 'asn.1', 'asterisk', 'brainfuck', 'clike', 'clojure', 'cmake', 'cobol', 
             'coffeescript', 'commonlisp', 'crystal', 'css', 'cypher', 'd', 'dart', 'diff', 'django', 
             'dockerfile', 'dtd', 'dylan', 'ebnf', 'ecl', 'eiffel', 'elm', 'erlang', 'factor', 'fcl', 'forth', 
             'fortran', 'gas', 'gfm', 'gherkin', 'go', 'groovy', 'haml', 'handlebars', 'haskell', 
             'haskell-literate', 'haxe', 'htmlembedded', 'htmlmixed', 'http', 'idl', 'javascript', 'jinja2', 
             'jsx', 'julia', 'livescript', 'lua', 'markdown', 'mathematica', 'mbox', 'mirc', 'mllike', 'modelica', 
             'mscgen', 'mumps', 'nginx', 'nsis', 'ntriples', 'octave', 'oz', 'pascal', 'pegjs', 'perl', 'php', 
             'pig', 'powershell', 'properties', 'protobuf', 'pug', 'puppet', 'python', 'q', 'r', 'rpm', 'rst', 
             'ruby', 'rust', 'sas', 'sass', 'scheme', 'shell', 'sieve', 'slim', 'smalltalk', 'smarty', 'solr', 
             'soy', 'sparql', 'spreadsheet', 'sql', 'stex', 'stylus', 'swift', 'tcl', 'textile', 'tiddlywiki', 
             'tiki', 'toml', 'tornado', 'troff', 'ttcn', 'ttcn-cfg', 'turtle', 'twig', 'vb', 'vbscript', 'velocity', 
             'verilog', 'vhdl', 'vue', 'wast', 'webidl', 'xml', 'xquery', 'yacas', 'yaml', 'yaml-frontmatter', 'z80']

app = Flask(__name__)
oauth = OAuth(app)

app.config.from_mapping(dict(os.environ))
app.secret_key = os.getenv("secret_key")
database = databases.Database(app.config['MONGO_URI'])

@app.route('/')
def index():
    if 'user' in session:
        snips = database.get10RandomSnips()
        return render_template('browse.html', snips=snips)
    return render_template('index.html')

@app.route('/auth/google/')
def google():
    GOOGLE_CLIENT_ID = config.GOOGLE_CLIENT_ID
    GOOGLE_CLIENT_SECRET = config.GOOGLE_CLIENT_SECRET
    
    CONF_URL = 'https://accounts.google.com/.well-known/openid-configuration'
    oauth.register(
        name='google',
        client_id=GOOGLE_CLIENT_ID,
        client_secret=GOOGLE_CLIENT_SECRET,
        server_metadata_url=CONF_URL,
        client_kwargs={
            'scope': 'openid email profile'
        }
    )
    redirect_uri = url_for('google_auth', _external=True)
    return oauth.google.authorize_redirect(redirect_uri)

@app.route('/auth/google/callback')
def google_auth():
    token = oauth.google.authorize_access_token()
    prof = token['userinfo']
    user = database.getUser(prof['email'])
    if user is None:
        user = database.addUser(prof['email'], prof['name'])
    session['user'] = database.getUser(prof['email'])
    return redirect('/home')

@app.route('/home')
def home():
    if 'user' in session:
        snips = database.getUserSnips(session['user']['email'])
        session['user']['snips'] = snips
        return render_template('home.html', user = session['user'])
    return redirect(url_for('index'))

@app.route('/new')
def new():
    if 'user' in session:
        return render_template('new.html', user = session['user'], themes = themes, langs = languages)
    return redirect(url_for('index'))
        

@app.route('/logout')
def logout():
    session.pop('user', None)
    return redirect(url_for('index'))

@app.route('/changeName', methods=['POST'])
def changeName():
    if 'user' in session:
        name = request.form.get('name')
        user = session['user']
        user['username'] = name
        session['user'] = user
        database.updateName(user['email'], name)
        return "Updated!"
    return abort(404)

@app.route('/uploadSnip', methods=['POST'])
def uploadSnip():
    if 'user' in session:
        name = request.form.get('name')
        description = request.form.get('description')
        code = request.form.get('code')
        language = request.form.get('language')
        theme = request.form.get('theme')
        response = database.uploadSnip(session['user']['email'], code, name, description, language, theme)
        return redirect(f'/snip/{response}')

@app.route('/snip/<id>', methods=['GET'])
def snip(id):
    if 'user' in session:
        return render_template('snip.html', snip = database.getSnip(id), themes = themes, langs = languages, user = session['user'])
    return render_template('snip.html', snip = database.getSnip(id), themes = themes, langs = languages, user = None)

@app.route('/user/<id>')
def user(id):
    if id == session['user']['_id']:
        return redirect(url_for('home'))
    if user := database.getUserWithId(id):
        snips = database.getUserSnips(user['email'])
        user['snips'] = snips
        return render_template('user.html', user = user)
    else:
        return abort(404)

@app.route('/save', methods=['POST'])
def save():
    if 'user' in session:
        snipID = request.args.get('snip')
        if snip := database.getSnip(snipID):
            database.saveSnip(snipID, session['user']['email'])
            session['user'] = database.getUser(session['user']['email'])
            return "ok"
        return abort(404)
    return abort(404)

@app.route('/remove', methods=['POST'])
def remove():
    if 'user' in session:
        snipID = request.args.get('snip')
        if snipID in session['user']['saves']:
            database.removeSnip(snipID, session['user']['email'])
            session['user'] = database.getUser(session['user']['email'])
            return "ok"
        return abort(404)
    return abort(404)

@app.route('/saves')
def saves():
    if 'user' not in session:
        return redirect(url_for('index'))
        # return jsonify(session['user']['saves'])
    snips = [database.getSnip(snip) for snip in session['user']['saves']]
    session['user']['saves'] = snips
    return render_template('saves.html', user = session['user'])
    
@app.route('/delete', methods=['POST'])
def delete():
    if 'user' in session:
        snipID = request.args.get('snip')
        if snipID in session['user']['snips']:
            database.delteSnip(snipID, session['user']['email'])
            session['user'] = database.getUser(session['user']['email'])
            return redirect(url_for('home'))
        return abort(404)
    return abort(404)

    
if __name__ == '__main__':
    app.run(debug=True)