freesitemgr

#!/usr/bin/env python3
#@+leo-ver=4
#@+node:@file freesitemgr
#@@first
#@+others
#@+node:freesitemgr-script
"""
A utility to update freesites from within a cron environment
"""
#@+others
#@+node:imports
import sys, os, time, subprocess, traceback, getopt
import os.path

import fcp3 as fcp
import fcp3.node
from fcp3.sitemgr import SiteMgr, fixUri, defaultMaxManifestSizeBytes, defaultMaxNumberSeparateFiles

#@-node:imports
#@+node:globals
progname = sys.argv[0]

# time we wait after starting fred, to allow the node to 'warm up'
# and make connections to its peers
startupTime = 180

# directory where we have freenet installed,
# change it as needed
#freenetDir = "/home/david/freenet"

homeDir = os.path.expanduser("~")

# derive path of freenet pid file, the (non)existence
# of which is the easiest test of whether the freenet
# node is running
#freenetPidFile = os.path.join(freenetDir, "Freenet.pid")

logFile = os.path.join(homeDir, "updatesites.log")
pidFile = os.path.join(homeDir, "updatesites.pid")

if sys.platform.startswith("win"):
    confDirName = "freesitemgr"
else:
    confDirName = ".freesitemgr"
confDir = os.path.join(homeDir, confDirName)

#@-node:globals
#@+node:editCreateConfig
def editCreateConfig(sitemgr):
    """
    Creates an initial config file interactively
    """
    print("Setting up configuration file %s" % sitemgr.conffile)

    # get fcp hostname
    fcpHost = input("FCP Hostname [%s] (* for all): " % sitemgr.fcpHost).strip()
    if not fcpHost:
        fcpHost = sitemgr.fcpHost
    if fcpHost == '*':
        fcpHost = ""
    
    # get fcp port
    while 1:
        fcpPort = input("FCP Port [%s]: " % sitemgr.fcpPort).strip()
        if not fcpPort:
            fcpPort = sitemgr.fcpPort
        try:
            fcpPort = int(fcpPort)
        except:
            continue
        break

    print("Trying FCP port at %s:%s" % (fcpHost, fcpPort))
    try:
        fcpnode = fcp.FCPNode(host=fcpHost, port=fcpPort)
    except:
        traceback.print_exc()
        print("Failed to connect to FCP Port")
        print("Please ensure your node is running, with its FCP port")
        print("reachable at %s:%s, and try this command again" % (fcpHost, fcpPort))
        print("Setup aborted")
        return

    fcpnode.shutdown()

    sitemgr.fcpHost = fcpHost
    sitemgr.fcpPort = fcpPort

    # confirm and save
    if getyesno("Save configuration", True):
        sitemgr.save()

    print("Configuration saved to %s" % sitemgr.conffile)

#@-node:editCreateConfig
#@+node:addSite
def addSite(sitemgr, sitepath=None):
    """
    Interactively adds a new site to config
    """
    if sitepath:
        print("Add new site for path %s" % sitepath)
    else:
        print("Add new site")
    
    if not sitemgr.node:
        print("Cannot add site - no contact with node on %s:%s" % (
            sitemgr.fcpHost, sitemgr.fcpPort))
        print("Please ensure your freenet node is running, or run")
        print("'%s setup' to edit your FCP access address" % progname)

    if sitepath:
        sitepathabs = os.path.abspath(sitepath)
        sitepathname = os.path.basename(sitepathabs)
    if sitepath and os.path.isdir(sitepathabs) and not sitemgr.hasSite(sitepathname):
        sitename, sitedir = sitepathname, sitepathabs
        print("Using sitename %s and path %s" % (sitename, sitedir))
    else:
        if sitepath:
            if sitemgr.hasSite(sitepathname):
                print("A site with name", sitepathname, "already exists.")
            else:
                print("The argument is not a valid site path.")
        while 1:
            sitename = input("Name of freesite, or empty line to cancel: ").strip()
            if not sitename:
                print("Add site aborted")
                return
            if sitemgr.hasSite(sitename):
                print("Freesite '%s' already exists" % sitename)
                continue
            break
    
        while 1:
            sitedir = input("Directory where freesite's files reside: ").strip()
            if not sitedir:
                print("Add site aborted")
                return
            sitedir = os.path.abspath(sitedir)
            if not os.path.isdir(sitedir):
                print("'%s' is not a directory, try again" % sitedir)
                continue
            # elif not os.path.isfile(os.path.join(sitedir, sitemgr.index)):
            #     print "'%s' has no %s, try again" % (sitedir, sitemgr.index)
            #     continue
            break

    while 1:
        uriPriv = input("Site private URI (if any - press ENTER for new\n: ").strip()
        if not uriPriv:
            uriPub, uriPriv = sitemgr.node.genkey()
        else:
            if not fcp.node.uriIsPrivate(uriPriv):
                print("Sorry, that's a public URI, we need a private URI")
                continue
            try:
                uriPub = sitemgr.node.invertprivate(uriPriv)
            except:
                traceback.print_exc()
                print("Invalid private URI:\n  %s" % uriPriv)
                continue
        break
        uriPub = fixUri(uriPub, sitename)
        uriPriv = fixUri(uriPriv, sitename)

    # good to go - add the site
    site = sitemgr.addSite(name=sitename, dir=sitedir, uriPub=uriPub, uriPriv=uriPriv)

    print("Added new freesite: '%s' => %s" % (site.name, sitedir))
    return site.name

