import glob
import os
import sys
import time
import tarfile
import Bcfg2.Server.Admin
import Bcfg2.Options

class Backup(Bcfg2.Server.Admin.MetadataCore):
    __shorthelp__ = "Make a backup of the Bcfg2 repository."
    __longhelp__ = (__shorthelp__ + "\n\nbcfg2-admin backup start"
                                    "\n\nbcfg2-admin backup restore")
    __usage__ = ("bcfg2-admin backup [start|restore]")

    def __init__(self, configfile):
        Bcfg2.Server.Admin.MetadataCore.__init__(self, configfile,
                                                 self.__usage__)

    def __call__(self, args):
        Bcfg2.Server.Admin.MetadataCore.__call__(self, args)
        # Get Bcfg2 repo directory
        opts = {'repo': Bcfg2.Options.SERVER_REPOSITORY}
        setup = Bcfg2.Options.OptionParser(opts)
        setup.parse(sys.argv[1:])
        self.datastore = setup['repo']
        
	if len(args) == 0:
            self.errExit("No argument specified.\n"
                         "Please see bcfg2-admin backup help for usage.")
        if args[0] == 'start':
            timestamp = time.strftime('%Y%m%d%H%M%S')
            format = 'gz'
            mode = 'w:' + format
            filename = timestamp + '.tar' + '.' + format
            out = tarfile.open(self.datastore + '/' + filename, mode=mode)
            content = os.listdir(self.datastore)    
            for item in content:
                out.add(item)
            out.close()
            print "Archive %s was stored.\nLocation: %s" % (filename, self.datastore)

        elif args[0] == 'restore':
            print 'Not implemented yet'
        
        else:
            print "No command specified"
            raise SystemExit(1)