app.py

import json
import struct
import requests

from flask import Flask, render_template, redirect
from flask_swagger_ui import get_swaggerui_blueprint
from apscheduler.schedulers.background import BackgroundScheduler

app = Flask(__name__, static_url_path='/static')
scheduler = BackgroundScheduler()

swaggerui_blueprint = get_swaggerui_blueprint('/api/docs', '/static/api.yaml')
app.register_blueprint(swaggerui_blueprint)

authors = {}
counts = {}
stats = {}
authorsCased = {}


@scheduler.scheduled_job('interval', minutes=15)
def updateCounts():
    global authors, counts, stats
    authors.clear()
    counts.clear()
    stats.clear()

    # Grab RuneLite version from bootstrap file
    bootstrap = requests.get("https://static.runelite.net/bootstrap.json").json()
    rl_version = bootstrap['version']

    # Grab the Plugin Hub manifest as a byte array
    manifest = requests.get("https://repo.runelite.net/plugins/manifest/" + rl_version + "_full.js").content

    # Read and remove the signature. The length is stored in the first 4 bytes
    signature_size = struct.unpack('>i', manifest[0:4])[0]
    manifest = json.loads(manifest[4 + signature_size:])

    # Link each plugin name to the respective author
    for plugin in manifest['display']:
        author = plugin['author'].strip().lower().replace("_", " ")
        authors[plugin['internalName']] = author
        counts[author] = 0
        authorsCased[author] = plugin['author']

    # Grab the recent stats for each plugin
    stats = requests.get("https://api.runelite.net/runelite-" + rl_version + "/pluginhub").json()

    # Consolidate plugin installs
    for plugin in stats:
        if plugin not in authors:
            continue

        author = authors[plugin]
        counts[author] = counts[author] + stats[plugin]

    # Sort counts and stats dicts by value for finding author/plugin ranks
    counts = {k: v for k, v in sorted(counts.items(), key=lambda item: item[1], reverse=True)}
    stats = {k: v for k, v in sorted(stats.items(), key=lambda item: item[1], reverse=True)}


@app.route('/')
def get():
    return render_template('index.html')


@app.route('/leaderboards')
def getLeaderboards():
    return render_template('leaderboards.html', cols=["Rank", "Author", "Total Installs"], rows=counts)


@app.route('/generate')
def getGenerate():
    return render_template('generate.html', plugins=sorted(stats.keys()), authors=sorted(authorsCased.values()))


# API section
@app.route('/installs/author/<username>')
def getAuthorInstalls(username):
    username = str(username).strip().lower().replace("_", " ")
    if username not in counts:
        return "-1"

    return str(counts[username])


@app.route('/installs/plugin/<plugin_name>')
def getPluginInstalls(plugin_name):
    plugin_name = str(plugin_name).strip().lower().replace(" ", "-")
    if plugin_name not in stats:
        return "-1"

    return str(stats[plugin_name])


@app.route('/rank/author/<username>')
def getAuthorRank(username):
    username = str(username).strip().lower().replace("_", " ")
    for idx, name in enumerate(counts, start=1):
        if username == name:
            return str(idx)

    return "-1"


@app.route('/rank/plugin/<plugin_name>')
def getPluginRank(plugin_name):
    plugin_name = str(plugin_name).strip().lower().replace(" ", "-")
    for idx, plugin in enumerate(stats, start=1):
        if plugin_name == plugin:
            return str(idx)

    return "-1"


@app.route('/shields/installs/author/<username>')
def getAuthorInstallsBadge(username):
    return generateShieldJson(int(getAuthorInstalls(username)))


@app.route('/shields/installs/plugin/<plugin_name>')
def getPluginInstallsBadge(plugin_name):
    return redirect("https://api.runelite.net/pluginhub/shields/installs/plugin/" + plugin_name, 301)


@app.route('/shields/rank/author/<username>')
def getAuthorRankBadge(username):
    return generateShieldJson(int(getAuthorRank(username)), "Author rank")


@app.route('/shields/rank/plugin/<plugin_name>')
def getPluginRankBadge(plugin_name):
    return redirect("https://api.runelite.net/pluginhub/shields/rank/plugin/" + plugin_name, 301)


# Utilizes https://shields.io/endpoint to generate the badge dynamically
def generateShieldJson(count: int, label="Total installs"):
    return {
        "schemaVersion": 1,
        "label": label,
        "message": str(count),
        "color": "007bff" if count >= 0 else "red",
        "isError": count == -1
    }


# Initialize counts and schedule them for every 15 minutes
updateCounts()
scheduler.start()

if __name__ == '__main__':
    app.run(host='0.0.0.0')