#!/usr/bin/python2.2
# This file is part of Python FAM.
# Copyright (C) 2002 Robert Sander <robert.sander@epigenomics.com>
# 
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2.1 of the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307  USA
#
# $Id: test2.py,v 1.5 2005/04/10 07:48:45 loewis Exp $
#

import os, sys

if os.name != 'posix':
    print 'I am sorry, but this script will run only on POSIX platforms.'
    sys.exit(1)

import _fam, time, signal, getopt, errno, select

# hash of filename => TestRequest
requests = {}

showreqid = None
suspend = None
cont = None
intr = None
usr1 = None
usr2 = None

class TestRequest:
    def __init__(self, userData, isDir):
        self.fr = None
        self.userData = userData
        self.isDir = isDir

def handle_stop(signum, frame):
    global suspend
    
    print 'Suspended!'
    suspend = 1

def handle_cont(signum, frame):
    global cont
    
    print 'Resumed!'
    signal.signal(signal.SIGCONT, handle_cont)
    cont = 1

def handle_int(signum, frame):
    global intr
    
    print 'Interupted!'
    signal.signal(signal.SIGINT, handle_int)
    intr = 1

def handle_usr1(signum, frame):
    global usr1
    
    print 'Got USR1!'
    signal.signal(signal.SIGUSR1, handle_usr1)
    usr1 = 1

def handle_usr2(signum, frame):
    global usr2
    
    print 'Got USR2!'
    signal.signal(signal.SIGUSR2, handle_usr2)
    usr2 = 1

def sendRequest(request, path):
    if request.isDir:
        request.fr = fc.monitorDirectory(path, request.userData)
    else:
        request.fr = fc.monitorFile(path, request.userData)
    if showreqid:
        print 'req %d' % request.fr.requestID(),
    if request.isDir:
        print 'FAMMonitorDirectory("%s")' % path
    else:
        print 'FAMMonitorFile("%s")' % path

def sendRequests():
    for path in requests.keys():
        sendRequest(requests[path], path)

def processDirEvents(fe):
    if fe.userData:
        print fe.userData,
    if showreqid:
        print 'req %d' % fe.requestID,
    print fe.filename, fe.code2str()
    newfile = os.path.join(fe.userData, fe.filename)
    if fe.code == _fam.Exists and fe.filename != fe.userData and os.path.isdir(newfile):
        requests[newfile] = TestRequest(newfile, 1)
        sendRequest(requests[newfile], newfile)
    if fe.code == _fam.Created and os.path.isdir(newfile):
        requests[newfile] = TestRequest(newfile, 1)
        sendRequest(requests[newfile], newfile)
    if fe.code == _fam.Deleted and os.path.isdir(newfile):
        del requests[newfile].fr
        del requests[newfile]
        
if len(sys.argv) < 2:
    print 'usage: %s [-r] [-f <filename>] [-d <dirname>]' % sys.argv[0]
    sys.exit(1)

fc = _fam.open()

optlist, args = getopt.getopt(sys.argv[1:], 'f:d:r')

for arg, value in optlist:
    if arg == '-r':
        showreqid = 1

for arg, filename in optlist:
    if arg == '-f':
        requests[filename] = TestRequest('%s' % filename, None)
    if arg == '-d':
        requests[filename] = TestRequest('%s' % filename, 1)
        
signal.signal(signal.SIGTSTP, handle_stop)
signal.signal(signal.SIGCONT, handle_cont)
signal.signal(signal.SIGINT, handle_int)
signal.signal(signal.SIGUSR1, handle_usr1)
signal.signal(signal.SIGUSR2, handle_usr2)

sendRequests()

while 1:
    if suspend:
        for request in requests.values():
            request.fr.suspendMonitor()
            print 'Suspended monitoring of request %i' % request.fr.requestID()
        suspend = None
        signal.signal(signal.SIGTSTP, handle_stop)
    if cont:
        for request in requests.values():
            request.fr.resumeMonitor()
            print 'Resumed monitoring of request %i' % request.fr.requestID()
        cont = None
    if intr:
        # The dealloc methods of all objects should handle
        # FAMCancelMonitor and FAMClose
        sys.exit(0)
    if usr1:
        # Cancel all requests, close the connection, and reopen it.
        # This makes sure long-lived clients can connect, monitor, and
        # disconnect repeatedly without leaking.
        usr1 = None
        sleeptime = 1
        for request in requests.values():
            print 'Cancelling monitoring of request %i' % request.fr.requestID()
            request.fr.cancelMonitor()
        fc.close()
        print 'Closed connection, sleeping %d...' % sleeptime
        time.sleep(sleeptime)
        fc = _fam.open()
        sendRequests()
    if usr2:
        # Clean things up like a well-behaved client and exit.
        for request in requests.values():
            print 'Cancelling monitoring of request %i' % request.fr.requestID()
            request.fr.cancelMonitor()
        fc.close()
        print 'Closed connection'
        sys.exit(0)
    try:
        ri, ro, re = select.select([fc], [], [])
    except select.error, er:
        errnumber, strerr = er
        if errnumber == errno.EINTR:
            continue
        else:
            print strerr
            sys.exit(1)
    while fc.pending():
        fe = fc.nextEvent()
        processDirEvents(fe)