MsRewards/Flask/app.py

from time import sleep
import subprocess
import os
from flask import Flask, Response, redirect, url_for, request, session, abort, render_template, send_from_directory
from apscheduler.schedulers.background import BackgroundScheduler
from apscheduler.triggers.cron import CronTrigger
from flask_login import LoginManager, UserMixin, login_required, login_user, logout_user 
from werkzeug.utils import secure_filename
import json
import re
from requests import get

global password
with open("/app/MsRewards-Reborn/user_data/flask.json", "r") as inFile:
    data = json.load(inFile)

password = data["password"]
secret = data["secret"]
if secret == "":
    import secrets
    secret = secrets.token_hex()
    with open("/app/MsRewards-Reborn/user_data/flask.json", "w") as inFile:
        data = {
            "password": password,
            "secret": secret
        }
        json.dump(data, inFile)

"""
#Automatic start of MsRewards
"""
def daily_command():
    subprocess.Popen(["git",'pull'])
    subprocess.Popen(["pkill -9 chrome"])
    subprocess.Popen(["pkill -9 Xvfb"])
    subprocess.Popen(["pkill -9 undetected_chromedriver"])


scheduler = BackgroundScheduler()
scheduler.start()
scheduler.add_job(                  # on relance le job
    daily_command,                       # ---
    trigger=CronTrigger(
        year="*", month="*", day="*", hour="0", minute="0", second="0"
    ),         # ---
    name="Daily refresh",             # ---
    id="99"                          # ---
)

def start_ms(i):
    print("\033[32m" + f"Starting config {i}" + "\033[0m")
    log = open(f"/app/MsRewards-Reborn/user_data/logs/{i}.txt", 'a')  # so that data written to it will be appended
    subprocess.Popen([f"python3 -u /app/MsRewards-Reborn/V6.py -c {i}"], stdout=log, stderr=log, shell=True)
    log.close()




TriggerDict = {}
def update_jobs():
    with open("/app/MsRewards-Reborn/user_data/configs.json", "r") as inFile:
        configs = json.load(inFile)
    for i in configs:
        h, m = configs[i]["time"].split(":")
        print("\033[36m" + f"config {i} : {h}:{m}" + "\033[0m")
        TriggerDict[i] = CronTrigger(
                            year="*", month="*", day="*", hour=h, minute=m, second="0"
                        )
        if configs[i]["enabled"]:
            try :
                scheduler.remove_job(i)             # on reset le job
            except Exception as e:
                print(f"\033[33merror with deleting config {i} : {e}\033[0m")

            try :
                scheduler.add_job(                  # on relance le job
                    start_ms,                       # ---
                    trigger=TriggerDict[i],         # ---
                    args=[i],                       # ---
                    name="Daily start",             # ---
                    id=i                            # ---
                )
                print("\033[36m" + f"successfully created config {i}" + "\033[0m")
            except Exception as e:
                print(f"\033[33merror with creating config {i} : {e}\033[0m")
        else :
            try :
                scheduler.remove_job(i)
            except Exception as e :
                print(f"\033[33merror with deleting config {i} : {e}\033[0m")

def edit_version():
    with open("/app/MsRewards-Reborn/version", "r") as f:
        version = f.readline().replace("\n", '')
    f = open("/app/MsRewards-Reborn/Flask/templates/base.html", "r")
    txt = f.readlines()
    f.close()
    f = open("/app/MsRewards-Reborn/Flask/templates/base.html", "w")
    for i in txt:
        i = re.sub('<div class="footer">([^<]+)</div>', f'<div class="footer">{version}</div>', i)
        f.write(i)
    f.close()


"""
#Flask app
"""

app = Flask(__name__)

"""
#Login stuff
"""
# config
app.config["TEMPLATES_AUTO_RELOAD"] = True
app.config.update(
    SECRET_KEY = secret
)

login_manager = LoginManager()
login_manager.init_app(app)
login_manager.login_view = "login"

# silly user model
class User(UserMixin):
    def __init__(self, id):
        self.id = id
        self.name = "user" + str(id)
        self.password = password

    def __repr__(self):
        return "%d/%s/%s" % (self.id, self.name, self.password)

users = [User(1)]

@app.route("/login/", methods=["GET", "POST"])
def login():
    if request.method == 'POST':
        if request.form['password'] == password:
            user = User(id)
            login_user(user)
            if password == "ChangeMe":
                return(render_template("change_password.html"))
            return(redirect('override'))
        else:
            return abort(401)
    else:
        return(render_template("login.html"))


@app.route("/change_password/", methods=["GET", "POST"])
@login_required
def change_password():
    global password
    if request.method == 'POST':
        password = request.form["password"]
        with open("/app/MsRewards-Reborn/user_data/flask.json", "w") as inFile:
            data = {
                "password": password, 
                "secret": secret
            }
            json.dump(data, inFile)   
    return(render_template("override.html"))


