@databricks/sdk-tokens 0.0.0-dev → 0.1.0-dev.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +203 -0
- package/dist/v1/client.d.ts +32 -0
- package/dist/v1/client.d.ts.map +1 -0
- package/dist/v1/client.js +152 -0
- package/dist/v1/client.js.map +1 -0
- package/dist/v1/index.d.ts +5 -0
- package/dist/v1/index.d.ts.map +1 -0
- package/dist/v1/index.js +5 -0
- package/dist/v1/index.js.map +1 -0
- package/dist/v1/model.d.ts +94 -0
- package/dist/v1/model.d.ts.map +1 -0
- package/dist/v1/model.js +138 -0
- package/dist/v1/model.js.map +1 -0
- package/dist/v1/transport.d.ts +5 -0
- package/dist/v1/transport.d.ts.map +1 -0
- package/dist/v1/transport.js +57 -0
- package/dist/v1/transport.js.map +1 -0
- package/dist/v1/utils.d.ts +22 -0
- package/dist/v1/utils.d.ts.map +1 -0
- package/dist/v1/utils.js +113 -0
- package/dist/v1/utils.js.map +1 -0
- package/package.json +38 -4
- package/src/v1/client.ts +206 -0
- package/src/v1/index.ts +19 -0
- package/src/v1/model.ts +236 -0
- package/src/v1/transport.ts +73 -0
- package/src/v1/utils.ts +156 -0
- package/README.md +0 -1
- package/index.js +0 -1
package/dist/v1/model.js
ADDED
|
@@ -0,0 +1,138 @@
|
|
|
1
|
+
// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
|
|
2
|
+
import { FieldMask } from '@databricks/sdk-core/wkt';
|
|
3
|
+
import { z } from 'zod';
|
|
4
|
+
/**
|
|
5
|
+
* State of inferred scope collection (autoscope) for an external PAT.
|
|
6
|
+
* Mirrored in databricks.identity.AutoscopeState in common/principal-context/api/proto/tokendetails.proto.
|
|
7
|
+
* Token store and token management proto can depend on this.
|
|
8
|
+
* Principal context proto should NOT depend on this proto definitions because too many services depend on the principal context proto.
|
|
9
|
+
*/
|
|
10
|
+
export var AutoscopeState;
|
|
11
|
+
(function (AutoscopeState) {
|
|
12
|
+
AutoscopeState["AUTOSCOPE_STATE_UNSPECIFIED"] = "AUTOSCOPE_STATE_UNSPECIFIED";
|
|
13
|
+
AutoscopeState["AUTOSCOPE_STATE_DISABLED"] = "AUTOSCOPE_STATE_DISABLED";
|
|
14
|
+
AutoscopeState["AUTOSCOPE_STATE_RUNNING"] = "AUTOSCOPE_STATE_RUNNING";
|
|
15
|
+
AutoscopeState["AUTOSCOPE_STATE_COMPLETED"] = "AUTOSCOPE_STATE_COMPLETED";
|
|
16
|
+
AutoscopeState["AUTOSCOPE_STATE_BACKFILLED"] = "AUTOSCOPE_STATE_BACKFILLED";
|
|
17
|
+
AutoscopeState["AUTOSCOPE_STATE_USER_SELECTED"] = "AUTOSCOPE_STATE_USER_SELECTED";
|
|
18
|
+
AutoscopeState["AUTOSCOPE_STATE_API_NOT_COVERED"] = "AUTOSCOPE_STATE_API_NOT_COVERED";
|
|
19
|
+
})(AutoscopeState || (AutoscopeState = {}));
|
|
20
|
+
// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
|
|
21
|
+
export const unmarshalCreateTokenRequest_ResponseSchema = z
|
|
22
|
+
.object({
|
|
23
|
+
token_value: z.string().optional(),
|
|
24
|
+
token_info: z.lazy(() => unmarshalPublicTokenInfoSchema).optional(),
|
|
25
|
+
})
|
|
26
|
+
.transform(d => ({
|
|
27
|
+
tokenValue: d.token_value,
|
|
28
|
+
tokenInfo: d.token_info,
|
|
29
|
+
}));
|
|
30
|
+
// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
|
|
31
|
+
export const unmarshalListTokensRequest_ResponseSchema = z
|
|
32
|
+
.object({
|
|
33
|
+
token_infos: z
|
|
34
|
+
.array(z.lazy(() => unmarshalPublicTokenInfoSchema))
|
|
35
|
+
.optional(),
|
|
36
|
+
})
|
|
37
|
+
.transform(d => ({
|
|
38
|
+
tokenInfos: d.token_infos,
|
|
39
|
+
}));
|
|
40
|
+
export const unmarshalPublicTokenInfoSchema = z
|
|
41
|
+
.object({
|
|
42
|
+
token_id: z.string().optional(),
|
|
43
|
+
creation_time: z
|
|
44
|
+
.union([z.number(), z.bigint()])
|
|
45
|
+
.transform(v => BigInt(v))
|
|
46
|
+
.optional(),
|
|
47
|
+
expiry_time: z
|
|
48
|
+
.union([z.number(), z.bigint()])
|
|
49
|
+
.transform(v => BigInt(v))
|
|
50
|
+
.optional(),
|
|
51
|
+
comment: z.string().optional(),
|
|
52
|
+
scopes: z.array(z.string()).optional(),
|
|
53
|
+
autoscope_state: z.enum(AutoscopeState).optional(),
|
|
54
|
+
inferred_scopes: z.array(z.string()).optional(),
|
|
55
|
+
backfill_scopes: z.array(z.string()).optional(),
|
|
56
|
+
})
|
|
57
|
+
.transform(d => ({
|
|
58
|
+
tokenId: d.token_id,
|
|
59
|
+
creationTime: d.creation_time,
|
|
60
|
+
expiryTime: d.expiry_time,
|
|
61
|
+
comment: d.comment,
|
|
62
|
+
scopes: d.scopes,
|
|
63
|
+
autoscopeState: d.autoscope_state,
|
|
64
|
+
inferredScopes: d.inferred_scopes,
|
|
65
|
+
backfillScopes: d.backfill_scopes,
|
|
66
|
+
}));
|
|
67
|
+
// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
|
|
68
|
+
export const unmarshalRevokeTokenRequest_ResponseSchema = z.object({});
|
|
69
|
+
export const unmarshalUpdateTokenResponseSchema = z.object({});
|
|
70
|
+
export const marshalCreateTokenRequestSchema = z
|
|
71
|
+
.object({
|
|
72
|
+
lifetimeSeconds: z.bigint().optional(),
|
|
73
|
+
comment: z.string().optional(),
|
|
74
|
+
scopes: z.array(z.string()).optional(),
|
|
75
|
+
autoscopeEnabled: z.boolean().optional(),
|
|
76
|
+
})
|
|
77
|
+
.transform(d => ({
|
|
78
|
+
lifetime_seconds: d.lifetimeSeconds,
|
|
79
|
+
comment: d.comment,
|
|
80
|
+
scopes: d.scopes,
|
|
81
|
+
autoscope_enabled: d.autoscopeEnabled,
|
|
82
|
+
}));
|
|
83
|
+
export const marshalPublicTokenInfoSchema = z
|
|
84
|
+
.object({
|
|
85
|
+
tokenId: z.string().optional(),
|
|
86
|
+
creationTime: z.bigint().optional(),
|
|
87
|
+
expiryTime: z.bigint().optional(),
|
|
88
|
+
comment: z.string().optional(),
|
|
89
|
+
scopes: z.array(z.string()).optional(),
|
|
90
|
+
autoscopeState: z.enum(AutoscopeState).optional(),
|
|
91
|
+
inferredScopes: z.array(z.string()).optional(),
|
|
92
|
+
backfillScopes: z.array(z.string()).optional(),
|
|
93
|
+
})
|
|
94
|
+
.transform(d => ({
|
|
95
|
+
token_id: d.tokenId,
|
|
96
|
+
creation_time: d.creationTime,
|
|
97
|
+
expiry_time: d.expiryTime,
|
|
98
|
+
comment: d.comment,
|
|
99
|
+
scopes: d.scopes,
|
|
100
|
+
autoscope_state: d.autoscopeState,
|
|
101
|
+
inferred_scopes: d.inferredScopes,
|
|
102
|
+
backfill_scopes: d.backfillScopes,
|
|
103
|
+
}));
|
|
104
|
+
export const marshalRevokeTokenRequestSchema = z
|
|
105
|
+
.object({
|
|
106
|
+
tokenId: z.string().optional(),
|
|
107
|
+
})
|
|
108
|
+
.transform(d => ({
|
|
109
|
+
token_id: d.tokenId,
|
|
110
|
+
}));
|
|
111
|
+
export const marshalUpdateTokenRequestSchema = z
|
|
112
|
+
.object({
|
|
113
|
+
tokenId: z.string().optional(),
|
|
114
|
+
token: z.lazy(() => marshalPublicTokenInfoSchema).optional(),
|
|
115
|
+
updateMask: z
|
|
116
|
+
.any()
|
|
117
|
+
.transform((m) => m.toString())
|
|
118
|
+
.optional(),
|
|
119
|
+
})
|
|
120
|
+
.transform(d => ({
|
|
121
|
+
token_id: d.tokenId,
|
|
122
|
+
token: d.token,
|
|
123
|
+
update_mask: d.updateMask,
|
|
124
|
+
}));
|
|
125
|
+
const publicTokenInfoFieldMaskSchema = {
|
|
126
|
+
autoscopeState: { wire: 'autoscope_state' },
|
|
127
|
+
backfillScopes: { wire: 'backfill_scopes' },
|
|
128
|
+
comment: { wire: 'comment' },
|
|
129
|
+
creationTime: { wire: 'creation_time' },
|
|
130
|
+
expiryTime: { wire: 'expiry_time' },
|
|
131
|
+
inferredScopes: { wire: 'inferred_scopes' },
|
|
132
|
+
scopes: { wire: 'scopes' },
|
|
133
|
+
tokenId: { wire: 'token_id' },
|
|
134
|
+
};
|
|
135
|
+
export function publicTokenInfoFieldMask(...paths) {
|
|
136
|
+
return FieldMask.build(paths, publicTokenInfoFieldMaskSchema);
|
|
137
|
+
}
|
|
138
|
+
//# sourceMappingURL=model.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"model.js","sourceRoot":"","sources":["../../src/v1/model.ts"],"names":[],"mappings":"AAAA,+EAA+E;AAE/E,OAAO,EAAC,SAAS,EAAC,MAAM,0BAA0B,CAAC;AAEnD,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AAEtB;;;;;GAKG;AACH,MAAM,CAAN,IAAY,cAQX;AARD,WAAY,cAAc;IACxB,6EAA2D,CAAA;IAC3D,uEAAqD,CAAA;IACrD,qEAAmD,CAAA;IACnD,yEAAuD,CAAA;IACvD,2EAAyD,CAAA;IACzD,iFAA+D,CAAA;IAC/D,qFAAmE,CAAA;AACrE,CAAC,EARW,cAAc,KAAd,cAAc,QAQzB;AA+ED,oGAAoG;AACpG,MAAM,CAAC,MAAM,0CAA0C,GACrD,CAAC;KACE,MAAM,CAAC;IACN,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAClC,UAAU,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,8BAA8B,CAAC,CAAC,QAAQ,EAAE;CACpE,CAAC;KACD,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACf,UAAU,EAAE,CAAC,CAAC,WAAW;IACzB,SAAS,EAAE,CAAC,CAAC,UAAU;CACxB,CAAC,CAAC,CAAC;AAER,oGAAoG;AACpG,MAAM,CAAC,MAAM,yCAAyC,GACpD,CAAC;KACE,MAAM,CAAC;IACN,WAAW,EAAE,CAAC;SACX,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,8BAA8B,CAAC,CAAC;SACnD,QAAQ,EAAE;CACd,CAAC;KACD,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACf,UAAU,EAAE,CAAC,CAAC,WAAW;CAC1B,CAAC,CAAC,CAAC;AAER,MAAM,CAAC,MAAM,8BAA8B,GAA+B,CAAC;KACxE,MAAM,CAAC;IACN,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC/B,aAAa,EAAE,CAAC;SACb,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;SAC/B,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;SACzB,QAAQ,EAAE;IACb,WAAW,EAAE,CAAC;SACX,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;SAC/B,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;SACzB,QAAQ,EAAE;IACb,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC9B,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;IACtC,eAAe,EAAE,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,QAAQ,EAAE;IAClD,eAAe,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;IAC/C,eAAe,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;CAChD,CAAC;KACD,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACf,OAAO,EAAE,CAAC,CAAC,QAAQ;IACnB,YAAY,EAAE,CAAC,CAAC,aAAa;IAC7B,UAAU,EAAE,CAAC,CAAC,WAAW;IACzB,OAAO,EAAE,CAAC,CAAC,OAAO;IAClB,MAAM,EAAE,CAAC,CAAC,MAAM;IAChB,cAAc,EAAE,CAAC,CAAC,eAAe;IACjC,cAAc,EAAE,CAAC,CAAC,eAAe;IACjC,cAAc,EAAE,CAAC,CAAC,eAAe;CAClC,CAAC,CAAC,CAAC;AAEN,oGAAoG;AACpG,MAAM,CAAC,MAAM,0CAA0C,GACrD,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;AAEf,MAAM,CAAC,MAAM,kCAAkC,GAC7C,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;AAEf,MAAM,CAAC,MAAM,+BAA+B,GAAc,CAAC;KACxD,MAAM,CAAC;IACN,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACtC,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC9B,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;IACtC,gBAAgB,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;CACzC,CAAC;KACD,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACf,gBAAgB,EAAE,CAAC,CAAC,eAAe;IACnC,OAAO,EAAE,CAAC,CAAC,OAAO;IAClB,MAAM,EAAE,CAAC,CAAC,MAAM;IAChB,iBAAiB,EAAE,CAAC,CAAC,gBAAgB;CACtC,CAAC,CAAC,CAAC;AAEN,MAAM,CAAC,MAAM,4BAA4B,GAAc,CAAC;KACrD,MAAM,CAAC;IACN,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC9B,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACnC,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACjC,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC9B,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;IACtC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,QAAQ,EAAE;IACjD,cAAc,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;IAC9C,cAAc,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;CAC/C,CAAC;KACD,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACf,QAAQ,EAAE,CAAC,CAAC,OAAO;IACnB,aAAa,EAAE,CAAC,CAAC,YAAY;IAC7B,WAAW,EAAE,CAAC,CAAC,UAAU;IACzB,OAAO,EAAE,CAAC,CAAC,OAAO;IAClB,MAAM,EAAE,CAAC,CAAC,MAAM;IAChB,eAAe,EAAE,CAAC,CAAC,cAAc;IACjC,eAAe,EAAE,CAAC,CAAC,cAAc;IACjC,eAAe,EAAE,CAAC,CAAC,cAAc;CAClC,CAAC,CAAC,CAAC;AAEN,MAAM,CAAC,MAAM,+BAA+B,GAAc,CAAC;KACxD,MAAM,CAAC;IACN,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CAC/B,CAAC;KACD,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACf,QAAQ,EAAE,CAAC,CAAC,OAAO;CACpB,CAAC,CAAC,CAAC;AAEN,MAAM,CAAC,MAAM,+BAA+B,GAAc,CAAC;KACxD,MAAM,CAAC;IACN,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC9B,KAAK,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,4BAA4B,CAAC,CAAC,QAAQ,EAAE;IAC5D,UAAU,EAAE,CAAC;SACV,GAAG,EAAE;SACL,SAAS,CAAC,CAAC,CAAY,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;SACzC,QAAQ,EAAE;CACd,CAAC;KACD,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACf,QAAQ,EAAE,CAAC,CAAC,OAAO;IACnB,KAAK,EAAE,CAAC,CAAC,KAAK;IACd,WAAW,EAAE,CAAC,CAAC,UAAU;CAC1B,CAAC,CAAC,CAAC;AAEN,MAAM,8BAA8B,GAAoB;IACtD,cAAc,EAAE,EAAC,IAAI,EAAE,iBAAiB,EAAC;IACzC,cAAc,EAAE,EAAC,IAAI,EAAE,iBAAiB,EAAC;IACzC,OAAO,EAAE,EAAC,IAAI,EAAE,SAAS,EAAC;IAC1B,YAAY,EAAE,EAAC,IAAI,EAAE,eAAe,EAAC;IACrC,UAAU,EAAE,EAAC,IAAI,EAAE,aAAa,EAAC;IACjC,cAAc,EAAE,EAAC,IAAI,EAAE,iBAAiB,EAAC;IACzC,MAAM,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAC;IACxB,OAAO,EAAE,EAAC,IAAI,EAAE,UAAU,EAAC;CAC5B,CAAC;AAEF,MAAM,UAAU,wBAAwB,CACtC,GAAG,KAAe;IAElB,OAAO,SAAS,CAAC,KAAK,CACpB,KAAK,EACL,8BAA8B,CAC/B,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,5 @@
|
|
|
1
|
+
import type { HttpClient } from '@databricks/sdk-core/http';
|
|
2
|
+
import type { ClientOptions } from '@databricks/sdk-options/client';
|
|
3
|
+
/** Creates a new HTTP client with the given options. */
|
|
4
|
+
export declare function newHttpClient(options?: ClientOptions): HttpClient;
|
|
5
|
+
//# sourceMappingURL=transport.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"transport.d.ts","sourceRoot":"","sources":["../../src/v1/transport.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EACV,UAAU,EAGX,MAAM,2BAA2B,CAAC;AAEnC,OAAO,KAAK,EAAC,aAAa,EAAC,MAAM,gCAAgC,CAAC;AAElE,wDAAwD;AACxD,wBAAgB,aAAa,CAAC,OAAO,CAAC,EAAE,aAAa,GAAG,UAAU,CAwBjE"}
|
|
@@ -0,0 +1,57 @@
|
|
|
1
|
+
// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
|
|
2
|
+
import { defaultCredentials } from '@databricks/sdk-auth/credentials';
|
|
3
|
+
import { newFetchHttpClient } from '@databricks/sdk-core/http';
|
|
4
|
+
/** Creates a new HTTP client with the given options. */
|
|
5
|
+
export function newHttpClient(options) {
|
|
6
|
+
const opts = options ?? {};
|
|
7
|
+
// If an HTTP client is provided, use it as-is. Throw if other options are
|
|
8
|
+
// also set, since they would be silently ignored.
|
|
9
|
+
if (opts.httpClient !== undefined) {
|
|
10
|
+
if (opts.credentials !== undefined || opts.timeout !== undefined) {
|
|
11
|
+
throw new Error('httpClient cannot be combined with credentials or timeout');
|
|
12
|
+
}
|
|
13
|
+
return opts.httpClient;
|
|
14
|
+
}
|
|
15
|
+
const credentials = opts.credentials ?? defaultCredentials();
|
|
16
|
+
const base = newFetchHttpClient();
|
|
17
|
+
let client = new AuthHttpClient(base, credentials);
|
|
18
|
+
if (opts.timeout !== undefined) {
|
|
19
|
+
client = new TimeoutHttpClient(client, opts.timeout);
|
|
20
|
+
}
|
|
21
|
+
return client;
|
|
22
|
+
}
|
|
23
|
+
/** Wraps an HttpClient and adds authentication headers to requests. */
|
|
24
|
+
class AuthHttpClient {
|
|
25
|
+
base;
|
|
26
|
+
credentials;
|
|
27
|
+
constructor(base, credentials) {
|
|
28
|
+
this.base = base;
|
|
29
|
+
this.credentials = credentials;
|
|
30
|
+
}
|
|
31
|
+
async send(request) {
|
|
32
|
+
const authHeaders = await this.credentials.authHeaders();
|
|
33
|
+
// Do not modify the original request.
|
|
34
|
+
const headers = new Headers(request.headers);
|
|
35
|
+
for (const h of authHeaders) {
|
|
36
|
+
headers.set(h.key, h.value);
|
|
37
|
+
}
|
|
38
|
+
return this.base.send({ ...request, headers });
|
|
39
|
+
}
|
|
40
|
+
}
|
|
41
|
+
/** Wraps an HttpClient and applies a default timeout to requests. */
|
|
42
|
+
class TimeoutHttpClient {
|
|
43
|
+
base;
|
|
44
|
+
timeout;
|
|
45
|
+
constructor(base, timeout) {
|
|
46
|
+
this.base = base;
|
|
47
|
+
this.timeout = timeout;
|
|
48
|
+
}
|
|
49
|
+
async send(request) {
|
|
50
|
+
const timeoutSignal = AbortSignal.timeout(this.timeout);
|
|
51
|
+
const signal = request.signal !== undefined
|
|
52
|
+
? AbortSignal.any([request.signal, timeoutSignal])
|
|
53
|
+
: timeoutSignal;
|
|
54
|
+
return this.base.send({ ...request, signal });
|
|
55
|
+
}
|
|
56
|
+
}
|
|
57
|
+
//# sourceMappingURL=transport.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"transport.js","sourceRoot":"","sources":["../../src/v1/transport.ts"],"names":[],"mappings":"AAAA,+EAA+E;AAG/E,OAAO,EAAC,kBAAkB,EAAC,MAAM,kCAAkC,CAAC;AAMpE,OAAO,EAAC,kBAAkB,EAAC,MAAM,2BAA2B,CAAC;AAG7D,wDAAwD;AACxD,MAAM,UAAU,aAAa,CAAC,OAAuB;IACnD,MAAM,IAAI,GAAG,OAAO,IAAI,EAAE,CAAC;IAE3B,0EAA0E;IAC1E,kDAAkD;IAClD,IAAI,IAAI,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;QAClC,IAAI,IAAI,CAAC,WAAW,KAAK,SAAS,IAAI,IAAI,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;YACjE,MAAM,IAAI,KAAK,CACb,2DAA2D,CAC5D,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,IAAI,kBAAkB,EAAE,CAAC;IAE7D,MAAM,IAAI,GAAG,kBAAkB,EAAE,CAAC;IAClC,IAAI,MAAM,GAAe,IAAI,cAAc,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;IAE/D,IAAI,IAAI,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;QAC/B,MAAM,GAAG,IAAI,iBAAiB,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;IACvD,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,uEAAuE;AACvE,MAAM,cAAc;IAEC;IACA;IAFnB,YACmB,IAAgB,EAChB,WAAwB;QADxB,SAAI,GAAJ,IAAI,CAAY;QAChB,gBAAW,GAAX,WAAW,CAAa;IACxC,CAAC;IAEJ,KAAK,CAAC,IAAI,CAAC,OAAoB;QAC7B,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,CAAC;QACzD,sCAAsC;QACtC,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC7C,KAAK,MAAM,CAAC,IAAI,WAAW,EAAE,CAAC;YAC5B,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC;QAC9B,CAAC;QACD,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAC,GAAG,OAAO,EAAE,OAAO,EAAC,CAAC,CAAC;IAC/C,CAAC;CACF;AAED,qEAAqE;AACrE,MAAM,iBAAiB;IAEF;IACA;IAFnB,YACmB,IAAgB,EAChB,OAAe;QADf,SAAI,GAAJ,IAAI,CAAY;QAChB,YAAO,GAAP,OAAO,CAAQ;IAC/B,CAAC;IAEJ,KAAK,CAAC,IAAI,CAAC,OAAoB;QAC7B,MAAM,aAAa,GAAG,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACxD,MAAM,MAAM,GACV,OAAO,CAAC,MAAM,KAAK,SAAS;YAC1B,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;YAClD,CAAC,CAAC,aAAa,CAAC;QACpB,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAC,GAAG,OAAO,EAAE,MAAM,EAAC,CAAC,CAAC;IAC9C,CAAC;CACF"}
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
import type { Call } from '@databricks/sdk-core/api';
|
|
2
|
+
import type { HttpClient, HttpRequest } from '@databricks/sdk-core/http';
|
|
3
|
+
import type { Logger } from '@databricks/sdk-core/logger';
|
|
4
|
+
import type { CallOptions } from '@databricks/sdk-options/call';
|
|
5
|
+
import type { z } from 'zod';
|
|
6
|
+
export interface HttpCallOptions {
|
|
7
|
+
readonly request: HttpRequest;
|
|
8
|
+
readonly httpClient: HttpClient;
|
|
9
|
+
readonly logger: Logger;
|
|
10
|
+
}
|
|
11
|
+
/**
|
|
12
|
+
* Translates public CallOptions to the internal Options shape accepted by
|
|
13
|
+
* execute(). Even though the shapes match today, this isolates the public
|
|
14
|
+
* API from the executor's internal type so they can diverge.
|
|
15
|
+
*/
|
|
16
|
+
export declare function executeCall(call: Call, options?: CallOptions): Promise<void>;
|
|
17
|
+
export declare function executeHttpCall(opts: HttpCallOptions): Promise<Uint8Array>;
|
|
18
|
+
export declare function buildHttpRequest(method: string, url: string, headers: Headers, signal?: AbortSignal, body?: string | ReadableStream<Uint8Array>): HttpRequest;
|
|
19
|
+
export declare function parseResponse<T>(body: Uint8Array, schema: z.ZodType<T>): T;
|
|
20
|
+
export declare function marshalRequest(data: unknown, schema: z.ZodType): string;
|
|
21
|
+
export declare function flattenQueryParams(prefix: string, value: unknown, params: URLSearchParams): void;
|
|
22
|
+
//# sourceMappingURL=utils.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/v1/utils.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAC,IAAI,EAAU,MAAM,0BAA0B,CAAC;AAG5D,OAAO,KAAK,EACV,UAAU,EACV,WAAW,EAEZ,MAAM,2BAA2B,CAAC;AACnC,OAAO,KAAK,EAAC,MAAM,EAAC,MAAM,6BAA6B,CAAC;AACxD,OAAO,KAAK,EAAC,WAAW,EAAC,MAAM,8BAA8B,CAAC;AAE9D,OAAO,KAAK,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AAO3B,MAAM,WAAW,eAAe;IAC9B,QAAQ,CAAC,OAAO,EAAE,WAAW,CAAC;IAC9B,QAAQ,CAAC,UAAU,EAAE,UAAU,CAAC;IAChC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;CACzB;AAED;;;;GAIG;AACH,wBAAsB,WAAW,CAC/B,IAAI,EAAE,IAAI,EACV,OAAO,CAAC,EAAE,WAAW,GACpB,OAAO,CAAC,IAAI,CAAC,CASf;AA2BD,wBAAsB,eAAe,CACnC,IAAI,EAAE,eAAe,GACpB,OAAO,CAAC,UAAU,CAAC,CA2BrB;AAED,wBAAgB,gBAAgB,CAC9B,MAAM,EAAE,MAAM,EACd,GAAG,EAAE,MAAM,EACX,OAAO,EAAE,OAAO,EAChB,MAAM,CAAC,EAAE,WAAW,EACpB,IAAI,CAAC,EAAE,MAAM,GAAG,cAAc,CAAC,UAAU,CAAC,GACzC,WAAW,CASb;AAED,wBAAgB,aAAa,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAI1E;AAED,wBAAgB,cAAc,CAAC,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,OAAO,GAAG,MAAM,CAEvE;AAED,wBAAgB,kBAAkB,CAChC,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,OAAO,EACd,MAAM,EAAE,eAAe,GACtB,IAAI,CAuBN"}
|
package/dist/v1/utils.js
ADDED
|
@@ -0,0 +1,113 @@
|
|
|
1
|
+
// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
|
|
2
|
+
import { execute } from '@databricks/sdk-core/api';
|
|
3
|
+
import { ApiError } from '@databricks/sdk-core/apierror';
|
|
4
|
+
import JSONBig from 'json-bigint';
|
|
5
|
+
// JSON codec that preserves int64 precision. On the way in, large integer
|
|
6
|
+
// literals come back as bigint instead of being rounded to JS Number. On the
|
|
7
|
+
// way out, bigint values are emitted as raw JSON number digits.
|
|
8
|
+
const jsonBigint = JSONBig({ useNativeBigInt: true });
|
|
9
|
+
/**
|
|
10
|
+
* Translates public CallOptions to the internal Options shape accepted by
|
|
11
|
+
* execute(). Even though the shapes match today, this isolates the public
|
|
12
|
+
* API from the executor's internal type so they can diverge.
|
|
13
|
+
*/
|
|
14
|
+
export async function executeCall(call, options) {
|
|
15
|
+
const opts = {
|
|
16
|
+
...(options?.retrier !== undefined && { retrier: options.retrier }),
|
|
17
|
+
...(options?.rateLimiter !== undefined && {
|
|
18
|
+
rateLimiter: options.rateLimiter,
|
|
19
|
+
}),
|
|
20
|
+
...(options?.timeout !== undefined && { timeout: options.timeout }),
|
|
21
|
+
};
|
|
22
|
+
return execute(options?.signal, call, opts);
|
|
23
|
+
}
|
|
24
|
+
async function readAll(body) {
|
|
25
|
+
if (body === null) {
|
|
26
|
+
return new Uint8Array(0);
|
|
27
|
+
}
|
|
28
|
+
const reader = body.getReader();
|
|
29
|
+
const chunks = [];
|
|
30
|
+
for (;;) {
|
|
31
|
+
const { done, value } = await reader.read();
|
|
32
|
+
if (done) {
|
|
33
|
+
break;
|
|
34
|
+
}
|
|
35
|
+
chunks.push(value);
|
|
36
|
+
}
|
|
37
|
+
const totalLength = chunks.reduce((acc, chunk) => acc + chunk.length, 0);
|
|
38
|
+
const result = new Uint8Array(totalLength);
|
|
39
|
+
let offset = 0;
|
|
40
|
+
for (const chunk of chunks) {
|
|
41
|
+
result.set(chunk, offset);
|
|
42
|
+
offset += chunk.length;
|
|
43
|
+
}
|
|
44
|
+
return result;
|
|
45
|
+
}
|
|
46
|
+
export async function executeHttpCall(opts) {
|
|
47
|
+
opts.logger.debug('HTTP request', {
|
|
48
|
+
method: opts.request.method,
|
|
49
|
+
url: opts.request.url,
|
|
50
|
+
});
|
|
51
|
+
let resp;
|
|
52
|
+
try {
|
|
53
|
+
resp = await opts.httpClient.send(opts.request);
|
|
54
|
+
}
|
|
55
|
+
catch (e) {
|
|
56
|
+
opts.logger.debug('HTTP request failed');
|
|
57
|
+
throw e;
|
|
58
|
+
}
|
|
59
|
+
const body = await readAll(resp.body);
|
|
60
|
+
opts.logger.debug('HTTP response', {
|
|
61
|
+
statusCode: resp.statusCode,
|
|
62
|
+
body: new TextDecoder().decode(body),
|
|
63
|
+
});
|
|
64
|
+
const apiErr = ApiError.fromHttpError(resp.statusCode, resp.headers, body);
|
|
65
|
+
if (apiErr !== undefined) {
|
|
66
|
+
throw apiErr;
|
|
67
|
+
}
|
|
68
|
+
return body;
|
|
69
|
+
}
|
|
70
|
+
export function buildHttpRequest(method, url, headers, signal, body) {
|
|
71
|
+
const req = { url, method, headers };
|
|
72
|
+
if (body !== undefined) {
|
|
73
|
+
req.body = body;
|
|
74
|
+
}
|
|
75
|
+
if (signal !== undefined) {
|
|
76
|
+
req.signal = signal;
|
|
77
|
+
}
|
|
78
|
+
return req;
|
|
79
|
+
}
|
|
80
|
+
export function parseResponse(body, schema) {
|
|
81
|
+
const text = new TextDecoder().decode(body);
|
|
82
|
+
const parsed = jsonBigint.parse(text);
|
|
83
|
+
return schema.parse(parsed);
|
|
84
|
+
}
|
|
85
|
+
export function marshalRequest(data, schema) {
|
|
86
|
+
return jsonBigint.stringify(schema.parse(data));
|
|
87
|
+
}
|
|
88
|
+
export function flattenQueryParams(prefix, value, params) {
|
|
89
|
+
if (value === null || value === undefined) {
|
|
90
|
+
return;
|
|
91
|
+
}
|
|
92
|
+
if (Array.isArray(value)) {
|
|
93
|
+
// arrays of objects are not yet supported
|
|
94
|
+
for (const item of value) {
|
|
95
|
+
params.append(prefix, String(item));
|
|
96
|
+
}
|
|
97
|
+
}
|
|
98
|
+
else if (typeof value === 'object') {
|
|
99
|
+
for (const [key, val] of Object.entries(value)) {
|
|
100
|
+
flattenQueryParams(`${prefix}.${key}`, val, params);
|
|
101
|
+
}
|
|
102
|
+
}
|
|
103
|
+
else if (typeof value === 'string' ||
|
|
104
|
+
typeof value === 'number' ||
|
|
105
|
+
typeof value === 'boolean' ||
|
|
106
|
+
typeof value === 'bigint') {
|
|
107
|
+
params.append(prefix, String(value));
|
|
108
|
+
}
|
|
109
|
+
else {
|
|
110
|
+
throw new Error(`Unsupported query parameter type: ${typeof value}`);
|
|
111
|
+
}
|
|
112
|
+
}
|
|
113
|
+
//# sourceMappingURL=utils.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/v1/utils.ts"],"names":[],"mappings":"AAAA,+EAA+E;AAG/E,OAAO,EAAC,OAAO,EAAC,MAAM,0BAA0B,CAAC;AACjD,OAAO,EAAC,QAAQ,EAAC,MAAM,+BAA+B,CAAC;AAQvD,OAAO,OAAO,MAAM,aAAa,CAAC;AAGlC,0EAA0E;AAC1E,6EAA6E;AAC7E,gEAAgE;AAChE,MAAM,UAAU,GAAG,OAAO,CAAC,EAAC,eAAe,EAAE,IAAI,EAAC,CAAC,CAAC;AAQpD;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,IAAU,EACV,OAAqB;IAErB,MAAM,IAAI,GAAY;QACpB,GAAG,CAAC,OAAO,EAAE,OAAO,KAAK,SAAS,IAAI,EAAC,OAAO,EAAE,OAAO,CAAC,OAAO,EAAC,CAAC;QACjE,GAAG,CAAC,OAAO,EAAE,WAAW,KAAK,SAAS,IAAI;YACxC,WAAW,EAAE,OAAO,CAAC,WAAW;SACjC,CAAC;QACF,GAAG,CAAC,OAAO,EAAE,OAAO,KAAK,SAAS,IAAI,EAAC,OAAO,EAAE,OAAO,CAAC,OAAO,EAAC,CAAC;KAClE,CAAC;IACF,OAAO,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;AAC9C,CAAC;AAED,KAAK,UAAU,OAAO,CACpB,IAAuC;IAEvC,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;QAClB,OAAO,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC;IAC3B,CAAC;IACD,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;IAChC,MAAM,MAAM,GAAiB,EAAE,CAAC;IAChC,SAAS,CAAC;QACR,MAAM,EAAC,IAAI,EAAE,KAAK,EAAC,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QAC1C,IAAI,IAAI,EAAE,CAAC;YACT,MAAM;QACR,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACrB,CAAC;IACD,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IACzE,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,WAAW,CAAC,CAAC;IAC3C,IAAI,MAAM,GAAG,CAAC,CAAC;IACf,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAC1B,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC;IACzB,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,IAAqB;IAErB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,cAAc,EAAE;QAChC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;QAC3B,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG;KACtB,CAAC,CAAC;IAEH,IAAI,IAAkB,CAAC;IACvB,IAAI,CAAC;QACH,IAAI,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAClD,CAAC;IAAC,OAAO,CAAU,EAAE,CAAC;QACpB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;QACzC,MAAM,CAAC,CAAC;IACV,CAAC;IAED,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAEtC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,eAAe,EAAE;QACjC,UAAU,EAAE,IAAI,CAAC,UAAU;QAC3B,IAAI,EAAE,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC;KACrC,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG,QAAQ,CAAC,aAAa,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IAC3E,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACzB,MAAM,MAAM,CAAC;IACf,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,gBAAgB,CAC9B,MAAc,EACd,GAAW,EACX,OAAgB,EAChB,MAAoB,EACpB,IAA0C;IAE1C,MAAM,GAAG,GAAgB,EAAC,GAAG,EAAE,MAAM,EAAE,OAAO,EAAC,CAAC;IAChD,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QACvB,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC;IAClB,CAAC;IACD,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACzB,GAAG,CAAC,MAAM,GAAG,MAAM,CAAC;IACtB,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,MAAM,UAAU,aAAa,CAAI,IAAgB,EAAE,MAAoB;IACrE,MAAM,IAAI,GAAG,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC5C,MAAM,MAAM,GAAY,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC/C,OAAO,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,IAAa,EAAE,MAAiB;IAC7D,OAAO,UAAU,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;AAClD,CAAC;AAED,MAAM,UAAU,kBAAkB,CAChC,MAAc,EACd,KAAc,EACd,MAAuB;IAEvB,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QAC1C,OAAO;IACT,CAAC;IACD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,0CAA0C;QAC1C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;SAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACrC,KAAK,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAgC,CAAC,EAAE,CAAC;YAC1E,kBAAkB,CAAC,GAAG,MAAM,IAAI,GAAG,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;SAAM,IACL,OAAO,KAAK,KAAK,QAAQ;QACzB,OAAO,KAAK,KAAK,QAAQ;QACzB,OAAO,KAAK,KAAK,SAAS;QAC1B,OAAO,KAAK,KAAK,QAAQ,EACzB,CAAC;QACD,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IACvC,CAAC;SAAM,CAAC;QACN,MAAM,IAAI,KAAK,CAAC,qCAAqC,OAAO,KAAK,EAAE,CAAC,CAAC;IACvE,CAAC;AACH,CAAC"}
|
package/package.json
CHANGED
|
@@ -1,7 +1,41 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@databricks/sdk-tokens",
|
|
3
|
-
"version": "0.
|
|
4
|
-
"description": "
|
|
5
|
-
"
|
|
6
|
-
"
|
|
3
|
+
"version": "0.1.0-dev.1",
|
|
4
|
+
"description": "",
|
|
5
|
+
"type": "module",
|
|
6
|
+
"exports": {
|
|
7
|
+
"./v1": {
|
|
8
|
+
"types": "./dist/v1/index.d.ts",
|
|
9
|
+
"import": "./dist/v1/index.js"
|
|
10
|
+
}
|
|
11
|
+
},
|
|
12
|
+
"files": [
|
|
13
|
+
"dist",
|
|
14
|
+
"src",
|
|
15
|
+
"LICENSE"
|
|
16
|
+
],
|
|
17
|
+
"scripts": {
|
|
18
|
+
"build": "tsc -b",
|
|
19
|
+
"lint": "eslint src --ext .ts",
|
|
20
|
+
"lint:fix": "eslint src --ext .ts --fix",
|
|
21
|
+
"format": "prettier --write \"src/**/*.ts\"",
|
|
22
|
+
"format:check": "prettier --check \"src/**/*.ts\"",
|
|
23
|
+
"typecheck": "tsc --noEmit",
|
|
24
|
+
"clean": "rm -rf dist tsconfig.tsbuildinfo",
|
|
25
|
+
"test": "echo 'no tests'",
|
|
26
|
+
"test:browser": "echo 'no tests'"
|
|
27
|
+
},
|
|
28
|
+
"author": "Databricks",
|
|
29
|
+
"license": "Apache-2.0",
|
|
30
|
+
"dependencies": {
|
|
31
|
+
"@databricks/sdk-auth": ">=0.1.0-dev.2 <1.0.0",
|
|
32
|
+
"@databricks/sdk-core": ">=0.1.0-dev.3 <1.0.0",
|
|
33
|
+
"@databricks/sdk-options": ">=0.1.0-dev.2 <1.0.0",
|
|
34
|
+
"@js-temporal/polyfill": "^0.5.0",
|
|
35
|
+
"json-bigint": "^1.0.0",
|
|
36
|
+
"zod": "^4.3.6"
|
|
37
|
+
},
|
|
38
|
+
"engines": {
|
|
39
|
+
"node": ">=22.0.0"
|
|
40
|
+
}
|
|
7
41
|
}
|
package/src/v1/client.ts
ADDED
|
@@ -0,0 +1,206 @@
|
|
|
1
|
+
// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
|
|
2
|
+
|
|
3
|
+
import {VERSION as AUTH_VERSION} from '@databricks/sdk-auth';
|
|
4
|
+
import type {Call} from '@databricks/sdk-core/api';
|
|
5
|
+
import {createDefault} from '@databricks/sdk-core/clientinfo';
|
|
6
|
+
import type {Logger} from '@databricks/sdk-core/logger';
|
|
7
|
+
import {NoOpLogger} from '@databricks/sdk-core/logger';
|
|
8
|
+
import type {CallOptions} from '@databricks/sdk-options/call';
|
|
9
|
+
import type {ClientOptions} from '@databricks/sdk-options/client';
|
|
10
|
+
import type {HttpClient} from '@databricks/sdk-core/http';
|
|
11
|
+
import {newHttpClient} from './transport';
|
|
12
|
+
import {
|
|
13
|
+
buildHttpRequest,
|
|
14
|
+
executeCall,
|
|
15
|
+
executeHttpCall,
|
|
16
|
+
marshalRequest,
|
|
17
|
+
parseResponse,
|
|
18
|
+
} from './utils';
|
|
19
|
+
import pkgJson from '../../package.json' with {type: 'json'};
|
|
20
|
+
import type {
|
|
21
|
+
CreateTokenRequest,
|
|
22
|
+
CreateTokenRequest_Response,
|
|
23
|
+
ListTokensRequest,
|
|
24
|
+
ListTokensRequest_Response,
|
|
25
|
+
RevokeTokenRequest,
|
|
26
|
+
RevokeTokenRequest_Response,
|
|
27
|
+
UpdateTokenRequest,
|
|
28
|
+
UpdateTokenResponse,
|
|
29
|
+
} from './model';
|
|
30
|
+
import {
|
|
31
|
+
marshalCreateTokenRequestSchema,
|
|
32
|
+
marshalRevokeTokenRequestSchema,
|
|
33
|
+
marshalUpdateTokenRequestSchema,
|
|
34
|
+
unmarshalCreateTokenRequest_ResponseSchema,
|
|
35
|
+
unmarshalListTokensRequest_ResponseSchema,
|
|
36
|
+
unmarshalRevokeTokenRequest_ResponseSchema,
|
|
37
|
+
unmarshalUpdateTokenResponseSchema,
|
|
38
|
+
} from './model';
|
|
39
|
+
|
|
40
|
+
// Package identity segment for this client to be used in the User-Agent header.
|
|
41
|
+
const PACKAGE_SEGMENT = {
|
|
42
|
+
key: 'sdk-js-' + pkgJson.name.replace(/^@[^/]+\/sdk-/, ''),
|
|
43
|
+
value: pkgJson.version,
|
|
44
|
+
};
|
|
45
|
+
|
|
46
|
+
export class TokensClient {
|
|
47
|
+
private readonly host: string;
|
|
48
|
+
// Workspace ID used to route workspace-level calls on unified hosts (SPOG).
|
|
49
|
+
// When set, workspace-level methods send X-Databricks-Org-Id on every
|
|
50
|
+
// request.
|
|
51
|
+
private readonly workspaceId: string | undefined;
|
|
52
|
+
private readonly httpClient: HttpClient;
|
|
53
|
+
private readonly logger: Logger;
|
|
54
|
+
// User-Agent header value. Composed once at construction from
|
|
55
|
+
// createDefault() merged with this package's identity and the active
|
|
56
|
+
// credential's name.
|
|
57
|
+
private readonly userAgent: string;
|
|
58
|
+
|
|
59
|
+
constructor(options: ClientOptions) {
|
|
60
|
+
if (options.host === undefined) {
|
|
61
|
+
throw new Error('Host is required.');
|
|
62
|
+
}
|
|
63
|
+
this.host = options.host.replace(/\/$/, '');
|
|
64
|
+
this.workspaceId = options.workspaceId;
|
|
65
|
+
this.logger = options.logger ?? new NoOpLogger();
|
|
66
|
+
const info = createDefault()
|
|
67
|
+
.with(PACKAGE_SEGMENT)
|
|
68
|
+
.with({key: 'sdk-js-auth', value: AUTH_VERSION})
|
|
69
|
+
.with({key: 'auth', value: options.credentials?.name() ?? 'default'});
|
|
70
|
+
this.userAgent = info.toString();
|
|
71
|
+
this.httpClient = newHttpClient(options);
|
|
72
|
+
}
|
|
73
|
+
|
|
74
|
+
/**
|
|
75
|
+
* Creates and returns a token for a user. If this call is made through token authentication, it creates
|
|
76
|
+
* a token with the same client ID as the authenticated token. If the user's token quota is exceeded, this call
|
|
77
|
+
* returns an error **QUOTA_EXCEEDED**.
|
|
78
|
+
*/
|
|
79
|
+
async createToken(
|
|
80
|
+
req: CreateTokenRequest,
|
|
81
|
+
options?: CallOptions
|
|
82
|
+
): Promise<CreateTokenRequest_Response> {
|
|
83
|
+
const url = `${this.host}/api/2.0/token/create`;
|
|
84
|
+
const body = marshalRequest(req, marshalCreateTokenRequestSchema);
|
|
85
|
+
let resp: CreateTokenRequest_Response | undefined;
|
|
86
|
+
const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
87
|
+
const headers = new Headers({'Content-Type': 'application/json'});
|
|
88
|
+
if (this.workspaceId !== undefined) {
|
|
89
|
+
headers.set('X-Databricks-Org-Id', this.workspaceId);
|
|
90
|
+
}
|
|
91
|
+
headers.set('User-Agent', this.userAgent);
|
|
92
|
+
const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
|
|
93
|
+
const respBody = await executeHttpCall({
|
|
94
|
+
request: httpReq,
|
|
95
|
+
httpClient: this.httpClient,
|
|
96
|
+
logger: this.logger,
|
|
97
|
+
});
|
|
98
|
+
resp = parseResponse(
|
|
99
|
+
respBody,
|
|
100
|
+
unmarshalCreateTokenRequest_ResponseSchema
|
|
101
|
+
);
|
|
102
|
+
};
|
|
103
|
+
await executeCall(call, options);
|
|
104
|
+
if (resp === undefined) {
|
|
105
|
+
throw new Error('API call completed without a result.');
|
|
106
|
+
}
|
|
107
|
+
return resp;
|
|
108
|
+
}
|
|
109
|
+
|
|
110
|
+
/** Lists all the valid tokens for a user-workspace pair. */
|
|
111
|
+
async listTokens(
|
|
112
|
+
_req: ListTokensRequest,
|
|
113
|
+
options?: CallOptions
|
|
114
|
+
): Promise<ListTokensRequest_Response> {
|
|
115
|
+
const url = `${this.host}/api/2.0/token/list`;
|
|
116
|
+
let resp: ListTokensRequest_Response | undefined;
|
|
117
|
+
const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
118
|
+
const headers = new Headers();
|
|
119
|
+
if (this.workspaceId !== undefined) {
|
|
120
|
+
headers.set('X-Databricks-Org-Id', this.workspaceId);
|
|
121
|
+
}
|
|
122
|
+
headers.set('User-Agent', this.userAgent);
|
|
123
|
+
const httpReq = buildHttpRequest('GET', url, headers, callSignal);
|
|
124
|
+
const respBody = await executeHttpCall({
|
|
125
|
+
request: httpReq,
|
|
126
|
+
httpClient: this.httpClient,
|
|
127
|
+
logger: this.logger,
|
|
128
|
+
});
|
|
129
|
+
resp = parseResponse(respBody, unmarshalListTokensRequest_ResponseSchema);
|
|
130
|
+
};
|
|
131
|
+
await executeCall(call, options);
|
|
132
|
+
if (resp === undefined) {
|
|
133
|
+
throw new Error('API call completed without a result.');
|
|
134
|
+
}
|
|
135
|
+
return resp;
|
|
136
|
+
}
|
|
137
|
+
|
|
138
|
+
/**
|
|
139
|
+
* Revokes an access token.
|
|
140
|
+
*
|
|
141
|
+
* If a token with the specified ID is not valid, this call returns an error **RESOURCE_DOES_NOT_EXIST**.
|
|
142
|
+
*/
|
|
143
|
+
async revokeToken(
|
|
144
|
+
req: RevokeTokenRequest,
|
|
145
|
+
options?: CallOptions
|
|
146
|
+
): Promise<RevokeTokenRequest_Response> {
|
|
147
|
+
const url = `${this.host}/api/2.0/token/delete`;
|
|
148
|
+
const body = marshalRequest(req, marshalRevokeTokenRequestSchema);
|
|
149
|
+
let resp: RevokeTokenRequest_Response | undefined;
|
|
150
|
+
const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
151
|
+
const headers = new Headers({'Content-Type': 'application/json'});
|
|
152
|
+
if (this.workspaceId !== undefined) {
|
|
153
|
+
headers.set('X-Databricks-Org-Id', this.workspaceId);
|
|
154
|
+
}
|
|
155
|
+
headers.set('User-Agent', this.userAgent);
|
|
156
|
+
const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
|
|
157
|
+
const respBody = await executeHttpCall({
|
|
158
|
+
request: httpReq,
|
|
159
|
+
httpClient: this.httpClient,
|
|
160
|
+
logger: this.logger,
|
|
161
|
+
});
|
|
162
|
+
resp = parseResponse(
|
|
163
|
+
respBody,
|
|
164
|
+
unmarshalRevokeTokenRequest_ResponseSchema
|
|
165
|
+
);
|
|
166
|
+
};
|
|
167
|
+
await executeCall(call, options);
|
|
168
|
+
if (resp === undefined) {
|
|
169
|
+
throw new Error('API call completed without a result.');
|
|
170
|
+
}
|
|
171
|
+
return resp;
|
|
172
|
+
}
|
|
173
|
+
|
|
174
|
+
/**
|
|
175
|
+
* Updates the comment or scopes of a token.
|
|
176
|
+
*
|
|
177
|
+
* If a token with the specified ID is not valid, this call returns an error **NOT_FOUND**.
|
|
178
|
+
*/
|
|
179
|
+
async updateToken(
|
|
180
|
+
req: UpdateTokenRequest,
|
|
181
|
+
options?: CallOptions
|
|
182
|
+
): Promise<UpdateTokenResponse> {
|
|
183
|
+
const url = `${this.host}/api/2.0/token/${req.tokenId ?? ''}`;
|
|
184
|
+
const body = marshalRequest(req, marshalUpdateTokenRequestSchema);
|
|
185
|
+
let resp: UpdateTokenResponse | undefined;
|
|
186
|
+
const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
187
|
+
const headers = new Headers({'Content-Type': 'application/json'});
|
|
188
|
+
if (this.workspaceId !== undefined) {
|
|
189
|
+
headers.set('X-Databricks-Org-Id', this.workspaceId);
|
|
190
|
+
}
|
|
191
|
+
headers.set('User-Agent', this.userAgent);
|
|
192
|
+
const httpReq = buildHttpRequest('PATCH', url, headers, callSignal, body);
|
|
193
|
+
const respBody = await executeHttpCall({
|
|
194
|
+
request: httpReq,
|
|
195
|
+
httpClient: this.httpClient,
|
|
196
|
+
logger: this.logger,
|
|
197
|
+
});
|
|
198
|
+
resp = parseResponse(respBody, unmarshalUpdateTokenResponseSchema);
|
|
199
|
+
};
|
|
200
|
+
await executeCall(call, options);
|
|
201
|
+
if (resp === undefined) {
|
|
202
|
+
throw new Error('API call completed without a result.');
|
|
203
|
+
}
|
|
204
|
+
return resp;
|
|
205
|
+
}
|
|
206
|
+
}
|
package/src/v1/index.ts
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
|
|
2
|
+
|
|
3
|
+
export {TokensClient} from './client';
|
|
4
|
+
|
|
5
|
+
export {AutoscopeState} from './model';
|
|
6
|
+
|
|
7
|
+
export type {
|
|
8
|
+
CreateTokenRequest,
|
|
9
|
+
CreateTokenRequest_Response,
|
|
10
|
+
ListTokensRequest,
|
|
11
|
+
ListTokensRequest_Response,
|
|
12
|
+
PublicTokenInfo,
|
|
13
|
+
RevokeTokenRequest,
|
|
14
|
+
RevokeTokenRequest_Response,
|
|
15
|
+
UpdateTokenRequest,
|
|
16
|
+
UpdateTokenResponse,
|
|
17
|
+
} from './model';
|
|
18
|
+
|
|
19
|
+
export {publicTokenInfoFieldMask} from './model';
|