template and handler (#16)

pareshpawar · web-flow · commit a77411e74fb2 · 2024-10-09T15:41:25.000+05:30
diff --git a/html/index.html b/html/index.html
@@ -4,14 +4,56 @@
         <meta charset="UTF-8" />
         <meta name="viewport" content="width=device-width, initial-scale=1.0" />
         <title>Simple HTTP Server</title>
+        <style>
+            body {
+              font-family: Arial, sans-serif;
+              margin: 20px;
+            }
+            h1, h3 {
+              text-align: center;
+            }
+            ul {
+              list-style-type: none;
+              padding: 0;
+            }
+            li {
+              margin-bottom: 10px;
+            }
+            table {
+              border-collapse: collapse;
+              width: 100%;
+            }
+            th, td {
+              border: 1px solid #ccc;
+              padding: 8px;
+              text-align: left;   
+        
+            }
+            th {
+              background-color: #f2f2f2;
+            }
+          </style>
     </head>
     <body>
         <h1>Welcome to Simple HTTP Server</h1>
         <h3>by Paresh Pawar</h3>
         <ul>
-            <li>Origin IP: {{.Origin}}</li>
-            <li>Request Method: {{.Type}}</li>
-            <li>Destination: {{.Hostname}}</li>
+            <li>Request: {{.Request.ReqTime}}</li>
+            <li>Type: {{.Request.ReqType}}</li>
+            <li>Host: {{.Request.Host}}</li>
+            <li>Remote Address: {{.Request.RemoteAddr}}</li>
         </ul>
+        <table>
+            <tr>
+                <th>Key</th>
+                <th>Value</th>
+            </tr>
+            {{ range $key, $value := .Headers }}
+            <tr>
+                <td>{{ $key }}</td>
+                <td>{{ $value }}</td>
+            </tr>
+            {{ end }}
+        </table>
     </body>
 </html>
diff --git a/simple_http_server.go b/simple_http_server.go
@@ -30,22 +30,42 @@ func check(err error) {
 	}
 }
 
-func htmlhandler(w http.ResponseWriter, r *http.Request){
+func htmlhandler(w http.ResponseWriter, r *http.Request) {
+	timestamp := time.Now()
+	if r.Method == "GET" {
+		fmt.Print(string("\033[34m"))
+	} else if r.Method == "POST" {
+		fmt.Print(string("\033[33m"))
+	} else if r.Method == "PUT" {
+		fmt.Print(string("\033[35m"))
+	} else if r.Method == "DELETE" {
+		fmt.Print(string("\033[31m"))
+	} else {
+		fmt.Print(string("\033[36m"))
+	}
+	fmt.Printf("%s %s %s %s  ===> from %s\n", timestamp.Local(), r.Method, r.URL, r.Proto, r.RemoteAddr)
 
 	file, err := os.ReadFile("html/index.html")
 	check(err)
 
 	template, err := template.New("webpage").Parse(string(file))
 	check(err)
 
+	type reqDataStruct struct{ ReqTime, ReqType, Host, Remote, RemoteAddr string }
+	reqData := reqDataStruct{
+		ReqTime:    timestamp.String(),
+		ReqType:    r.Proto + " " + r.Method + " " + r.URL.Path,
+		Host:       r.Host,
+		Remote:     r.RemoteAddr,
+		RemoteAddr: utils.GetMyOutboundIP().String(),
+	}
+
 	data := struct {
-		Origin string
-		Type string
-		Hostname string
+		Request reqDataStruct
+		Headers http.Header
 	}{
-		Origin: r.RemoteAddr,
-		Type: r.Method,
-		Hostname: r.Host,
+		Request: reqData,
+		Headers: r.Header,
 	}
 
 	err = template.Execute(w, data)
