summary refs log tree commit diff stats
path: root/bouml/python_imports
diff options
context:
space:
mode:
authorhut <hut@lavabit.com>2009-12-06 14:01:49 +0100
committerhut <hut@lavabit.com>2009-12-06 14:01:49 +0100
commit65cb1a32d4d5f83f48169846afec201ab3b6f9c3 (patch)
treeccd022db0eedc2377e0318085174b96a20d614b9 /bouml/python_imports
parent465bff736d234e57efb2e9232df8882f3fd3a5cb (diff)
downloadranger-65cb1a32d4d5f83f48169846afec201ab3b6f9c3.tar.gz
"SettingsAware", random cleanups
Diffstat (limited to 'bouml/python_imports')
0 files changed, 0 insertions, 0 deletions
ommit/http.go?h=v0.3.0&id=1168570b071d8bb483e4028f3a8c45a222c93ecb'>1168570 ^
ec2c4ed ^
1168570 ^

df79157 ^
1168570 ^
d3d9df2 ^

1168570 ^
df1d1ef ^
747f4fb ^
1168570 ^


893123c ^
fc49c57 ^
df1d1ef ^
893123c ^
b9d16fb ^
1168570 ^

747f4fb ^
1168570 ^



b9d16fb ^
d3d9df2 ^
1168570 ^

b6e7bb3 ^
1168570 ^

b9d16fb ^
d3d9df2 ^
505c5d8 ^
37be25f ^
505c5d8 ^

37be25f ^
505c5d8 ^

1168570 ^

d2680fb ^
1168570 ^


d2680fb ^

1168570 ^

fc49c57 ^
1168570 ^


d6fbc25 ^
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68


            
                 
             
             
                  
                 

 

                                                                         
 

                                                
 
                                                  

 
                                                              
                                               
 


                                                                 
         
 
                               
 
 

                                                          
 



                                                                              
 
 

                              
 

                                                                    
 
 
                      
                                                                 

                                                                               
                                                                     

 

                                                                
 


                                                                               

 

                                                                
 


                                                                               
 
package main

import (
	"context"
	"log"
	"net"
	"net/http"
	"strings"
)

// Attaches a request's IP address to the request's context
func newCtxUserIP(ctx context.Context, r *http.Request) context.Context {

	base := strings.Split(r.RemoteAddr, ":")
	uip := base[0]

	return context.WithValue(ctx, ctxKey, uip)
}

// Retrieves a request's IP address from the request's context
func getIPFromCtx(ctx context.Context) net.IP {

	uip, ok := ctx.Value(ctxKey).(string)
	if !ok {
		log.Printf("Couldn't retrieve IP from request\n")
	}

	return net.ParseIP(uip)
}

// Shim function to modify/pass context value to a handler
func ipMiddleware(hop http.Handler) http.Handler {

	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		ctx := newCtxUserIP(r.Context(), r)
		hop.ServeHTTP(w, r.WithContext(ctx))
	})
}

// log output for 200s
func log200(r *http.Request) {

	uip := getIPFromCtx(r.Context())
	log.Printf("*** %v :: 200 :: %v %v\n", uip, r.Method, r.URL)
}

// log output for 400s
func log400(w http.ResponseWriter, r *http.Request, err string) {
	uip := getIPFromCtx(r.Context())
	log.Printf("*** %v :: 400 :: %v %v :: %v\n", uip, r.Method, r.URL, err)
	http.Error(w, "400 Bad Request: "+err, http.StatusBadRequest)
}

// log output for 404s
func log404(w http.ResponseWriter, r *http.Request, err error) {

	uip := getIPFromCtx(r.Context())
	log.Printf("*** %v :: 404 :: %v %v :: %v\n", uip, r.Method, r.URL, err)
	http.Error(w, err.Error(), http.StatusNotFound)
}

// log output for 500s
func log500(w http.ResponseWriter, r *http.Request, err error) {

	uip := getIPFromCtx(r.Context())
	log.Printf("*** %v :: 500 :: %v %v :: %v\n", uip, r.Method, r.URL, err)
	http.Error(w, err.Error(), http.StatusInternalServerError)
}