1 module dweb.server; 2 3 import std.socket; 4 import std.stdio; 5 import std.array; 6 import std.conv; 7 import dweb.utils; 8 import dweb.view_engine; 9 10 public class HttpServer{ 11 private Address address; 12 private Socket socket; 13 private void function(Request, Response)[string] gettable; 14 private void function(Request, Response)[string] posttable; 15 private void function(Request, Response) notf; 16 private void function(Request, Response, void function(Request, Response)) ever; 17 public ViewEngine view_engine; 18 this(string host, ushort port){ 19 this.address = new InternetAddress(host, port); 20 this.socket = new TcpSocket(AddressFamily.UNSPEC); 21 this.socket.bind(this.address); 22 this.notf = &nf; 23 this.view_engine = new ViewEngine(); 24 this.ever = null; 25 } 26 public void setViewEngine(ViewEngine newve){ 27 this.view_engine = newve; 28 } 29 public void listen(){ 30 this.socket.listen(0); 31 while(true){ 32 Socket client = this.socket.accept(); 33 char[] buffer = new char[1024 * 1024]; 34 uint received = client.receive(buffer).to!uint(); 35 if(received != 0){ 36 buffer = buffer[0 .. received]; 37 Request req = parseRequest(buffer); 38 Response res = new Response(client, this); 39 40 res.setHeader("Content-Type", "text/html; charset=utf-8"); 41 42 if(req.Method == "GET"){ 43 44 if(containsKeyAssoc!(string, void function(Request, Response))(req.Path, gettable)){ 45 if(ever !is null) 46 ever(req, res, gettable[req.Path]); 47 else 48 gettable[req.Path](req, res); 49 } 50 else 51 notf(req, res); 52 } 53 else if(req.Method == "POST"){ 54 if(containsKeyAssoc!(string, void function(Request, Response))(req.Path, posttable)){ 55 if(ever !is null) 56 ever(req, res, posttable[req.Path]); 57 else 58 posttable[req.Path](req, res); 59 } 60 else 61 notf(req, res); 62 } 63 if(!res.isSended){ 64 res.send(); 65 } 66 client.shutdown(SocketShutdown.BOTH); 67 client.close(); 68 } 69 70 } 71 } 72 public void get(string path, void function(Request, Response) cb){ 73 gettable[path] = cb; 74 } 75 public void post(string path, void function(Request, Response) cb){ 76 posttable[path] = cb; 77 } 78 public void notfound(void function(Request, Response) cb){ 79 this.notf = cb; 80 } 81 82 } 83 void nf(Request req, Response res){ 84 res.text("<center style=\"margin-top: 30vh;\">"); 85 res.text("<h1>404 Not Found</h1>"); 86 res.text("<h3>Path \"" ~ req.Path ~ "\" not found"); 87 res.text("</center>"); 88 res.send(404); 89 }