import Bcfg2.Server.Admin
import sys

try:
    import sqlalchemy, sqlalchemy.orm
except:
    # FIXME should probably do something smarter here for folks without SA
    pass
import Bcfg2.Server.Snapshots
import Bcfg2.Server.Snapshots.model
from Bcfg2.Server.Snapshots.model import Snapshot, Client, Metadata, Base, \
     Group, Package

class Snapshots(Bcfg2.Server.Admin.Mode):
    __shorthelp__ = "Interact with the Snapshots system"
    __longhelp__ = (__shorthelp__)
    __usage__ = ("bcfg2-admin snapshots [init|query qtype] ")

    q_dispatch = {'client':Client, 'group':Group,
                  'snapshot':Snapshot,'package': Package, 'metadata':Metadata, 
                  }

    def __init__(self, configfile):
        Bcfg2.Server.Admin.Mode.__init__(self, configfile)
        self.session = Bcfg2.Server.Snapshots.setup_session(debug=True)

    def __call__(self, args):
        if args[0] == 'query':
            if args[1] in self.q_dispatch:
                q_obj = self.q_dispatch[args[1]]
                results = self.session.query(q_obj).all()
            else:
                print 'error'
                raise SystemExit, 1
            for result in results:
                print result.name
        elif args[0] == 'init':
            dbpath = Bcfg2.Server.Snapshots.db_from_config()
            engine = sqlalchemy.create_engine(dbpath, echo=True)
            metadata = Base.metadata
            metadata.create_all(engine) 
            Session = sqlalchemy.orm.sessionmaker()
            Session.configure(bind=engine)
            session = Session()
            session.commit()
        elif args[0] == 'dump':
            client, etype, ename = args[1:]
            snap = Snapshot.get_current(self.session, unicode(client))
            if not snap:
                print "Current snapshot for %s not found" % client
                sys.exit(1)
            for pkg in snap.packages:
                print pkg.correct, pkg.modified
                print pkg.start.name, pkg.start.version
            print snap.packages