forked from progrium/envy
/
http.go
52 lines (49 loc) · 1.27 KB
/
http.go
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
package envy
import (
"fmt"
"log"
"net/http"
"os"
"os/exec"
"strings"
"github.com/progrium/envy/pkg/hterm"
)
func init() {
http.HandleFunc("/u/", func(w http.ResponseWriter, r *http.Request) {
parts := strings.Split(r.URL.Path, "/")
if len(parts) < 3 {
http.NotFound(w, r)
return
}
pathUser := parts[2]
var pathEnv, sshUser string
if len(parts) > 3 && parts[3] != "hterm" {
pathEnv = parts[3]
sshUser = pathUser + "+" + pathEnv
} else {
sshUser = pathUser
}
// passthrough auth for hterm. use cookie to do this right
if !strings.Contains(r.URL.Path, "hterm") {
user, passwd, ok := r.BasicAuth()
if !ok || user != pathUser || !githubUserAuth(user, passwd) {
w.Header().Set("WWW-Authenticate", fmt.Sprintf("Basic realm=\"%s\"", pathUser))
http.Error(w, "Unauthorized", http.StatusUnauthorized)
log.Println("auth[http]: not allowing", user)
return
}
log.Println("auth[http]: allowing", user)
}
w.Header().Set("Hterm-Title", "Envy Term")
hterm.Handle(w, r, func(args string) *hterm.Pty {
cmd := exec.Command("/bin/enter", parts[2])
cmd.Env = os.Environ()
cmd.Env = append(cmd.Env, fmt.Sprintf("USER=%s", sshUser))
pty, err := hterm.NewPty(cmd)
if err != nil {
log.Fatal(err)
}
return pty
})
})
}