#!/bin/env python
#*-* coding: utf-8 *-*
 
"""
###############################################################################
#                        MODARCHIVE JUKEBOX SCRIPT
#   
#  Made by: Fernando Sancho AKA ‘toptnc’
#  email: toptnc@gmail.com
#
#  This script plays mods from <a href="http://modarchive.org" title="http://modarchive.org">http://modarchive.org</a> in random order
#  It can fetch files from various categories
#
#   This scripts needs the program mikmod installed under $PATH
#
#  This script is released under the terms of GNU GPL License 
#
###############################################################################
 
GPLv3 03-18-2011 
Port to python by: Ángel Guzmán Maeso (shakaran at gmail dot com) <a href="http://shakaran.net/blog/wp-content/modarchive/modarchive.py
" title="http://shakaran.net/blog/wp-content/modarchive/modarchive.py
">http://shakaran.net/blog/wp-content/modarchive/modarchive.py
</a> 
"""
from sys import exit
from os import path, makedirs, mkdir, remove, system
from urllib import urlopen
import re
import zipfile
 
VERSION = '1.0'
DEBUG = False
MODPATH = '/tmp/modarchive'
RANDOMSONG = None
MIKMOD = '/usr/bin/mikmod'
MODURL = None
 
def get_mikmod_bin():
    return MIKMOD
 
def check_mikmod():
    from os.path import exists as e
    return e(get_mikmod_bin())
 
if not check_mikmod():
    print 'This scripts needs the binary MIKMOD to run. Please install it or change the script'
    exit(1)
 
def get_url(url = None):
    try:
        return urlopen(url).read()
    except IOError, e:
        print _('Error: could not connect to the url' % url)
        return None
 
def get_data(url = None, uregex = None):
    data = get_url(url)
    #print data
    data.replace('<>', '\n')
    if uregex:
        p = re.compile(r'(%s)' % uregex)
        m = p.search(data)
 
        if m:
            return m.group()
        else:
            print 'No se encontraron resultados'
            return None
    else:
        return data
 
def get_mod_file(modurl, page):
    data = get_url('%s&page=%s' %(modurl, page))
    data.replace('href=', '\n')
    data.replace('>', '\n')
 
    p = re.compile(r'(http://modarchive.org/data/downloads.php\?moduleid=(.*)\#)')
    m = p.search(data)
 
    if m:
        module = m.group()[50 : -150]
        module = module[0:- (len(module) - module.find('#')) ]
        debug(module)
        return ('http://modarchive.org/data/downloads.php?moduleid=%s' % module, module)
    else:
        print 'No se encontraron resultados'
        return (None, None)
 
def unzip_file_into_dir(file, modID, dir):
    if not path.exists(dir + '/' + modID):
        mkdir(dir + '/' + modID, 0777)
    zfobj = zipfile.ZipFile(file)
    for name in zfobj.namelist():
        if name.endswith('/'):
            if not path.exists(path.join(dir, name)):
                mkdir(path.join(dir, name))
        else:
            outfile = open(path.join(dir + '/' + modID, name), 'wb')
            outfile.write(zfobj.read(name))
            outfile.close()
 
def debug(string = None):
    if DEBUG:
        print 'DEBUG:', string
 
def usage():
    return """
Modarchive Jukebox can be used with one of the following options:
\n\n
Hint: Use + symbol instead blankspaces in search strings."""
 
from argparse import ArgumentParser, RawTextHelpFormatter
 
parser = ArgumentParser(prog='modarchive', formatter_class=RawTextHelpFormatter, add_help=True, description=usage(), version=VERSION)
parser.add_argument('-d', '--debug', action='store_true', dest='debug', help='Enable debug mode')
 
