Put access_key_type to plugin request context

pkg/server/plugin/exec/process_test.go

@@ -175,14 +175,16 @@ func TestRun(t *testing.T) {
 			executed := false
 			startCommand = func(cmd *exec.Cmd, in []byte) (out []byte, err error) {
 				So(cmd, shouldRunWithContext, map[string]interface{}{
-					"user_id": "user",
+					"user_id":         "user",
+					"access_key_type": "master",
 				})
 				So(cmd, shouldRunWithConfig, appconfig)
 				executed = true
 				return []byte(`{"result": {}}`), nil
 			}
 
 			ctx := context.WithValue(context.Background(), router.UserIDContextKey, "user")
+			ctx = context.WithValue(ctx, router.AccessKeyTypeContextKey, router.MasterAccessKey)
 			transport.RunLambda(ctx, "work", []byte{})
 			So(executed, ShouldBeTrue)
 		})
@@ -625,14 +627,16 @@ func TestRun(t *testing.T) {
 			executed := false
 			startCommand = func(cmd *exec.Cmd, in []byte) (out []byte, err error) {
 				So(cmd, shouldRunWithContext, map[string]interface{}{
-					"user_id": "user",
+					"user_id":         "user",
+					"access_key_type": "master",
 				})
 				So(cmd, shouldRunWithConfig, appconfig)
 				executed = true
 				return []byte(`{"result": {}}`), nil
 			}
 
 			ctx := context.WithValue(context.Background(), router.UserIDContextKey, "user")
+			ctx = context.WithValue(ctx, router.AccessKeyTypeContextKey, router.MasterAccessKey)
 			transport.RunHook(ctx, "note_afterSave", &recordin, nil)
 			So(executed, ShouldBeTrue)
 		})

pkg/server/plugin/http/http_test.go

@@ -114,11 +114,12 @@ func TestRun(t *testing.T) {
 
 		Convey("run lambda", func() {
 			ctx := context.WithValue(context.Background(), router.UserIDContextKey, "user")
+			ctx = context.WithValue(ctx, router.AccessKeyTypeContextKey, router.MasterAccessKey)
 			data := `{"data": "bye"}`
 			httpmock.RegisterResponder("POST", "http://localhost:8000",
 				func(req *http.Request) (*http.Response, error) {
 					out, _ := ioutil.ReadAll(req.Body)
-					So(out, ShouldEqualJSON, `{"context":{"user_id":"user"},"kind":"op","name":"john","param":{"data":"bye"}}`)
+					So(out, ShouldEqualJSON, `{"context":{"user_id":"user","access_key_type":"master"},"kind":"op","name":"john","param":{"data":"bye"}}`)
 					return httpmock.NewJsonResponse(200, map[string]interface{}{
 						"result": map[string]interface{}{"data": "hello"},
 					})
@@ -132,6 +133,7 @@ func TestRun(t *testing.T) {
 
 		Convey("run hook", func() {
 			ctx := context.WithValue(context.Background(), router.UserIDContextKey, "user")
+			ctx = context.WithValue(ctx, router.AccessKeyTypeContextKey, router.ClientAccessKey)
 
 			recordin := skydb.Record{
 				ID:      skydb.NewRecordID("note", "id"),
@@ -169,7 +171,7 @@ func TestRun(t *testing.T) {
 				func(req *http.Request) (*http.Response, error) {
 					in, _ := ioutil.ReadAll(req.Body)
 					So(in, ShouldEqualJSON, `{
-						"context":{"user_id":"user"},
+						"context":{"user_id":"user","access_key_type":"client"},
 						"kind":"hook",
 						"name":"beforeSave",
 						"param":{

pkg/server/plugin/transport.go

@@ -104,5 +104,13 @@ func ContextMap(ctx context.Context) map[string]interface{} {
 	if userID, ok := ctx.Value(router.UserIDContextKey).(string); ok {
 		pluginCtx["user_id"] = userID
 	}
+	if accessKeyType, ok := ctx.Value(router.AccessKeyTypeContextKey).(router.AccessKeyType); ok {
+		switch accessKeyType {
+		case router.ClientAccessKey:
+			pluginCtx["access_key_type"] = "client"
+		case router.MasterAccessKey:
+			pluginCtx["access_key_type"] = "master"
+		}
+	}
 	return pluginCtx
 }

pkg/server/plugin/transport_test.go

@@ -115,8 +115,10 @@ func TestContextMap(t *testing.T) {
 	Convey("UserID", t, func() {
 		ctx := context.Background()
 		ctx = context.WithValue(ctx, router.UserIDContextKey, "42")
+		ctx = context.WithValue(ctx, router.AccessKeyTypeContextKey, router.MasterAccessKey)
 		So(ContextMap(ctx), ShouldResemble, map[string]interface{}{
-			"user_id": "42",
+			"user_id":         "42",
+			"access_key_type": "master",
 		})
 	})
 }

pkg/server/preprocessor/auth.go

@@ -36,6 +36,7 @@ func checkRequestAccessKey(payload *router.Payload, clientKey string, masterKey
 	} else {
 		return skyerr.NewErrorf(skyerr.AccessKeyNotAccepted, "Cannot verify api key: `%v`", apiKey)
 	}
+	payload.Context = context.WithValue(payload.Context, router.AccessKeyTypeContextKey, payload.AccessKey)
 	return nil
 }
 

pkg/server/router/model.go

@@ -29,6 +29,7 @@ import (
 type ContextKey string
 
 var UserIDContextKey ContextKey = "UserID"
+var AccessKeyTypeContextKey ContextKey = "AccessKeyType"
 
 // HandlerFunc specifies the function signature of a request handler function
 type HandlerFunc func(*Payload, *Response)