#!/usr/bin/env python
"""
A command line helper for common operations performed by Galaxy maintainers.
Encodes and decodes IDs, returns Dataset IDs if provided an HDA or LDDA id,
returns the disk path of a dataset.
"""

import os, sys
from ConfigParser import ConfigParser
from optparse import OptionParser

default_config = os.path.abspath( os.path.join( os.path.dirname( __file__ ), '..', 'universe_wsgi.ini') )

parser = OptionParser()
parser.add_option( '-c', '--config', dest='config', help='Path to Galaxy config file (universe_wsgi.ini)', default=default_config )
parser.add_option( '-e', '--encode-id', dest='encode_id', help='Encode an ID' )
parser.add_option( '-d', '--decode-id', dest='decode_id', help='Decode an ID' )
parser.add_option( '--hda', dest='hda_id', help='Display HistoryDatasetAssociation info' )
parser.add_option( '--ldda', dest='ldda_id', help='Display LibraryDatasetDatasetAssociation info' )
( options, args ) = parser.parse_args()

try:
    assert options.encode_id or options.decode_id or options.hda_id or options.ldda_id
except:
    parser.print_help()
    sys.exit( 1 )

options.config = os.path.abspath( options.config )
os.chdir( os.path.dirname( options.config ) )
sys.path.append( 'lib' )

from galaxy import eggs
import pkg_resources

config = ConfigParser( dict( file_path = 'database/files',
                             id_secret = 'USING THE DEFAULT IS NOT SECURE!',
                             database_connection = 'sqlite:///database/universe.sqlite?isolation_level=IMMEDIATE' ) )
config.read( os.path.basename( options.config ) )

from galaxy.web import security
from galaxy.model import mapping

helper = security.SecurityHelper( id_secret = config.get( 'app:main', 'id_secret' ) )
model = mapping.init( config.get( 'app:main', 'file_path' ), config.get( 'app:main', 'database_connection' ), create_tables = False )

if options.encode_id:
    print 'Encoded "%s": %s' % ( options.encode_id, helper.encode_id( options.encode_id ) )

if options.decode_id:
    print 'Decoded "%s": %s' % ( options.decode_id, helper.decode_id( options.decode_id ) )

if options.hda_id:
    try:
        hda_id = int( options.hda_id )
    except:
        hda_id = int( helper.decode_id( options.hda_id ) )
    hda = model.context.current.query( model.HistoryDatasetAssociation ).get( hda_id )
    print 'HDA "%s" is Dataset "%s" at: %s' % ( hda.id, hda.dataset.id, hda.file_name )

if options.ldda_id:
    try:
        ldda_id = int( options.ldda_id )
    except:
        ldda_id = int( helper.decode_id( options.ldda_id ) )
    ldda = model.context.current.query( model.HistoryDatasetAssociation ).get( ldda_id )
    print 'LDDA "%s" is Dataset "%s" at: %s' % ( ldda.id, ldda.dataset.id, ldda.file_name )