import argparse
import os
import sys
import unpacker
import setup

'''
obtain new info for sites database
install core files
move wp-content to correct place
import db to website
check for any issues

package.json

function ai1wm_activate_plugins( $plugins ) {
	$current = get_option( AI1WM_ACTIVE_PLUGINS, array() );

	// Add plugins
	foreach ( $plugins as $plugin ) {
		if ( ! in_array( $plugin, $current ) && ! is_wp_error( validate_plugin( $plugin ) ) ) {
			$current[] = $plugin;
		}
	}

	return update_option( AI1WM_ACTIVE_PLUGINS, $current );
}
'''

def setup_args():
    parser = argparse.ArgumentParser(
        prog='AiO-Restore',
        description='Unpack and Restore All-in-One WP Migration Packages',
        epilog='aio_restore ',
    )
    parser.add_argument(
        'source',
        type=str,
        help='Source Path'
        )
    parser.add_argument(
        'target',
        type=str,
        help='Target Path'
        )
    parser.add_argument(
        'db_name',
        type=str,
        help='Database Name'
        )
    parser.add_argument(
        'db_pass',
        type=str,
        help='Database Password'
        )
    parser.add_argument(
        name='site_url',
        name='-s',
        name='--site',
        type=str,
        help='Site URL, domain only'
        )
    parser.add_argument(
        '-h',
        '--home',
        'home_url',
        type=str,
        help='Home URL, domain only'
        )
    parser.add_argument(
        '-v',
        '--version',
        action='version',
        version='%(prog)s v0.0.1',
        help='Show version information'
    )
    
    return parser

plugin_prefix = 'SERVMASK_PREFIX_'

if __name__ == '__main__':
    """ Entry of the ai1wm program. """

    parser = setup_args()
    args = parser.parse_args()

    try:
        # Ensure we have source and target
        if not args.source or not args.target:
            parser.print_help()
            exit(1)
        
        # Unpack the archive and files to the source destination
        if os.path.isfile(args.source):
            unpacker.Ai1wmUnpacker().unpack(args.source, args.target)
    except Exception as e:
        print(e)
        sys.exit(-1)
        
    # Process WordPress setup and install
    try:
        # TODO: check for package.json for processing or exit
        if os.path.isfile(args.target):
            # check for database.sql and package.json
            # Setup Site files
            setup.WPSetup()
    except Exception as e:
        print(e)
        sys.exit(-1)
            

    sys.exit(0)