X-Git-Url: http://nitlanguage.org

diff --git a/src/web/api_graph.nit b/src/web/api_graph.nit
index 67a32e2..13a2177 100644
--- a/src/web/api_graph.nit
+++ b/src/web/api_graph.nit
@@ -19,18 +19,10 @@ import web_base
 import dot
 import uml
 
-# Group all api handlers in one router.
-class APIGraphRouter
-	super Router
-
-	# Model to pass to handlers.
-	var model: Model
-
-	# Mainmodule to pass to handlers.
-	var mainmodule: MModule
-
+redef class APIRouter
 	init do
-		use("/inheritance/:id", new APIInheritanceGraph(model, mainmodule))
+		super
+		use("/graph/inheritance/:id", new APIInheritanceGraph(config))
 	end
 end
 
@@ -39,14 +31,11 @@ class APIInheritanceGraph
 	super APIHandler
 
 	redef fun get(req, res) do
+		var mentity = mentity_from_uri(req, res)
+		if mentity == null then return
 		var pdepth = req.int_arg("pdepth")
 		var cdepth = req.int_arg("cdepth")
-		var mentity = mentity_from_uri(req, res)
-		if mentity == null then
-			res.error 404
-			return
-		end
-		var g = new InheritanceGraph(mentity, view)
+		var g = new InheritanceGraph(mentity, config.view)
 		res.send g.draw(pdepth, cdepth).to_svg
 	end
 end