from risotto.controller import Controller
from risotto.register import register

class Risotto(Controller):
    @register('v1.servermodel.list', None)
    async def servermodel_list(self, source_id):
        return [{'servermodel_id': 1,
                 'servermodel_name': 'name1',
                 'release_id': 1,
                 'servermodel_description': 'description1'},
                {'servermodel_id': 2,
                 'servermodel_name': 'name2',
                 'release_id': 2,
                 'servermodel_description': 'description2',
                 'servermodel_parents_id': [1]}]

    @register('v1.servermodel.describe', None)
    async def servermodel_describe(self, inheritance, creolefuncs, servermodel_id, schema, conffiles, resolvdepends, probes):
        schema = """<?xml version='1.0' encoding='UTF-8'?>
<creole>
  <family name="containers">
    <family name="container0" doc="test">
      <family doc="files" name="files">
        <family doc="file0" name="file0">
          <variable doc="" multi="False" name="mkdir" type="boolean">
            <value>False</value>
          </variable>
          <variable doc="" multi="False" name="name" type="string">
            <value>/etc/mailname</value>
          </variable>
          <variable doc="" multi="False" name="rm" type="boolean">
            <value>False</value>
          </variable>
          <variable doc="" multi="False" name="source" type="string">
            <value>mailname</value>
          </variable>
          <variable doc="" multi="False" name="activate" type="boolean">
            <value>True</value>
          </variable>
        </family>
      </family>
      <property>basic</property>
    </family>
  </family>
  <family doc="" name="creole">
    <family doc="general" name="general">
      <property>normal</property>
      <variable doc="No change" multi="False" name="mode_conteneur_actif" type="choice">
        <choice type="string">oui</choice>
        <choice type="string">non</choice>
        <property>mandatory</property>
        <property>normal</property>
        <value type="string">non</value>
      </variable>
      <leader doc="master" name="master">
        <property>normal</property>
        <variable doc="master" multi="True" name="master" type="string"/>
        <variable doc="slave1" multi="True" name="slave1" type="string">
          <property>normal</property>
        </variable>
        <variable doc="slave2" multi="True" name="slave2" type="string">
          <property>normal</property>
        </variable>
      </leader>
    </family>
    <separators/>
  </family>
</creole>"""
        return {'servermodel_id': 1, 'servermodel_name': 'name', 'servermodel_description': 'description', 'release_id': 1, 'schema': schema, 'creolefuncs': ''}