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

Add logic - evaluate ip for eth1 and eth0, determine which to use in the update.

"""

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('eth1')
proxyname = "administrative"
login_url = "/001/"
sql = """UPDATE 'psiphon'.'proxy' SET hostname = \"%s\" WHERE name = "administrative";""" % (address)

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(sql)
    #m.execute("""UPDATE "psiphon"."proxy" SET hostname = \"%s\"  WHERE name = "administrative";""" , (address))
    #m.execute('UPDATE "psiphon"."proxy" SET hostname = \"%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()