from route.tool.func import *
from route.tool.mark import load_conn2, namumark

try:
    set_data = json.loads(open('data/set.json').read())
except:
    if os.getenv('NAMU_DB') != None:
        set_data = { "db" : os.getenv('NAMU_DB') }
    else:
        print('DB name (data) : ', end = '')
        
        new_json = str(input())
        if new_json == '':
            new_json = 'data'
            
        with open('data/set.json', 'w') as f:
            f.write('{ "db" : "' + new_json + '" }')
            
        set_data = json.loads(open('data/set.json').read())
        
print('DB name : ' + set_data['db'])
db_name = set_data['db']

conn = sqlite3.connect(db_name + '.db', check_same_thread = False)
curs = conn.cursor()

load_conn(conn)

print('----')
print('1. Backlink reset')
print('2. reCAPTCHA delete')
print('3. Ban delete')
print('4. Change host')
print('5. Change port')
print('6. Change skin')
print('7. Change password')
print('8. Reset version')
print('9. Delete set.json')
print('10. Change name')

print('----')
print('Select : ', end = '')
what_i_do = input()

if what_i_do == '1':
    def parser(data):
        namumark(data[0], data[1], 1)

    curs.execute(db_change("delete from back"))
    conn.commit()

    curs.execute(db_change("select title, data from data"))
    data = curs.fetchall()
    num = 0

    for test in data:
        num += 1

        t = threading.Thread(target = parser, args = [test])
        t.start()
        t.join()

        if num % 10 == 0:
            print(num)
elif what_i_do == '2':
    curs.execute(db_change("delete from other where name = 'recaptcha'"))
    curs.execute(db_change("delete from other where name = 'sec_re'"))
elif what_i_do == '3':
    print('----')
    print('IP or Name : ', end = '')
    user_data = input()

    if re.search("^([0-9]{1,3}\.[0-9]{1,3})$", user_data):
        band = 'O'
    else:
        band = ''

        curs.execute(db_change("insert into rb (block, end, today, blocker, why, band) values (?, ?, ?, ?, ?, ?)"), 
            [user_data, 
            'release', 
            get_time(), 
            'tool:emergency', 
            '', 
            band
        ])
    curs.execute(db_change("delete from ban where block = ?"), [user_data])
elif what_i_do == '4':
    print('----')
    print('Host : ', end = '')
    host = input()

    curs.execute(db_change("update other set data = ? where name = 'host'"), [host])
elif what_i_do == '5':
    print('----')
    print('Port : ', end = '')
    port = int(input())

    curs.execute(db_change("update other set data = ? where name = 'port'"), [port])
elif what_i_do == '6':
    print('----')
    print('Skin name : ', end = '')
    skin = input()

    curs.execute(db_change("update other set data = ? where name = 'skin'"), [skin])
elif what_i_do == '7':
    print('----')
    print('1. sha256')
    print('2. sha3')
    
    print('----')
    print('Select : ', end = '')
    what_i_do = int(input())

    print('----')
    print('User name : ', end = '')
    user_name = input()

    print('----')
    print('User password : ', end = '')
    user_pw = input()

    if what_i_do == '1':
        hashed = hashlib.sha256(bytes(user_pw, 'utf-8')).hexdigest()
    else:
        if sys.version_info < (3, 6):
            hashed = sha3.sha3_256(bytes(user_pw, 'utf-8')).hexdigest()
        else:
            hashed = hashlib.sha3_256(bytes(user_pw, 'utf-8')).hexdigest()
       
    curs.execute(db_change("update user set pw = ? where id = ?"), [hashed, user_name])
elif what_i_do == '8':
    curs.execute(db_change("update other set data = '00000' where name = 'ver'"))
elif what_i_do == '9':
    try:
        os.remove('data/set.json')
    except:
        pass
else:
    print('----')
    print('User name : ', end = '')
    user_name = input()

    print('----')
    print('New name : ', end = '')
    new_name = input()

    curs.execute(db_change("update user set id = ? where id = ?"), [new_name, user_name])

conn.commit()

print('----')
print('OK')