#@-node:addSite
#@+node:removeSite
def removeSite(sitemgr, sitename):
    """
    tries to remove site from config
    """
    if not sitemgr.hasSite(sitename):
        print("No such freesite '%s'" % sitename)
        return

    if getyesno("Are you sure you wish to delete the configuration (metadata and keys) for freesite '%s'" % sitename, False):
        sitemgr.removeSite(sitename)
        print("Removed freesite '%s'" % sitename)

#@-node:removeSite
#@+node:cancelUpdate
def cancelUpdate(sitemgr, sitename, force=False):
    """
    tries to remove site from config
    """
    if not sitemgr.hasSite(sitename):
        print("No such freesite '%s'" % sitename)
        return

    doit = False
    if force:
        doit = True
    elif getyesno("Are you sure you wish to cancel update for freesite '%s'" \
                    % sitename, False):
        doit = True
    if doit:
        sitemgr.cancelUpdate(sitename)
        print("Cancelled update for freesite '%s'" % sitename)
    else:
        print("Not cancelling update for freesite '%s'" % sitename)

#@-node:cancelUpdate
#@+node:getChkCalcNode
def getChkCalcNode(addr):
    """
    yuck - using a separate node for chk calculation
    """
    parts = addr.split(":")
    nparts = len(parts)
    if nparts == 1:
        chkHost = addr
        chkPort = fcp.node.defaultFCPPort
    elif nparts == 2:
        chkHost = parts[0] or fcp.node.defaultFCPHost
        chkPort = parts[1] or fcp.node.defaultFCPPort
    
    try:
        chkNode = fcp.node.FCPNode(host=chkHost, port=chkPort)
        return chkNode
    except:
        return None

#@-node:getChkCalcNode
#@+node:getYesNo
def getyesno(ques, default=False):
    """
    prompt for yes/no answer, with default
    """
    if default:
        prmt = "[Y/n]"
    else:
        prmt = "[y/N]"
        
    resp = input(ques + " " + prmt + " ").strip().lower()
    
    if not resp:
        return default
    #elif resp[0] in ['y', 't']:
    elif resp[0] == 'y':
        return True
    else:
        return False
