import asm3.al
import asm3.cachedisk
import asm3.db
from asm3.sitedefs import MULTIPLE_DATABASES, MULTIPLE_DATABASES_TYPE

import datetime
import re
import os, sys
import web

# Regex to remove invalid chars from an entered database
INVALID_REMOVE = re.compile(r'[\/\.\*\?\ ]')

try:
    sys.path.append("/root/asmdb")
    import smcom_client
except:
    # sys.stderr.write("warn: no smcom_client\n")
    pass

def active():
    """
    Returns true if we're sheltermanager.com
    """
    return MULTIPLE_DATABASES and MULTIPLE_DATABASES_TYPE == "smcom"

def get_account(alias):
    """
    Returns the smcom account object for alias/db
    Uses a read through 48 hour cache to save unnecessary calls
    """
    # Attackers have tried to overflow alias in the past, we'll never use more than 20 chars
    # fail fast and save us a load of processing.
    if len(alias) > 20: return None     
    TTL = 86400 * 2
    cachekey = "smcom_dbinfo_%s" % alias
    a = asm3.cachedisk.get(cachekey, "smcom")
    if a is None:
        a = smcom_client.get_account(alias)
        if a is not None and "user" in a:
            asm3.cachedisk.put(cachekey, "smcom", a, TTL)
    return a

def get_database_info(alias):
    """
    Returns the dbo object for a sheltermanager.com account or alias.  
    Also returns a dbo with a database property of "DISABLED" for a 
    disabled account, "FAIL" for a problem or "WRONGSERVER" to indicate
    that the database does not exist on this server.
    """
    alias = re.sub(INVALID_REMOVE, '', alias).lower()
    dbo = asm3.db.get_dbo("POSTGRESQL")
    dbo.host = "/var/run/postgresql/" # use socket dir to use UNIX sockets to connect to local pgbouncer /var/run/postgresql/
    dbo.port = 6432
    dbo.dbtype = "POSTGRESQL"
    dbo.alias = alias

    a = get_account(alias)
    if a is None:
        dbo.database = "FAIL"
        return dbo

    dbo.database = str(a["user"])
    dbo.username = dbo.database
    dbo.password = dbo.database

    # Is this sm.com account disabled or removed from the server?
    if a["expired"] or a["archived"]:
        dbo.database = "DISABLED"

    # Is this the wrong server?
    if smcom_client.get_this_server() != a["server"]: 
        dbo.database = "WRONGSERVER"
        asm3.al.error("failed login, wrong server: %s not present in %s" % (a["server"], smcom_client.get_this_server()))

    return dbo

def get_expiry_date(dbo):
    """
    Returns the account expiry date or None for a problem.
    """
    a = get_account(dbo.database)
    try:
        expiry = datetime.datetime.strptime(a["expiry"], "%Y-%m-%d")
        asm3.al.debug("retrieved account expiry date: %s" % expiry, "smcom.get_expiry_date", dbo)
        return expiry
    except:
        return None

def go_smcom_my(dbo):
    """
    Goes to the my account page for this database
    """
    raise web.seeother(smcom_client.get_my_url(dbo.database))

def vacuum_full(dbo):
    """ Performs a full vacuum on the database via command line (transaction problems via db.py) """
    os.system("psql -U %s -c \"VACUUM FULL;\"" % dbo.database)