mirror of
https://github.com/syumai/workers.git
synced 2025-03-11 01:39:11 +00:00
59 lines
2.0 KiB
Go
59 lines
2.0 KiB
Go
package cfruntimecontext
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"syscall/js"
|
|
|
|
"github.com/syumai/workers/internal/runtimecontext"
|
|
)
|
|
|
|
/**
|
|
* The type definition of RuntimeContext for Cloudflare Worker expects:
|
|
* ```ts
|
|
* type RuntimeContext {
|
|
* env: Env;
|
|
* ctx: ExecutionContext;
|
|
* ...
|
|
* }
|
|
* ```
|
|
* This type is based on the type definition of ExportedHandlerFetchHandler.
|
|
* - see: https://github.com/cloudflare/workers-types/blob/c8d9533caa4415c2156d2cf1daca75289d01ae70/index.d.ts#LL564
|
|
*/
|
|
|
|
// MustGetRuntimeContextEnv gets object which holds environment variables bound to Cloudflare worker.
|
|
// - see: https://github.com/cloudflare/workers-types/blob/c8d9533caa4415c2156d2cf1daca75289d01ae70/index.d.ts#L566
|
|
func MustGetRuntimeContextEnv(ctx context.Context) js.Value {
|
|
return MustGetRuntimeContextValue(ctx, "env")
|
|
}
|
|
|
|
// MustGetExecutionContext gets ExecutionContext object from context.
|
|
// - see: https://github.com/cloudflare/workers-types/blob/c8d9533caa4415c2156d2cf1daca75289d01ae70/index.d.ts#L567
|
|
// - see also: https://github.com/cloudflare/workers-types/blob/c8d9533caa4415c2156d2cf1daca75289d01ae70/index.d.ts#L554
|
|
func MustGetExecutionContext(ctx context.Context) js.Value {
|
|
return MustGetRuntimeContextValue(ctx, "ctx")
|
|
}
|
|
|
|
// MustGetRuntimeContextValue gets value for specified key from RuntimeContext.
|
|
// - if the value is undefined, this function panics.
|
|
func MustGetRuntimeContextValue(ctx context.Context, key string) js.Value {
|
|
val, err := GetRuntimeContextValue(ctx, key)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return val
|
|
}
|
|
|
|
var ErrValueNotFound = errors.New("execution context value for specified key not found")
|
|
|
|
// GetRuntimeContextValue gets value for specified key from RuntimeContext.
|
|
// - if the value is undefined, return error.
|
|
func GetRuntimeContextValue(ctx context.Context, key string) (js.Value, error) {
|
|
runtimeObj := runtimecontext.MustExtractRuntimeObj(ctx)
|
|
v := runtimeObj.Get(key)
|
|
if v.IsUndefined() {
|
|
return js.Value{}, ErrValueNotFound
|
|
}
|
|
return v, nil
|
|
}
|