
import fcntl
import time
import sys
import os, os.path

update_config_file = '/etc/panopta-agent/panopta_agent.cfg'
if not os.path.exists(update_config_file):
    home = os.path.expanduser("~")
    update_config_file = os.path.join(home, "lib", "fm-agent", "update_config")

unlock_attempts = 10

def update_config(manifest_file):
    if not os.path.exists(manifest_file):
        print "Manifest file %s does not exist" % manifest_file
        sys.exit(0)

    try:
        mf = open(manifest_file).read().strip().split('\n')
        props = dict([map(str.strip, line.split("=")) for line in mf])
    except:
        print "Error reading manifest file %s" % manifest_file
        sys.exit(0)

    manfile = open(update_config_file, 'wt')

    # Acquire lock
    locked = True
    for i in range(unlock_attempts):
        try:
            fcntl.flock(manfile, fcntl.LOCK_EX | fcntl.LOCK_NB)
            locked = False
            break
        except:
            time.sleep(1.0)
    if locked:
        print "Could not unlock %s after %s attempts" % (update_config_file, unlock_attempts)
        sys.exit(1)

    # Write new properties
    manfile.write(open(manifest_file).read())

    # Release lock
    fcntl.flock(manfile, fcntl.LOCK_UN)
    manfile.close()

    # Change perms to make sure the agent user can read the config file
    os.system("chmod 666 %s" % update_config_file)

if __name__ == "__main__":
    if len(sys.argv) < 2:
        print "Please specify the manifest file"
        sys.exit(0)
    else:
        update_config(sys.argv[1])