# handle login failed
@app.errorhandler(401)
def unauthorized(e):
    return(render_template("login.html"))


# callback to reload the user object
@login_manager.user_loader
def load_user(userid):
    return User(userid)

"""
#end of login stuff
"""

@app.route("/", methods=["post"])
def dev():
    action = request.form
    print(action)
    if action == "dev":
        print("dev action test")
    return(f"<h1> TO IMPLEMENT - {action}</h1>")


@app.route("/")
def main():
    with open("/app/MsRewards-Reborn/user_data/configs.json", "r") as inFile:
        configs = json.load(inFile)
    return(render_template("override.html", data=configs))


@app.route("/discord/")
def discord_get():
    with open("/app/MsRewards-Reborn/user_data/discord.json", "r") as inFile:
        data = json.load(inFile)
    return(render_template("discord.html", data=data, len=maxi(data)))


@app.route("/discord/", methods=["post"])
def discord_post():
    with open("/app/MsRewards-Reborn/user_data/discord.json", "r") as inFile:
        data = json.load(inFile)    
    action = request.form
    if action['DISCORD'] == "delete" :
        data.pop(action["select"], None)
    else :
        config = action["select"]
        successL = action["successL"]
        try :
            a = action["successT"] 
            successT = "True"
        except:
            successT = "False"
        try :
            a = action["errorsT"] 
            errorsT = "True"
        except:
            errorsT = "False"
        errorsL = action["errorsL"]
        name = action["name"] if action["name"] else f"unnamed{action['select']}"
        data[config] = {"errorsL" : errorsL, "errorsT": errorsT, "successT": successT, "successL": successL, "name": name}

    with open("/app/MsRewards-Reborn/user_data/discord.json", "w") as outFile:
        json.dump(data, outFile)
    return(render_template("discord.html", data=data, len=maxi(data)))


@app.route("/dev/")
def dev2():
    with open("/app/MsRewards-Reborn/user_data/proxy.json", "r") as inFile:
        j = json.load(inFile)
    new_proxy = {"address": "ADDRESS", "port": "PORT", "name":"NAME"}
    max_index = 0
    for i in range(1, 50):
        try :
            print(j[str(i)])
        except :
            print(f"found {i - 1} proxys")
            max_index = i
            break
    j[f"{max_index}"] = new_proxy
    print(j)
    with open("/app/MsRewards-Reborn/user_data/proxy.json", "w") as outfile:
        json.dump(j, outfile)
    return(render_template("dev.html"))


@app.route("/settings/")
def settings_get():
    with open("/app/MsRewards-Reborn/user_data/settings.json", "r") as inFile:
        settings = json.load(inFile)
    return(render_template("settings.html", data=settings))


@app.route("/settings/", methods=["post"])
def settings_post():
    settings = {}
    action = request.form
    settings['avatarlink'] = action["avatarlink"]
    with open("/app/MsRewards-Reborn/user_data/settings.json", "w") as inFile:
        json.dump(settings, inFile)
    return(render_template("settings.html", data=settings))


@app.route("/proxy/")
def proxy_get():
    with open("/app/MsRewards-Reborn/user_data/proxy.json", "r") as inFile:
        j = json.load(inFile)
    return(render_template("proxy.html", data=j, len=maxi(j)))


@app.route("/proxy/", methods=["post"])
def proxy_post():
    with open("/app/MsRewards-Reborn/user_data/proxy.json", "r") as inFile:
        data = json.load(inFile)    
    action = request.form
    print(action)
    if action['PROXY'] == "delete" :
        print(action)
        data.pop(action["select"], None)
    else :
        try :
            config = action["select"]
            address = action["address"]
            port = action["port"]
            name = action["name"] if action["name"] else f"@unnamed{action['select']}"
            data[config] = {"address" : address, "port": port, "name": name}
        except :
            print("error : probably bad config")

    with open("/app/MsRewards-Reborn/user_data/proxy.json", "w") as outFile:
        json.dump(data, outFile)
    return(render_template("proxy.html", data=data, len=maxi(data)))


@app.route("/override/")
def override_get():
    edit_version()
    with open("/app/MsRewards-Reborn/user_data/configs.json", "r") as inFile:
        configs = json.load(inFile)
    return(render_template("override.html", data=configs))


@app.route("/override/", methods=["post"])
def override_post():
    edit_version()
    with open("/app/MsRewards-Reborn/user_data/configs.json", "r") as inFile:
        configs = json.load(inFile)

    data = dict(request.form)
    for i in configs:
        try :
            data[f'switch{i}']
        except :
            data[f'switch{i}'] = "off"
        
    for i in configs:
        configs[i]["time"] = data[f"time{i}"]
        configs[i]["enabled"] = data[f"switch{i}"] == "on"

    with open("/app/MsRewards-Reborn/user_data/configs.json", "w") as inFile:
        json.dump(configs, inFile)
    update_jobs()
    return(render_template("override.html", data=configs))


