67 lines
1.4 KiB
TypeScript
67 lines
1.4 KiB
TypeScript
import { BBContext } from "@budibase/types"
|
|
|
|
export class ResourceIdGetter {
|
|
parameter: string
|
|
main: string | null
|
|
sub: string | null
|
|
|
|
constructor(ctxProperty: string) {
|
|
this.parameter = ctxProperty
|
|
this.main = null
|
|
this.sub = null
|
|
return this
|
|
}
|
|
|
|
mainResource(field: string) {
|
|
this.main = field
|
|
return this
|
|
}
|
|
|
|
subResource(field: string) {
|
|
this.sub = field
|
|
return this
|
|
}
|
|
|
|
build() {
|
|
const parameter = this.parameter,
|
|
main = this.main,
|
|
sub = this.sub
|
|
return (ctx: BBContext, next: any) => {
|
|
// @ts-ignore
|
|
const request = ctx.request[parameter] || ctx[parameter]
|
|
if (request == null) {
|
|
return next()
|
|
}
|
|
if (main != null && request[main]) {
|
|
ctx.resourceId = request[main]
|
|
}
|
|
if (sub != null && request[sub]) {
|
|
ctx.subResourceId = request[sub]
|
|
}
|
|
return next()
|
|
}
|
|
}
|
|
}
|
|
|
|
export function paramResource(main: string) {
|
|
return new ResourceIdGetter("params").mainResource(main).build()
|
|
}
|
|
|
|
export function paramSubResource(main: string, sub: string) {
|
|
return new ResourceIdGetter("params")
|
|
.mainResource(main)
|
|
.subResource(sub)
|
|
.build()
|
|
}
|
|
|
|
export function bodyResource(main: string) {
|
|
return new ResourceIdGetter("body").mainResource(main).build()
|
|
}
|
|
|
|
export function bodySubResource(main: string, sub: string) {
|
|
return new ResourceIdGetter("body")
|
|
.mainResource(main)
|
|
.subResource(sub)
|
|
.build()
|
|
}
|