-
Notifications
You must be signed in to change notification settings - Fork 1.7k
/
vrf_keys_controller.go
125 lines (106 loc) · 3.46 KB
/
vrf_keys_controller.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
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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
package web
import (
"io"
"net/http"
"github.com/gin-gonic/gin"
"github.com/smartcontractkit/chainlink/v2/core/logger/audit"
"github.com/smartcontractkit/chainlink/v2/core/services/chainlink"
"github.com/smartcontractkit/chainlink/v2/core/web/presenters"
)
// VRFKeysController manages VRF keys
type VRFKeysController struct {
App chainlink.Application
}
// Index lists VRF keys
// Example:
// "GET <application>/keys/vrf"
func (vrfkc *VRFKeysController) Index(c *gin.Context) {
keys, err := vrfkc.App.GetKeyStore().VRF().GetAll()
if err != nil {
jsonAPIError(c, http.StatusInternalServerError, err)
return
}
jsonAPIResponse(c, presenters.NewVRFKeyResources(keys, vrfkc.App.GetLogger()), "vrfKey")
}
// Create and return a VRF key
// Example:
// "POST <application>/keys/vrf"
func (vrfkc *VRFKeysController) Create(c *gin.Context) {
pk, err := vrfkc.App.GetKeyStore().VRF().Create()
if err != nil {
jsonAPIError(c, http.StatusInternalServerError, err)
return
}
vrfkc.App.GetAuditLogger().Audit(audit.KeyCreated, map[string]interface{}{
"type": "vrf",
"id": pk.ID(),
"vrfPublicKey": pk.PublicKey,
"vrfPublicKeyAddress": pk.PublicKey.Address(),
})
jsonAPIResponse(c, presenters.NewVRFKeyResource(pk, vrfkc.App.GetLogger()), "vrfKey")
}
// Delete a VRF key
// Example:
// "DELETE <application>/keys/vrf/:keyID"
// "DELETE <application>/keys/vrf/:keyID?hard=true"
func (vrfkc *VRFKeysController) Delete(c *gin.Context) {
keyID := c.Param("keyID")
key, err := vrfkc.App.GetKeyStore().VRF().Get(keyID)
if err != nil {
jsonAPIError(c, http.StatusNotFound, err)
return
}
_, err = vrfkc.App.GetKeyStore().VRF().Delete(keyID)
if err != nil {
jsonAPIError(c, http.StatusInternalServerError, err)
return
}
vrfkc.App.GetAuditLogger().Audit(audit.KeyDeleted, map[string]interface{}{
"type": "vrf",
"id": keyID,
})
jsonAPIResponse(c, presenters.NewVRFKeyResource(key, vrfkc.App.GetLogger()), "vrfKey")
}
// Import imports a VRF key
// Example:
// "Post <application>/keys/vrf/import"
func (vrfkc *VRFKeysController) Import(c *gin.Context) {
defer vrfkc.App.GetLogger().ErrorIfFn(c.Request.Body.Close, "Error closing Import request body")
bytes, err := io.ReadAll(c.Request.Body)
if err != nil {
jsonAPIError(c, http.StatusBadRequest, err)
return
}
oldPassword := c.Query("oldpassword")
key, err := vrfkc.App.GetKeyStore().VRF().Import(bytes, oldPassword)
if err != nil {
jsonAPIError(c, http.StatusInternalServerError, err)
return
}
vrfkc.App.GetAuditLogger().Audit(audit.KeyImported, map[string]interface{}{
"type": "vrf",
"id": key.ID(),
"vrfPublicKey": key.PublicKey,
"vrfPublicKeyAddress": key.PublicKey.Address(),
})
jsonAPIResponse(c, presenters.NewVRFKeyResource(key, vrfkc.App.GetLogger()), "vrfKey")
}
// Export exports a VRF key
// Example:
// "Post <application>/keys/vrf/export/:keyID"
func (vrfkc *VRFKeysController) Export(c *gin.Context) {
defer vrfkc.App.GetLogger().ErrorIfFn(c.Request.Body.Close, "Error closing Export request body")
keyID := c.Param("keyID")
// New password to re-encrypt the export with
newPassword := c.Query("newpassword")
bytes, err := vrfkc.App.GetKeyStore().VRF().Export(keyID, newPassword)
if err != nil {
jsonAPIError(c, http.StatusInternalServerError, err)
return
}
vrfkc.App.GetAuditLogger().Audit(audit.KeyExported, map[string]interface{}{
"type": "vrf",
"id": keyID,
})
c.Data(http.StatusOK, MediaType, bytes)
}