app.py

import re
import bcrypt
import flask_login
import json
import logging
import os
import pyrebase

import threading
from dotenv import load_dotenv
from helpers import *
from slack_bolt import App
from slack_bolt.adapter.flask import SlackRequestHandler
from flask import Flask, redirect, request, render_template, redirect, url_for

load_dotenv()

tars_token = os.environ.get("TARS_TOKEN")  # Bot user OAuth
tars_user_token = os.environ.get("TARS_USER_TOKEN")  # User OAuth
tars_admin = os.environ.get("TARS_ADMIN")  # Channel for testing
tars_secret = os.environ.get("TARS_SECRET")  # signing secret

firebase_api_key = os.environ.get("FIREBASE_API_KEY")  # Firebase API key
tars_fb_ad = os.environ.get("TARS_FB_AD")  # Firebase authDomain
tars_fb_url = os.environ.get("TARS_FB_URL")  # Firebase databaseURL
tars_fb_sb = os.environ.get("TARS_FB_SB")  # Firebase storageBucket
key_fb_tars = os.environ.get("KEY_FB_TARS")  # SHA used to access child

vineethv_id = os.environ.get("VINEETHV_ID")  # Sir's user ID
general = os.environ.get("GENERAL_ID")  # `general` channel ID
tars = os.environ.get("TARS_ID")  # TARS ID
vineeth_emailid = os.environ.get("VINEETH_EMAIL_ID")  # Sir's email ID

username = os.environ.get("USERNAME")  # Webpage login username
password = os.environ.get("PASSWORD").encode()  # Webpage login password
secret = os.environ.get("SECRET")  # Don't change

office_hours_form_url = os.environ.get(
    "OFFICE_HOURS_FORM"
)  # Google Form for office hours

config = {
    "apiKey": firebase_api_key,
    "authDomain": tars_fb_ad,
    "databaseURL": tars_fb_url,
    "storageBucket": tars_fb_sb,
}

firebase = pyrebase.initialize_app(config)
db = firebase.database()

app = App(
    token=tars_token,
    signing_secret=tars_secret,
)

flask_app = Flask(__name__)
flask_app.secret_key = secret
handler = SlackRequestHandler(app)

login_manager = flask_login.LoginManager()
login_manager.init_app(flask_app)
login_manager.login_view = "login"


class User(flask_login.UserMixin):
    pass


@flask_app.route("/slack/events", methods=["POST"])
def slack_events():
    return handler.handle(request)


@login_manager.user_loader
def load_user(id):
    if id != username:
        return
    user = User()
    user.id = id
    return user


@flask_app.route("/", methods=["GET"])
def index():
    return redirect("https://solarillionfoundation.org/")


@flask_app.route("/login", methods=["GET", "POST"])
def login():
    if request.method == "GET":
        data1 = {}
        data1["status"] = "Enter credentials."
        return render_template("login.html", data=data1)
    if request.method == "POST":
        username_form = request.form.get("username")
        password_form = request.form.get("password").encode()
        print("Password: %s\tPass Form: %s" % (password, password_form))
        if username_form == username and bcrypt.checkpw(password_form, password):
            user = User()
            user.id = username
            flask_login.login_user(user)
            data1 = {}
            data1["status"] = "Enter the details and submit."
            print("Validated user")
            return redirect(request.args.get("next") or url_for("index"))
        else:
            data1 = {}
            data1["status"] = "Incorrect credentials."
            print("Invalid user")
            return render_template("login.html", data=data1)


@flask_app.route("/logout")
@flask_login.login_required
def logout():
    flask_login.logout_user()
    return redirect(url_for("login"))


@flask_app.route("/interact", methods=["POST"])
def interact():
    payload = json.loads(request.form.get("payload"))
    thread = threading.Thread(target=interact_handler,
                              args=(app, db, key_fb_tars, payload,))
    thread.start()
    return "OK", 200


@app.event("app_mention")
def app_mention_function(event, say):
    thread = threading.Thread(
        target=app_mention_event_handler, args=(app, db, key_fb_tars, event,))
    thread.start()
    return "OK", 200


@app.message("request office hours")
def request_office_hours(message, say):
    admin = list(db.child(key_fb_tars).child("admin").get().val())
    if message["user"] not in admin:
        say("You're not allowed to do this!")
        return
    msg = "Sir, please fill your office hours in this form: " + office_hours_form_url
    app.client.chat_postMessage(channel=vineethv_id, text=msg)
    logging.info("Sent request to Sir!")


