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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
package devices
import (
"log"
"net/http"
"penguins.lan/api/types"
"penguins.lan/database"
"penguins.lan/utils"
)
// DeviceView is one device row for the profile page.
type DeviceView struct {
Mac string // canonical colon form, used as the set-primary value
Name string // <compact-mac>.<username>.<zone> FQDN
IP string
Online bool
LastSeen string
Primary bool
}
// ListDevicesContinuation loads the user's devices (with liveness + which is
// primary) and the zone facts the profile page and wizard need.
func ListDevicesContinuation(context *types.RequestContext, req *http.Request, resp http.ResponseWriter) types.ContinuationChain {
return func(success types.Continuation, failure types.Continuation) types.ContinuationChain {
if context.User == nil {
return failure(context, req, resp)
}
macs, err := database.ListUserMacs(context.DBConn, context.User.ID)
if err != nil {
log.Println("devices: list failed:", err)
resp.WriteHeader(http.StatusInternalServerError)
return failure(context, req, resp)
}
primary, err := database.GetUserPrimaryMAC(context.DBConn, context.User.ID)
if err != nil {
log.Println("devices: primary lookup failed:", err)
resp.WriteHeader(http.StatusInternalServerError)
return failure(context, req, resp)
}
zone := context.Args.DnsZone
userDomain := context.User.Username + "." + zone
views := make([]DeviceView, 0, len(macs))
for _, d := range macs {
name := d.Mac
if mac, err := utils.ParseMac(d.Mac); err == nil {
name = mac.Compact() + "." + userDomain
}
views = append(views, DeviceView{
Mac: d.Mac,
Name: name,
IP: d.IP,
Online: d.Online,
LastSeen: utils.FormatSince(d.LastSeen),
Primary: d.Mac == primary,
})
}
(*context.TemplateData)["Devices"] = views
(*context.TemplateData)["PrimaryMAC"] = primary
(*context.TemplateData)["DnsZone"] = zone
(*context.TemplateData)["UserDomain"] = userDomain
return success(context, req, resp)
}
}
// SetPrimaryContinuation marks one of the caller's devices as primary.
func SetPrimaryContinuation(context *types.RequestContext, req *http.Request, resp http.ResponseWriter) types.ContinuationChain {
return func(success types.Continuation, failure types.Continuation) types.ContinuationChain {
mac := req.FormValue("mac")
owned, err := database.MACBelongsToUser(context.DBConn, mac, context.User.ID)
if err != nil {
log.Println("devices: ownership check failed:", err)
resp.WriteHeader(http.StatusInternalServerError)
return failure(context, req, resp)
}
if !owned {
resp.WriteHeader(http.StatusUnauthorized)
return failure(context, req, resp)
}
if err := database.SetUserPrimaryMAC(context.DBConn, context.User.ID, mac); err != nil {
log.Println("devices: set primary failed:", err)
resp.WriteHeader(http.StatusInternalServerError)
return failure(context, req, resp)
}
http.Redirect(resp, req, "/profile", http.StatusSeeOther)
return success(context, req, resp)
}
}
|
