tfb-groupme-bridge/tfbbridge/__init__.py

from flask import Flask, request, render_template, g, redirect
import requests
from .dbutils import get_and_init_db, get_db
from .groupme import group_sharing_url_to_id
import json
import uuid

app = Flask(__name__)
app.config.from_mapping(
    DATABASE_LOCATION = "tfbbridge.sqlite3",
    BOT_ID = "4b841d36f2a0788192b5375335"
)

@app.route("/")
def testing():
    return "testing"

@app.route("/add_bot", methods=["GET", "POST"])
def add_bot():
    if request.method == "GET":
        return render_template("add_new.html")
    elif request.method == "POST":
        try:
            url = request.form["chaturl"]
            try:
                group_id = group_sharing_url_to_id(url)
            except:
                return "Bad URL", 400

            db = get_db()
            try:
                db.execute(
                    "INSERT INTO child_groups (group_id) VALUES (?)",
                    (group_id,),
                )
                db.commit()
            except db.IntegrityError:
                return "This group is already registered."

            bot_id = app.config["BOT_ID"]
            print(bot_id)

            requests.post("https://api.groupme.com/v3/bots?token={}".format(bot_id))

            return url

        except KeyError:
            return "Bad URL", 400
    else:
        return "Invalid Method", 400

@app.route("/list_child_groups")
def list_child_groups():
    db = get_db()
    cursor = db.cursor()
    cursor.execute('SELECT group_id FROM child_groups')
    results = cursor.fetchall()
    groups = [i[0] for i in results]
    return groups

@app.route("/groupme/oauth_callback")
def oauth_callback():
    token = request.args.get("access_token")
    print(token)

    r = requests.get("https://api.groupme.com/v3/groups?token={}".format(token))
    data = r.json()["response"]
    data = [
        {
            "group_id": i["group_id"],
            "name": i["name"],
        }
    for i in data]

    return render_template("what_group.html", groups=data, token=token)

@app.route("/groupme/add_group", methods=["POST"])
def add_group():
    token = request.form["token"]
    target_group = request.form["to_add"]
    print(target_group)

    print(token, target_group)

    headers = {
        'Content-Type': 'application/json',
    }

    params = {
        'token': token,
    }

    json_data = {
        'bot': {
            'name': 'tfbbot',
            'group_id': target_group,
        },
    }

    r = requests.post('https://api.groupme.com/v3/bots', params=params, headers=headers, json=json_data)

    payload = r.json()
    bot_id = payload["response"]["bot_id"]

    return r.text

@app.route("/groupme/authorize")
def authorize():
    return redirect(
        "https://oauth.groupme.com/oauth/authorize?client_id=SgAPMhA6H1UuBE2QD2O5JE1BM6WnysRu6AP1ib6jQsqrH3QA",
    )

@app.route("/groupme/create_organization", methods=["GET", "POST"])
def create_organization():
    if request.method == "GET":
        return render_template("new_org.html")
    elif request.method == "POST":
        name = request.form["orgname"]

        db = get_db()
        try:
            db.execute(
                "INSERT INTO organizations (org_name, admin_url, addition_url) VALUES (?, ?, ?)",
                (name, uuid.uuid4().hex, uuid.uuid4().hex,),
            )
            db.commit()

            return "testing"
        except db.IntegrityError:
            return "This group is already registered."

@app.route("/groupme/list_orgs")
def list_orgs():
    db = get_db()
    cursor = db.cursor()
    cursor.execute('SELECT * FROM organizations')
    results = cursor.fetchall()
    results = [(i[0], i[1], i[2], i[3]) for i in results]
    print(results)
    return results

@app.route("/groupme/org_info/<org>")
def org_info(org):
    db = get_db()
    cursor = db.cursor()
    cursor.execute('SELECT id, org_name FROM organizations WHERE admin_url=?', (org,))
    org_id, name = cursor.fetchone()

    cursor.execute('SELECT group_name, group_id, bot_id, chan_type FROM channels WHERE organization=?', (org_id,))
    results = cursor.fetchall()
    results = [
        {
            "name": i[0],
            "group_id": i[1],
            "bot_id": i[2],
            "chan_type": i[3],
        }
    for i in results]
    return results

@app.route("/groupme/add_chan_to_org/<org>")
def add_chan_to_org(org):
    db = get_db()
    cursor = db.cursor()
    cursor.execute('SELECT id, org_name FROM organizations WHERE addition_url=?', (org,))
    org_id, name = cursor.fetchone()
    return redirect(
        "https://oauth.groupme.com/oauth/authorize?client_id=SgAPMhA6H1UuBE2QD2O5JE1BM6WnysRu6AP1ib6jQsqrH3QA&state=" + org,
    )