#!/usr/bin/env python
#
# Copyright C 2002-2003 by Isilon Systems, Inc. All rights not expressly
# granted herein are reserved.
# Created in the United States of America.
# 
# This digital media is protected by U.S. and international copyright and
# intellectual property laws. Unless otherwise specified, all information
# and screens appearing as part of this digital medium, including
# software, services, documents, text, images, icons, and logos design;
# the selection, assembly, arrangement, and design thereof; and the code
# that enables its presentation, are the sole property of Isilon Systems,
# Inc.
# 
# Except as otherwise required by applicable law, no part of any
# copyrighted material may be reproduced, modified, translated,
# distributed, published, or transmitted by any means, electronic or
# mechanical, for any purpose, without the prior written consent of Isilon
# Systems. 
#  
# Isilon and Isilon Systems are registered trademarks of Isilon Systems,
# Inc.  Legion is a trademark of Isilon Systems, Inc.  All other
# trademarks and registered trademarks are the property of their
# respective holders.
#
# $Id: upgrade 65844 2007-03-14 21:06:27Z qlu $
import sys
import os, os.path

import isi.sys.update.engine as engine
import isi.sys.update.config as config
import isi.sys.update.util as util
    
def exit_on_err(eng):
    if eng._status == config.State.FAILED:
        os._exit(1)
        
if __name__ == '__main__':
    if not len(sys.argv) >= 3:
        os._exit(1)
        
    try:
        tarball = os.path.join(sys.argv[1], sys.argv[2])
        if not os.path.isfile(tarball):
            os._exit(1)

        reimage = False
        ignore_md5 = False
        if len(sys.argv) > 3:
            for arg in sys.argv[3:]:
                if arg == 'reimage':
                    reimage = True
                elif arg == 'ignore_md5':
                    ignore_md5 = True

        upgrade_engine = engine.Engine()
                
        # preparing for installation
        upgrade_engine._lkg_init()
        exit_on_err(upgrade_engine)

        upgrade_engine._load(tarball, None, None, None, None, None)
        exit_on_err(upgrade_engine)

        upgrade_engine.install_count = 0
        upgrade_engine._install(reimage, ignore_md5) 
        exit_on_err(upgrade_engine)

        if not reimage:
            upgrade_engine._restore(False)
            exit_on_err(upgrade_engine)

        os._exit(0)
    except:
        util.Logging.exception()
        os._exit(1)