@app.message("remind office hours")
def remind_office_hours(message, say):
    admin = list(db.child(key_fb_tars).child("admin").get().val())

    if message["user"] not in admin:
        say("You're not allowed to do this!")
        return

    msg = (
        "Sir, if you haven't filled your office hours yet, please do so by 9 pm tonight. Here's the link to the form: "
        + office_hours_form_url
    )

    app.client.chat_postMessage(channel=vineethv_id, text=msg)
    logging.info("Sent reminder to Sir!")


@app.message("post office hours")
def post_office_hours(message, say):
    admin = list(db.child(key_fb_tars).child("admin").get().val())
    if message["user"] not in admin:
        say("You're not allowed to do this!")
        return
    data = db.child(key_fb_tars).child("officehours").get().val()
    message = "Sir's office hours for the week:\n"
    for item in data[1:]:
        item["start"] = reformat_time(item["start"])
        item["end"] = reformat_time(item["end"])
        message += item["days"] + ": " + \
            item["start"] + " - " + item["end"] + "\n"
    app.client.chat_postMessage(channel=general, text=message)


@app.message("book meeting")
def book_meeting(message, say):
    slack_id = message["user"]
    meetings = db.child(key_fb_tars).child("meetings").get().val()
    id = "0"
    if meetings is not None:
        for i in list(meetings):
            if slack_id in i:
                id = i
    if id == "0":
        id = slack_id + "_1"
    else:
        id = slack_id + "_" + str(int(id.split("_")[1]) + 1)
    lines = message['text'].lower().split("\n")
    meeting_description = " ".join(lines[0].split(" ")[2:])
    people = [app.client.users_info(
        user=slack_id).data["user"]["profile"]["email"]]
    people = people + [vineeth_emailid]
    people_slack = [slack_id, vineethv_id]
    if len(lines) == 2:
        attendees = lines[1].replace("@", "").replace(
            "<", "").replace(">", "").upper().split()
        people_slack += attendees
        attendees = list(map(lambda x: app.client.users_info(
            user=x).data["user"]["profile"]["email"], attendees))
        people = people + attendees
        db.child(key_fb_tars).child("bookings").child(id).set(
            {"meeting": meeting_description, "people": people, "people_slack": people_slack})
    say("The meeting has been booked!")


@app.message("show meeting")
def show_meeting(message, say):
    slack_id = message["user"]
    meetings = db.child(key_fb_tars).child("meetings").get().val()
    if meetings is not None:
        meetings = dict(meetings)
        count = 0
        meet_keys = list(meetings.keys())
        for meet in meet_keys:
            if slack_id in meet:
                count += 1
                item = db.child(key_fb_tars).child(
                    "meetings").child(meet).get().val()
                meeting_info = f'{meet.split("_")[1]}. {item["desc"]}, {meet_reformat_time(item["start"])}-{meet_reformat_time(item["end"])}\n Meet Link : {item["meet_link"]}\n'
                if count == 1:
                    say("*Meetings you've booked:*")
                say(meeting_info)
                meetings.pop(meet)
        invites = 0
        meet_keys = list(meetings.keys())
        for meet in meet_keys:
            if (slack_id in meetings[meet]["people"]):
                invites += 1
                item = db.child(key_fb_tars).child(
                    "meetings").child(meet).get().val()
                meeting_info = f'{meet.split("_")[1]}. {item["desc"]}, {meet_reformat_time(item["start"])}-{meet_reformat_time(item["end"])}\n Meet Link : {item["meet_link"]}\n'
                if invites == 1:
                    say("*Meetings you've been invited to:*")
                say(meeting_info)
                meetings.pop(meet)
        if count == 0 and invites == 0:
            say("You have no upcoming meetings!")
    else:
        say("You haven't booked any meetings this week!")


@app.message("cancel meeting")
def cancel_meeting(message, say):
    slack_id = message["user"]
    meetings = db.child(key_fb_tars).child("meetings").get().val()
    if meetings is None:
        say("You haven't booked any meetings!")
        return
    id = message["text"].lower().split(" ")[2]
    cancel = False
    for meet in meetings:
        if slack_id in meet and meet.split("_")[1] == id:
            db.child(key_fb_tars).child("cancels").update({meet: "cancel"})
            say(f"Meeting with ID {id} has been cancelled!")
            cancel = True
            break
    if not cancel:
        say("Sorry! You've entered the incorrect meeting ID. Verify the meeting number using `show meeting`.")


@app.event("message")
def handle_message_events(body):
    logging.info(body)


if __name__ == "__main__":
    flask_app.run(port=5000)