#!/usr/bin/python
# Copyright (c) 2008 Alon Swartz <alon@turnkeylinux.org> - all rights reserved
#Modified significantly by Rik Goldman
"""
Configure Psiphon Proxy in MySQL

"""

import sys
import time
import getopt
import socket
import fcntl
import struct

from executil import ExecError, system

DEBIAN_CNF = "/etc/mysql/debian.cnf"

class Error(Exception):
    pass

def escape_chars(s):
    """escape special characters: required by nested quotes in query"""
    s = s.replace("\\", "\\\\")  # \  ->  \\
    s = s.replace('"', '\\"')    # "  ->  \"
    s = s.replace("'", "'\\''")  # '  ->  '\''
    return s

#Set Proxy: verbatim excerpt from code.activestate.com/recipes/439094

def get_ip_address ( ifname ):
    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    return socket.inet_ntoa(fcntl.ioctl(
        s.fileno(),
        0x8915,
        struct.pack('256s', ifname[:15])
    )[20:24])
#Set variables for psiphon.proxy
address = "get_ip_address('eth0')"
proxyname = "administrative"
login_url = "/001/"

class MySQL:
    def __init__(self):
        system("mkdir -p /var/run/mysqld")
        system("chown mysql:root /var/run/mysqld")

        self.selfstarted = False
        if not self._is_alive():
            self._start()
            self.selfstarted = True

    def _is_alive(self):
        try:
            system('mysqladmin -s ping >/dev/null 2>&1')
        except ExecError:
            return False

        return True

    def _start(self):
        system("mysqld --skip-networking >/dev/null 2>&1 &")
        for i in range(6):
            if self._is_alive():
                return

            time.sleep(1)

        raise Error("could not start mysqld")

    def _stop(self):
        if self.selfstarted:
            system("mysqladmin --defaults-file=%s shutdown" % DEBIAN_CNF)

    def __del__(self):
        self._stop()

    def execute(self, query):
        system("mysql --defaults-file=%s -B -e '%s'" % (DEBIAN_CNF, query))

def usage(s=None):
    if s:
        print >> sys.stderr, "Error:", s
    print >> sys.stderr, "Syntax: %s [options]" % sys.argv[0]
    print >> sys.stderr, __doc__
    sys.exit(1)

def main():

    m = MySQL()

    #Set Proxy URL
    m.execute('UPDATE psiphon.proxy SET hostneme = \"%s\" WHERE name = "administrative";') % (address)
    #Correct query without variables:  m.execute('update psiphon.proxy set hostname = "192.168.1.198" where name = "administrative";')
    #WORKS with Substitutions:  m.execute('REPLACE INTO psiphon.proxy (hostname) VALUES (\"%s\") (name="administrative");') % (hostname)
    # set password
    #m.execute('update mysql.user set Password=PASSWORD(\"%s\") where User=\"%s\"; flush privileges;' % (escape_chars(password), username))

    # edge case: update DEBIAN_CNF
    #if username == "debian-sys-maint":
    #    old = file(DEBIAN_CNF).read()
    #    new = re.sub("password = (.*)\n", "password = %s\n" % password, old)
    #    file(DEBIAN_CNF, "w").write(new)


if __name__ == "__main__":
    main()