#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# Copyright (c) 2019 Bryan Davis and Wikimedia Foundation. All Rights Reserved.
from __future__ import print_function

import argparse

import redis
import sqlite3


def main():
    parser = argparse.ArgumentParser(description='Fix orphan redis sets')
    parser.add_argument(
        '--do-it', dest='do_it', action='store_true',
        help='Update database and redis')
    args = parser.parse_args()

    conn = sqlite3.connect('/etc/dynamicproxy-api/data.db')
    cur = conn.cursor()
    r = redis.Redis()

    cur.execute("""
    SELECT
      r.domain as domain
    FROM
      project p
      JOIN route r on p.id = r.project_id
      JOIN backend b on r.id = b.route_id
    """)
    domains = [domain for domain, in cur.fetchall()]
    conn.close()

    orphans = [
        key for key in r.scan_iter(match='frontend:*')
        if key[9:] not in domains
    ]
    for orphan in orphans:
        print(orphan)
    if args.do_it and orphans:
        r.delete(*orphans)

main()