parser.add_argument('-s', dest='section', nargs=1, type=str, choices=('uploads', 'featured', 'favourites', 'downloads', 'topscore', 'new', 'random'),
          help="""Play from selected section: Can be one of this
uploads     This is a list of the recent member upload activity.
featured    These modules have been nominated by the crew for either \n\t    outstanding quality, technique or creativity (or combination of).
favourites  These modules have been nominated by the members via their favourites.
downloads   The top 1000 most downloaded modules, recorded since circa 2002.
topscore    This chart lists the most revered modules on the archive.
new         Same than uploads but using search engine.
random      Ramdom module from entire archive.""")
 
search_group = parser.add_argument_group('Searching')
search_group.add_argument('-a', dest='artist', nargs=1, type=str, help='Search in artist database')        
search_group.add_argument('-m', dest='module', nargs=1, type=str, help="Search in module database (Title and Filename)")
 
try:
    results = parser.parse_args()
except IOError, msg:
    parser.error(str(msg))
 
if results.debug:
    DEBUG = True
 
debug(results)
 
pages = None
if results.section:
    if results.section[0] == 'uploads':
        MODURL = 'http://modarchive.org/index.php?request=view_actions_uploads'
        pages = None
    elif results.section[0] == 'featured':
        MODURL = 'http://modarchive.org/index.php?request=view_chart&query=featured'
        pages = get_data(MODURL, 'page=(\w)#mods')[5:-5]
    elif results.section[0] == 'favourites':
        MODURL = 'http://modarchive.org/index.php?request=view_top_favourites'
        pages = get_data(MODURL, 'page=(\w)#mods')[5:-5]
    elif results.section[0] == 'downloads':
        MODURL = 'http://modarchive.org/index.php?request=view_chart&query=tophits'
        pages = get_data(MODURL, 'page=(\w)#mods')[5:-5]
    elif results.section[0] == 'topscore':
        MODURL = 'http://modarchive.org/index.php?request=view_chart&query=topscore'
        pages = get_data(MODURL, 'page=(\w)#mods')[5:-5]
    elif results.section[0] == 'new':
        MODURL = 'http://modarchive.org/index.php?request=search&search_type=new_additions'
        pages = get_data(MODURL, 'page=(\w)#mods')[5:-5]
    elif results.section[0] == 'random':
        MODURL = 'http://modarchive.org/index.php?request=view_random'
        pages = 0
elif results.artist:
    MODURL = 'http://modarchive.org/index.php?query=%s&submit=Find&request=search&search_type=guessed_artist' % results.artist[0]
    pages = get_data(MODURL, 'page=(\w)#mods')[5:-5]
elif results.module:
    MODURL = 'http://modarchive.org/index.php?request=search&query=%s&submit=Find&search_type=filename_or_songtitle' % results.module[0]
    res = get_data(MODURL, 'page=(\w)#mods')[5:-5]
 
print "Starting Modarchive JukeBox Player"
debug("Pages: %s" % pages)
 
if not path.exists(MODPATH):
    debug('Creating dir %s' % MODPATH)
    makedirs(MODPATH)
 
while 1:
    print 'Press Ctrl-C twice to stop'
 
    (modfile, modID) = get_mod_file(MODURL, pages)
 
    if modfile:
        zip_file = '%s/%s.zip' %(MODPATH, modID)
        print "Downloading %s to %s" %(modfile, zip_file)
        f = open(zip_file, 'w')
        f.write(get_url(modfile))
        f.close()
        unzip_file_into_dir(zip_file, modID, MODPATH)
        remove(zip_file)
 
#   if [ ! -z $(pgrep pidgin) ];
#   then
#       purple-remote "setstatus?status=available&message=Modarchive JukeBox:  ${MODFILE}"
#   fi;
	print 'mikmod i -hqmixer --surrond X -noloops %s/%s/*' % (MODPATH, modID)
    print system('mikmod i -hqmixer --surrond X -noloops %s/%s/*' % (MODPATH, modID))
#   if [ ! -z $(pgrep pidgin) ];
#   then
#       purple-remote "setstatus?status=available&message="
#   fi;