Integrate boot.ipfire.org application with the main webapp.

[people/shoehn/ipfire.org.git] / www / webapp / handlers_boot.py

diff --git a/www/webapp/handlers_boot.py b/www/webapp/handlers_boot.py
new file mode 100644 (file)
index 0000000..22acce5
--- /dev/null
+++ b/www/webapp/handlers_boot.py
@@ -0,0 +1,117 @@
+#!/usr/bin/python
+
+import logging
+import os
+import tornado.httpserver
+import tornado.ioloop
+import tornado.locale
+import tornado.options
+import tornado.web
+
+import backend
+
+BASEDIR = os.path.dirname(__file__)
+
+def word_wrap(s, width=45):
+       paragraphs = s.split('\n')
+       lines = []
+       for paragraph in paragraphs:
+               while len(paragraph) > width:
+                       pos = paragraph.rfind(' ', 0, width)
+                       if not pos:
+                               pos = width
+                       lines.append(paragraph[:pos])
+                       paragraph = paragraph[pos:]
+               lines.append(paragraph.lstrip())
+       return '\n'.join(lines)
+
+class BootBaseHandler(tornado.web.RequestHandler):
+       @property
+       def netboot(self):
+               return backend.NetBoot()
+
+
+class MenuGPXEHandler(BootBaseHandler):
+       """
+               menu.gpxe
+       """
+       def get(self):
+               # XXX Check if version of the bootloader is allright
+
+               # Devliver content
+               self.set_header("Content-Type", "text/plain")
+               self.write("#!gpxe\n")
+               self.write("chain menu.c32 premenu.cfg\n")
+
+
+class MenuCfgHandler(BootBaseHandler):
+       def _menu_string(self, menu, level=0):
+               s = ""
+
+               for entry in menu:
+                       s += self._menu_entry(entry, level=level)
+
+               return s
+
+       def _menu_entry(self, entry, level=0):
+               lines = []
+
+               ident = "\t" * level
+
+               if entry.type == "seperator":
+                       lines.append(ident + "menu separator")
+
+               elif entry.type == "header":
+                       lines.append(ident + "menu begin %d" % entry.id)
+                       lines.append(ident + "\tmenu title %s" % entry.title)
+
+                       # Add "Back..." entry
+                       lines.append(ident + "\tlabel %d.back" % entry.id)
+                       lines.append(ident + "\t\tmenu label Back...")
+                       lines.append(ident + "\t\tmenu exit")
+                       lines.append(ident + "\tmenu separator")
+
+                       lines.append("%s" % self._menu_string(entry.submenu, level=level+1))
+                       lines.append(ident + "menu end")
+
+               elif entry.type == "config":
+                       lines.append(ident + "label %d" % entry.id)
+                       lines.append(ident + "\tmenu label %s" % entry.title)
+                       if entry.description:
+                               lines.append(ident + "\ttext help")
+                               lines.append(word_wrap(entry.description))
+                               lines.append(ident + "\tendtext")
+                       lines.append(ident + "\tkernel /config/%s/boot.gpxe" % entry.item)
+
+               return "\n".join(lines + [""])
+
+       def get(self):
+               self.set_header("Content-Type", "text/plain")
+
+               menu = self._menu_string(self.netboot.get_menu(1))
+
+               self.render("menu.cfg", menu=menu)
+
+
+class BootGPXEHandler(BootBaseHandler):
+       def get(self, id):
+               config = self.netboot.get_config(id)
+               if not config:
+                       raise tornado.web.HTTPError(404, "Configuration with ID '%s' was not found" % id)
+
+               lines = ["#!gpxe", "imgfree",]
+
+               line = "kernel -n img %s" % config.image1
+               if line.endswith(".iso"):
+                       line += " iso"
+               if config.args:
+                       line += " %s" % config.args
+               lines.append(line)
+
+               if config.image2:
+                       lines.append("initrd -n img %s" % config.image2)
+
+               lines.append("boot img")
+
+               for line in lines:
+                       self.write("%s\n" % line)