Open Source Projects Europe

#!/usr/bin/env python

#

# Copyright (C) 2016 J.F.Dockes

#

# This program is free software: you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program.  If not, see <http://www.gnu.org/licenses/>.

from __future__ import print_function

import sys

import os

import json

import posixpath

import re

import conftree

import cmdtalkplugin

import folders

from uprclutils import *

# Func name to method mapper

dispatcher = cmdtalkplugin.Dispatch()

# Pipe message handler

msgproc = cmdtalkplugin.Processor(dispatcher)

is_logged_in = False

def maybelogin():

    global httphp

    global pathprefix

    global is_logged_in

    if is_logged_in:

        return True

    if "UPMPD_HTTPHOSTPORT" not in os.environ:

        raise Exception("No UPMPD_HTTPHOSTPORT in environment")

    httphp = os.environ["UPMPD_HTTPHOSTPORT"]

    if "UPMPD_PATHPREFIX" not in os.environ:

        raise Exception("No UPMPD_PATHPREFIX in environment")

    pathprefix = os.environ["UPMPD_PATHPREFIX"]

    if "UPMPD_CONFIG" not in os.environ:

        raise Exception("No UPMPD_CONFIG in environment")

    upconfig = conftree.ConfSimple(os.environ["UPMPD_CONFIG"])

@dispatcher.record('trackuri')

def trackuri(a):

    msgproc.log("trackuri: [%s]" % a)

    trackid = trackid_from_urlpath(pathprefix, a)

    maybelogin()

    mime, kbs = get_mimeandkbs()

    return {'media_url' : media_url, 'mimetype' : mime, 'kbs' : kbs}

@dispatcher.record('browse')

def browse(a):

    msgproc.log("browse: [%s]" % a)

    if 'objid' not in a:

        raise Exception("No objid in args")

    objid = a['objid']

    bflg = a['flag'] if 'flag' in a else 'children'

    if re.match('0\$uprcl\$', objid) is None:

        raise Exception("bad objid [%s]" % objid)

    maybelogin()

    idpath = objid.replace('0$uprcl$', '', 1)

    msgproc.log("browse: idpath: %s" % idpath)

    entries = []

    if bflg == 'meta':

        m = re.match('.*\$(.+)$', idpath)

        if m:

            trackid = m.group(1)

            # get doc from trackid or whatever

            doc = {}

            entries += trackentries(httphp, pathprefix,

                                    objid, [])

    else:

        if not idpath:

            # Root dir

            entries.append(rcldirentry('0$uprcl$' + 'folders', '0$uprcl$',

                                        '[folders]'))

        if idpath.startswith("folders"):

            entries = folders.browse(objid, bflg)

        else:

            pass

    #msgproc.log("%s" % entries)

    encoded = json.dumps(entries)

    return {"entries" : encoded}

@dispatcher.record('search')

def search(a):

    msgproc.log("search: [%s]" % a)

    objid = a['objid']

    field = a['field'] if 'field' in a else None

    value = a['value']

    objkind = a['objkind'] if 'objkind' in a else None

    if re.match('0\$uprcl\$', objid) is None:

        raise Exception("bad objid [%s]" % objid)

    maybelogin()

    searchresults = session.search(value)

    if objkind and objkind not in ['artist', 'album', 'playlist', 'track']:

        msgproc.log('Unknown objkind \'%s\'' % objkind)

        objkind = None

    if objkind is None or objkind == 'artist':

        view(searchresults.artists,

             urls_from_id(artist_view, searchresults.artists), end=False)

    if objkind is None or objkind == 'album':

        view(searchresults.albums,

             urls_from_id(album_view, searchresults.albums), end=False)

    if objkind is None or objkind == 'playlist':

        view(searchresults.playlists,

             urls_from_id(playlist_view, searchresults.playlists), end=False)

    if objkind is None or objkind == 'track':

        track_list(searchresults.tracks)

    #msgproc.log("%s" % xbmcplugin.entries)

    encoded = json.dumps(xbmcplugin.entries)

    return {"entries" : encoded}

msgproc.log("Uprcl running")

msgproc.mainloop()