#@-node:getYesNo
#@+node:help
def help():
    """
    dump help info and exit
    """
    maxMankiB = defaultMaxManifestSizeBytes / 1024
    print("%s: a console-based USK freesite insertion utility" % progname)
    
    print("Usage: %s [options] <command> <args>" % progname)
    print("Options:")
    print("  -h, --help")
    print("          - display this help message")
    print("  -c, --config-dir=path")
    print("          - use different config directory (default is %s)" % confDir)
    print("  -v, --verbose")
    print("          - run verbosely, use multiple times for even more noise")
    print("  -q, --quiet")
    print("          - run quietly")
    print("  -n, --no-update")
    print("          - do not update the site after adding it (only for add)")
    print("  -i, --index")
    print("          - index file (default is index.html)")
    print("  -m, --mime-type")
    print("          - mime-type of the index file (default is \"text/html\")")
    print("  -l, --logfile=filename")
    print("          - location of logfile (default %s)" % logFile)
    print("  -r, --priority")
    print("     Set the priority (0 highest, 6 lowest, default 3)")
    print("  -C, --cron")
    print("     Set options suitable for putting freesitemgr in your crontab,")
    print("     and output a dated header with each site insert")
    print("  --max-manifest-size=sizekiB")
    print("     Insert at most files of the given cumulative size as manifest.")
    print("     The files are chosen based on whether they are referenced in the")
    print("     index-file and by size - smallest first. If you have a site with")
    print("     many small files, you should increase this to include them all")
    print("     and benefit from better compression (default: %s)." % maxMankiB)
    print("     It will only go above this to avoid inserting more than")
    print("     %s files separately." % defaultMaxNumberSeparateFiles)
    print("  --chk-calculation-node=hostname[:port]")
    print("     Use a different node for CHK calculations, which can be a")
    print("     timesaver when inserting large amounts of data into a remote node")
    print("     (port defaults to %s)" % fcp.node.defaultFCPPort)
    print()
    print("Available Commands:")
    print("  setup              - create/edit freesite config file interactively")
    print("  add                - add new freesite - user will be prompted for")
    print("                       details")
    print("  list [<name>]      - display a summary of all freesites, or a")
    print("                       detailed report of one site if <name> given")
    print("  listall            - print detailed report of all sites")
    print("  remove <name>      - remove metadata and keys for given freesite")
    print("  update [<name>...] - insert freesites which have changed since")
    print("                       they were last inserted. If no site names are")
    print("                       given, then all freesites will be updated")
    print("  reinsert [<name>...] - update with forceb reinsert of all files")
    print("  cancel <name>...   - cancel any pending insert of freesite(s) <name>.")
    print("  cleanup <name>...  - clean up node queue for site(s) <name>")
    print("  help               - same as '-h', display this help page")

    sys.exit(0)

#@-node:help
#@+node:usage
def usage(ret=-1, msg=None):
    if msg != None:
        print(msg)
    print("Usage: %s [options] <command> [<arguments>]" % progname)
    print("Do '%s -h' for help" % progname)

    sys.exit(ret)

#@-node:usage
#@+node:noNodeError
def noNodeError(sitemgr, msg):
    print(msg + ": cannot connect to node FCP port")
    print("To use this command, you must have a freenet node running")
    print("Please ensure your node FCP port is reachable at %s:%s" % (
        sitemgr.fcpHost, sitemgr.fcpPort))
    print("or run '%s setup' to configure a different FCP port" % progname)
    sys.exit(1)

