"""Level menu."""

from mamba.habitats.levelmenu import LevelMenu
from mamba.level import Level

import urllib2


class UserLevelMenu(LevelMenu):

    LEVEL_SERVER_URL = "http://localhost:5000/"

    def list_levels(self):
        url = "%s%s" % (self.LEVEL_SERVER_URL, "index")
        data = urllib2.urlopen(url).read()
        levels = [x.strip() for x in data.splitlines()]
        return levels

    def get_level(self, name):
        url = "%s%s/%s" % (self.LEVEL_SERVER_URL, "level", name)
        source = urllib2.urlopen(url).read()
        return Level(name, source)