Initial open source release

internal/rpc/context.go

@@ -0,0 +1,96 @@
+package rpc
+
+import "context"
+
+type ctxKey int
+
+const (
+	layerKey ctxKey = iota
+	clientInfoKey
+	rawAuthKeyIDKey
+	authKeyIDKey
+	sessionIDKey
+	userIDKey
+)
+
+// ClientInfo 是 initConnection 携带的客户端信息。
+type ClientInfo struct {
+	APIID          int
+	DeviceModel    string
+	SystemVersion  string
+	AppVersion     string
+	SystemLangCode string
+	LangPack       string
+	LangCode       string
+}
+
+// WithLayer 在 ctx 注入客户端 layer（来自 invokeWithLayer）。
+func WithLayer(ctx context.Context, layer int) context.Context {
+	return context.WithValue(ctx, layerKey, layer)
+}
+
+// LayerFrom 返回 ctx 中的客户端 layer，未设置时为 0。
+func LayerFrom(ctx context.Context) int {
+	if v, ok := ctx.Value(layerKey).(int); ok {
+		return v
+	}
+	return 0
+}
+
+// WithClientInfo 在 ctx 注入客户端信息（来自 initConnection）。
+func WithClientInfo(ctx context.Context, info ClientInfo) context.Context {
+	return context.WithValue(ctx, clientInfoKey, info)
+}
+
+// ClientInfoFrom 返回 ctx 中的客户端信息。
+func ClientInfoFrom(ctx context.Context) (ClientInfo, bool) {
+	v, ok := ctx.Value(clientInfoKey).(ClientInfo)
+	return v, ok
+}
+
+// WithRawAuthKeyID 在 ctx 注入连接实际使用的 auth_key_id。
+func WithRawAuthKeyID(ctx context.Context, id [8]byte) context.Context {
+	return context.WithValue(ctx, rawAuthKeyIDKey, id)
+}
+
+// RawAuthKeyIDFrom 返回连接实际使用的 auth_key_id。
+func RawAuthKeyIDFrom(ctx context.Context) ([8]byte, bool) {
+	v, ok := ctx.Value(rawAuthKeyIDKey).([8]byte)
+	return v, ok
+}
+
+// WithAuthKeyID 在 ctx 注入业务视角的 auth_key_id；temp auth_key 绑定后会解析为 perm auth_key。
+func WithAuthKeyID(ctx context.Context, id [8]byte) context.Context {
+	return context.WithValue(ctx, authKeyIDKey, id)
+}
+
+// AuthKeyIDFrom 返回 ctx 中业务视角的 auth_key_id。已握手连接均有（即便尚未登录）。
+func AuthKeyIDFrom(ctx context.Context) ([8]byte, bool) {
+	v, ok := ctx.Value(authKeyIDKey).([8]byte)
+	return v, ok
+}
+
+// WithSessionID 在 ctx 注入调用方的 MTProto session_id。
+func WithSessionID(ctx context.Context, id int64) context.Context {
+	return context.WithValue(ctx, sessionIDKey, id)
+}
+
+// SessionIDFrom 返回 ctx 中调用方的 MTProto session_id。
+func SessionIDFrom(ctx context.Context) (int64, bool) {
+	v, ok := ctx.Value(sessionIDKey).(int64)
+	return v, ok
+}
+
+// WithUserID 在 ctx 注入当前已登录用户 id。
+func WithUserID(ctx context.Context, id int64) context.Context {
+	return context.WithValue(ctx, userIDKey, id)
+}
+
+// UserIDFrom 返回 ctx 中当前已登录用户 id。
+func UserIDFrom(ctx context.Context) (int64, bool) {
+	v, ok := ctx.Value(userIDKey).(int64)
+	if !ok || v == 0 {
+		return 0, false
+	}
+	return v, true
+}