#@-node:noNodeError
#@+node:main
def main():

    force = False
    cron = False
    chkCalcNode = None

    # default job options
    opts = {
            "verbosity" : fcp.node.ERROR,
            "Verbosity" : 0,
            #"logfile" : logFile,
            "priority"  : 3,
            "update-after-add"    : True
            }

    # process command line switches
    try:
        cmdopts, args = getopt.getopt(
            sys.argv[1:],
	    "?hvc:l:r:qfnCVi:m:",
            ["help", "verbose", "config-dir=", "logfile=",
             "max-concurrent=", "quiet", "force", "no-update",
             "priority", "cron",
             "chk-calculation-node=", "max-manifest-size=",
             "version", "index", "mime-type",
             ]
            )
    except getopt.GetoptError:
        # print help information and exit:
        usage()
        sys.exit(2)
    output = None
    verbose = False
    #print cmdopts
    for o, a in cmdopts:

        if o in ("-?", "-h", "--help"):
            help()
            sys.exit(0)

        if o in ("-V", "--version"):
            print("This is %s, version %s" % (progname, fcp.node.fcpVersion))
            sys.exit(0)

        if o in ("-v", "--verbosity"):
            opts['verbosity'] += 1
            opts['Verbosity'] = 1023
        
        if o in ("-q", "--quiet"):
            opts['verbosity'] = fcp.node.SILENT
        
        if o in ("-n", "--no-update"):
            opts['update-after-add'] = False
        
        if o in ("-c", "--config-dir"):
            opts['basedir'] = a
        
        if o in ("-l", "--logfile"):
            opts['logfile'] = a

        if o in ("-i", "--index"):
            opts['index'] = a

        if o in ("-m", "--mime-type"):
            opts['mtype'] = a

        if o == '--max-manifest-size':
            opts['maxManifestSizeBytes'] = int(float(a)*1024)

        if o == '--chk-calculation-node':
            chkNode = getChkCalcNode(a)
            if not chkNode:
                usage("Failed to connect to specified CHK calc node '%s'" % a)
            opts['chkCalcNode'] = chkNode
        
        if o in ("-r", "--priority"):
            try:
                pri = int(a)
                if pri < 0 or pri > 6:
                    raise hell
            except:
                usage("Invalid priority '%s'" % pri)
            opts['priority'] = int(a)

        if o in ("-f", "--force"):
            force = True

        if o in ("-C", "--cron"):
            opts['verbosity'] = fcp.node.INFO
            opts['Verbosity'] = 1023
            cron = True

    # process command
    if len(args) < 1:
        usage(msg="No command given")

    cmd = args.pop(0)

    if cmd not in [
            'setup','config','init',
            'add',
            'remove',
            'list', 'listall',
            'update',
            'cancel', "help", "cleanup", "reinsert",
            ]:
        usage(msg="Unrecognised command '%s'" % cmd)

    # we now have a likely valid command, so now we need a sitemgr
    sitemgr = SiteMgr(*args, **opts)

    if cmd in ['setup', 'init', 'config']:
        editCreateConfig(sitemgr)

    elif cmd == 'help':
        help()
        sys.exit(0)

    elif cmd == 'add':
        if not sitemgr.node:
            noNodeError(sitemgr, "Cannot add site")
        if args:
            sitepath = args[0]
            sitename = addSite(sitemgr, sitepath)
        else:
            sitename = addSite(sitemgr)
    
        if opts["update-after-add"]: 
            # renew sitemgr, then update 
            # FIXME: Move the whole if-elif-else into a function to
            # make it clear that nothing spills out.
            try:
                sitemgr.node.shutdown()
            except:
                try:
                    sitemgr.node.socket.close()
                except:
                    pass
                pass
            sitemgr = SiteMgr(**opts)
            sitemgr.insert(sitename, cron=cron)

    elif cmd == 'remove':
        if not args:
            print("Remove site: no freesites selected")
            return
        for sitename in args:
            removeSite(sitemgr, sitename)

    elif cmd == 'cancel':
        if not args:
            print("Cancel site update: no freesites selected")
            return
        for sitename in args:
            cancelUpdate(sitemgr, sitename, force)

    elif cmd == 'cleanup':
        sitemgr.cleanup(*args)

    elif cmd in ['list', 'listall']:
        if cmd == 'listall':
            sites = [site.name for site in sitemgr.sites]
        else:
            sites = args
        if not sites:
            # summary list
            names = []
            for site in sitemgr.sites:
                if site.updateInProgress:
                    names.append("*"+site.name)
                else:
                    names.append(site.name)
            print(" ".join(names))
        else:
            for sitename in sites:
                if not sitemgr.hasSite(sitename):
                    print("No such site '%s'" % sitename)
                else:
                    site = sitemgr.getSite(sitename)
                    if site.updateInProgress:
                        state = "updating"
                    else:
                        state = "idle"
                    print("%s:" % sitename)
                    print("    state: %s" % state)
                    print("    dir: %s" % site.dir)
                    print("    uri: %s" % site.uriPub)
                    print("    privkey: %s" % site.uriPriv)
                    #print "    version: %s" % info['version']

            pass

    elif cmd == 'reinsert':
        if not sitemgr.node:
            noNodeError(sitemgr, "Cannot update freesites")
        try:
            if not args:
                sites = sitemgr.getSiteNames()
            else:
                sites = args
            sitemgr.reinsert(cron=cron, *args)
        except KeyboardInterrupt:
            print("freesitemgr: site reinsert cancelled by user")

    elif cmd == 'update':
        if not sitemgr.node:
            noNodeError(sitemgr, "Cannot update freesites")
        try:
            if not args:
                sites = sitemgr.getSiteNames()
            else:
                sites = args
            sitemgr.insert(cron=cron, *args)
        except KeyboardInterrupt:
            print("freesitemgr: site update cancelled by user")

    try:
        sitemgr.node.shutdown()
    except:
        try:
            sitemgr.node.socket.close()
        except:
            pass
        pass

#@-node:main
#@+node:mainline
if __name__ == '__main__':
    main()

#@-node:mainline
#@-others
#@-node:freesitemgr-script
#@-others
#@-node:@file freesitemgr
#@-leo