@app.route("/database/")
def database_get():
    with open("/app/MsRewards-Reborn/user_data/database.json", "r") as inFile:
        database = json.load(inFile)
    return(render_template("database.html", data = database))


@app.route("/database/", methods=["post"])
def database_post():
    action = request.form
    data = {
        "host": action['address'],
        "table": action['table'],
        "usr": action['user'],
        "pwd": action['password'],
        "checked": ""
        }  

    try :
        if action["switch"] :
            data['checked'] = "checked"
    except:
        pass

    with open("/app/MsRewards-Reborn/user_data/database.json", "w") as inFile:
        json.dump(data, inFile)

    return(render_template("database.html", data = data))


@app.route("/config/")
def config_get():
    with open("/app/MsRewards-Reborn/user_data/proxy.json", "r") as inFile:
        proxys = json.load(inFile)
    with open("/app/MsRewards-Reborn/user_data/discord.json", "r") as inFile:
        discords = json.load(inFile)
    with open("/app/MsRewards-Reborn/user_data/configs.json", "r") as inFile:
        configs = json.load(inFile)
    return(render_template("config.html", data=configs, discords=discords, proxys=proxys, configs=configs, len=maxi(configs)))


@app.route("/config/", methods=["POST"])
def config_post():
    action = request.form
    with open("/app/MsRewards-Reborn/user_data/proxy.json", "r") as inFile:
        proxys = json.load(inFile)
    with open("/app/MsRewards-Reborn/user_data/discord.json", "r") as inFile:
        discords = json.load(inFile)
    with open("/app/MsRewards-Reborn/user_data/configs.json", "r") as inFile:
        configs = json.load(inFile)

    if action["data"] == "delete":
        print(action["config"])
        configs.pop(action["config"])
    else :
        comptes = {
            "1":{"mail": action["mail1"], "pwd": action["pwd1"], "2fa": action["2fa1"]},
            "2":{"mail": action["mail2"], "pwd": action["pwd2"], "2fa": action["2fa2"]},
            "3":{"mail": action["mail3"], "pwd": action["pwd3"], "2fa": action["2fa3"]},
            "4":{"mail": action["mail4"], "pwd": action["pwd4"], "2fa": action["2fa4"]},
            "5":{"mail": action["mail5"], "pwd": action["pwd5"], "2fa": action["2fa5"]}
        }

        configs[action["config"]] = {
            "name" : action["name"] if action["name"] != "" else f"unnamed{action['config']}",
            "proxy": action["proxy"],
            "discord": action["discord"],
            "time":"",
            "enabled":"False",
            "accounts": comptes
            }
    with open("/app/MsRewards-Reborn/user_data/configs.json", "w") as outFile:
        json.dump(configs, outFile)
    return(render_template("config.html", data=configs, discords=discords, proxys=proxys, configs=configs, len=maxi(configs)))

@app.route("/logs/", methods=["GET", "POST"])
def logs():
    return(render_template("logs.html"))



@app.route('/download/<path:filename>', methods=['GET', 'POST'])
@login_required
def download(filename):
    print("file send !")
    return send_from_directory(directory='/app/MsRewards-Reborn/user_data/', path=filename, as_attachment=True)


def allowed_file(filename):
    ALLOWED_EXTENSIONS = ["json"]
    return '.' in filename and \
           filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS

@app.route('/upload_file/', methods=['POST'])
@login_required
def upload_file():
    print(request.files)
    i = 1
    while f'file{i}' in request.files :
        file = request.files[f'file{i}']
        if file.filename == '':
                print('end of files')
                return redirect(url_for('settings_get'))

        elif file and allowed_file(file.filename):
            filename = secure_filename(file.filename)
            print(os.path.join('/app/MsRewards-Reborn/user_data/', filename))
            file.save(os.path.join('/app/MsRewards-Reborn/user_data/', filename))
        
        i += 1
    print(i)
    print(f'file{i}' in request.files)
    print("requete bizarre")            
    return redirect(url_for('settings_get'))


def maxi(dict):
    m = 0
    for i in dict :
        if int(i) >= m:
            m = int(i)
    return(m+1)

def setup_grafana():
    if get("http://localhost:3000/api/datasources").text == "[]" :
        subprocess.Popen(["bash",'/app/MsRewards-Reborn/grafana-config/request.sh'])
setup_grafana()
#if __name__ == '__main__':
#    print("this is launched")
#    setup_grafana()
#    update_jobs()
#    edit_version()
#    app.run(host='0.0.0.0', port=1234, debug=True)
#