Add urlfilter

2 files changed, 30 insertions, 5 deletions

diff --git a/src/io/loader.nim b/src/io/loader.nim
index 150831a0..dbca256b 100644
--- a/src/io/loader.nim
+++ b/src/io/loader.nim
@@ -22,6 +22,7 @@ import bindings/curl
 import io/about
 import io/http
 import io/request
+import io/urlfilter
 import ips/serialize
 import ips/serversocket
 import ips/socketstream
@@ -83,7 +84,7 @@ proc loadResource(request: Request, ostream: Stream) =
     ostream.flush()
 
 var ssock: ServerSocket
-proc runFileLoader*(fd: cint, defaultHeaders: HeaderList) =
+proc runFileLoader*(fd: cint, defaultHeaders: HeaderList, filter: URLFilter) =
   if curl_global_init(CURL_GLOBAL_ALL) != CURLE_OK:
     raise newException(Defect, "Failed to initialize libcurl.")
   ssock = initServerSocket()
@@ -108,10 +109,14 @@ proc runFileLoader*(fd: cint, defaultHeaders: HeaderList) =
       of LOAD:
         var request: Request
         stream.sread(request)
-        for k, v in defaultHeaders.table:
-          if k notin request.headers.table:
-            request.headers.table[k] = v
-        loadResource(request, stream)
+        if not filter.match(request.url):
+          stream.swrite(-1) # error
+          stream.flush()
+        else:
+          for k, v in defaultHeaders.table:
+            if k notin request.headers.table:
+              request.headers.table[k] = v
+          loadResource(request, stream)
         stream.close()
       of QUIT:
         stream.close()
diff --git a/src/io/urlfilter.nim b/src/io/urlfilter.nim
new file mode 100644
index 00000000..6c0b71c9
--- /dev/null
+++ b/src/io/urlfilter.nim
@@ -0,0 +1,20 @@
+import options
+
+import types/url
+
+type URLFilter* = object
+  scheme: Option[string]
+  host: Option[string]
+
+proc newURLFilter*(scheme = none(string), host = none(string)): URLFilter =
+  return URLFilter(
+    scheme: scheme,
+    host: host
+  )
+
+func match*(filter: URLFilter, url: URL): bool =
+  if filter.scheme.isSome and filter.scheme.get != url.scheme:
+    return false
+  if filter.host.isSome and filter.host.get != url.host:
+    return false
+  return true