@datadog/datadog-api-client 1.59.0 → 1.60.0

package/dist/packages/datadog-api-client-v2/apis/CloudCostManagementApi.js

@@ -343,6 +343,33 @@ class CloudCostManagementApiRequestFactory extends baseapi_1.BaseAPIRequestFacto
             return requestContext;
         });
     }
+    deleteCustomForecast(budgetId, _options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const _config = _options || this.configuration;
+            logger_1.logger.warn("Using unstable operation 'deleteCustomForecast'");
+            if (!_config.unstableOperations["v2.deleteCustomForecast"]) {
+                throw new Error("Unstable operation 'deleteCustomForecast' is disabled");
+            }
+            // verify required parameter 'budgetId' is not null or undefined
+            if (budgetId === null || budgetId === undefined) {
+                throw new baseapi_1.RequiredError("budgetId", "deleteCustomForecast");
+            }
+            // Path Params
+            const localVarPath = "/api/v2/cost/budget/{budget_id}/custom-forecast".replace("{budget_id}", encodeURIComponent(String(budgetId)));
+            // Make Request Context
+            const requestContext = _config
+                .getServer("v2.CloudCostManagementApi.deleteCustomForecast")
+                .makeRequestContext(localVarPath, http_1.HttpMethod.DELETE);
+            requestContext.setHeaderParam("Accept", "*/*");
+            requestContext.setHttpConfig(_config.httpConfig);
+            // Apply auth methods
+            (0, configuration_1.applySecurityAuthentication)(_config, requestContext, [
+                "apiKeyAuth",
+                "appKeyAuth",
+            ]);
+            return requestContext;
+        });
+    }
     deleteTagPipelinesRuleset(rulesetId, _options) {
         return __awaiter(this, void 0, void 0, function* () {
             const _config = _options || this.configuration;
@@ -391,7 +418,7 @@ class CloudCostManagementApiRequestFactory extends baseapi_1.BaseAPIRequestFacto
             return requestContext;
         });
     }
-    getBudget(budgetId, _options) {
+    getBudget(budgetId, actual, forecast, start, end, _options) {
         return __awaiter(this, void 0, void 0, function* () {
             const _config = _options || this.configuration;
             // verify required parameter 'budgetId' is not null or undefined
@@ -406,6 +433,19 @@ class CloudCostManagementApiRequestFactory extends baseapi_1.BaseAPIRequestFacto
                 .makeRequestContext(localVarPath, http_1.HttpMethod.GET);
             requestContext.setHeaderParam("Accept", "application/json");
             requestContext.setHttpConfig(_config.httpConfig);
+            // Query Params
+            if (actual !== undefined) {
+                requestContext.setQueryParam("actual", ObjectSerializer_1.ObjectSerializer.serialize(actual, "boolean", ""), "");
+            }
+            if (forecast !== undefined) {
+                requestContext.setQueryParam("forecast", ObjectSerializer_1.ObjectSerializer.serialize(forecast, "boolean", ""), "");
+            }
+            if (start !== undefined) {
+                requestContext.setQueryParam("start", ObjectSerializer_1.ObjectSerializer.serialize(start, "number", "int64"), "");
+            }
+            if (end !== undefined) {
+                requestContext.setQueryParam("end", ObjectSerializer_1.ObjectSerializer.serialize(end, "number", "int64"), "");
+            }
             // Apply auth methods
             (0, configuration_1.applySecurityAuthentication)(_config, requestContext, [
                 "apiKeyAuth",
@@ -2059,6 +2099,40 @@ class CloudCostManagementApiRequestFactory extends baseapi_1.BaseAPIRequestFacto
             return requestContext;
         });
     }
+    upsertCustomForecast(body, _options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const _config = _options || this.configuration;
+            logger_1.logger.warn("Using unstable operation 'upsertCustomForecast'");
+            if (!_config.unstableOperations["v2.upsertCustomForecast"]) {
+                throw new Error("Unstable operation 'upsertCustomForecast' is disabled");
+            }
+            // verify required parameter 'body' is not null or undefined
+            if (body === null || body === undefined) {
+                throw new baseapi_1.RequiredError("body", "upsertCustomForecast");
+            }
+            // Path Params
+            const localVarPath = "/api/v2/cost/budget/custom-forecast";
+            // Make Request Context
+            const requestContext = _config
+                .getServer("v2.CloudCostManagementApi.upsertCustomForecast")
+                .makeRequestContext(localVarPath, http_1.HttpMethod.PUT);
+            requestContext.setHeaderParam("Accept", "application/json");
+            requestContext.setHttpConfig(_config.httpConfig);
+            // Body Params
+            const contentType = ObjectSerializer_1.ObjectSerializer.getPreferredMediaType([
+                "application/json",
+            ]);
+            requestContext.setHeaderParam("Content-Type", contentType);
+            const serializedBody = ObjectSerializer_1.ObjectSerializer.stringify(ObjectSerializer_1.ObjectSerializer.serialize(body, "CustomForecastUpsertRequest", ""), contentType);
+            requestContext.setBody(serializedBody);
+            // Apply auth methods
+            (0, configuration_1.applySecurityAuthentication)(_config, requestContext, [
+                "apiKeyAuth",
+                "appKeyAuth",
+            ]);
+            return requestContext;
+        });
+    }
     validateBudget(body, _options) {
         return __awaiter(this, void 0, void 0, function* () {
             const _config = _options || this.configuration;
@@ -2559,6 +2633,41 @@ class CloudCostManagementApiResponseProcessor {
             throw new exception_1.ApiException(response.httpStatusCode, 'Unknown API Status Code!\nBody: "' + body + '"');
         });
     }
+    /**
+     * Unwraps the actual response sent by the server from the response context and deserializes the response content
+     * to the expected objects
+     *
+     * @params response Response returned by the server for a request to deleteCustomForecast
+     * @throws ApiException if the response code was not in [200, 299]
+     */
+    deleteCustomForecast(response) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const contentType = ObjectSerializer_1.ObjectSerializer.normalizeMediaType(response.headers["content-type"]);
+            if (response.httpStatusCode === 204) {
+                return;
+            }
+            if (response.httpStatusCode === 400 ||
+                response.httpStatusCode === 404 ||
+                response.httpStatusCode === 429) {
+                const bodyText = ObjectSerializer_1.ObjectSerializer.parse(yield response.body.text(), contentType);
+                let body;
+                try {
+                    body = ObjectSerializer_1.ObjectSerializer.deserialize(bodyText, "APIErrorResponse");
+                }
+                catch (error) {
+                    logger_1.logger.debug(`Got error deserializing error: ${error}`);
+                    throw new exception_1.ApiException(response.httpStatusCode, bodyText);
+                }
+                throw new exception_1.ApiException(response.httpStatusCode, body);
+            }
+            // Work around for missing responses in specification, e.g. for petstore.yaml
+            if (response.httpStatusCode >= 200 && response.httpStatusCode <= 299) {
+                return;
+            }
+            const body = (yield response.body.text()) || "";
+            throw new exception_1.ApiException(response.httpStatusCode, 'Unknown API Status Code!\nBody: "' + body + '"');
+        });
+    }
     /**
      * Unwraps the actual response sent by the server from the response context and deserializes the response content
      * to the expected objects
@@ -2643,7 +2752,9 @@ class CloudCostManagementApiResponseProcessor {
                 const body = ObjectSerializer_1.ObjectSerializer.deserialize(ObjectSerializer_1.ObjectSerializer.parse(yield response.body.text(), contentType), "BudgetWithEntries");
                 return body;
             }
-            if (response.httpStatusCode === 429) {
+            if (response.httpStatusCode === 400 ||
+                response.httpStatusCode === 404 ||
+                response.httpStatusCode === 429) {
                 const bodyText = ObjectSerializer_1.ObjectSerializer.parse(yield response.body.text(), contentType);
                 let body;
                 try {
@@ -4479,6 +4590,43 @@ class CloudCostManagementApiResponseProcessor {
             throw new exception_1.ApiException(response.httpStatusCode, 'Unknown API Status Code!\nBody: "' + body + '"');
         });
     }
+    /**
+     * Unwraps the actual response sent by the server from the response context and deserializes the response content
+     * to the expected objects
+     *
+     * @params response Response returned by the server for a request to upsertCustomForecast
+     * @throws ApiException if the response code was not in [200, 299]
+     */
+    upsertCustomForecast(response) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const contentType = ObjectSerializer_1.ObjectSerializer.normalizeMediaType(response.headers["content-type"]);
+            if (response.httpStatusCode === 200) {
+                const body = ObjectSerializer_1.ObjectSerializer.deserialize(ObjectSerializer_1.ObjectSerializer.parse(yield response.body.text(), contentType), "CustomForecastResponse");
+                return body;
+            }
+            if (response.httpStatusCode === 400 ||
+                response.httpStatusCode === 404 ||
+                response.httpStatusCode === 429) {
+                const bodyText = ObjectSerializer_1.ObjectSerializer.parse(yield response.body.text(), contentType);
+                let body;
+                try {
+                    body = ObjectSerializer_1.ObjectSerializer.deserialize(bodyText, "APIErrorResponse");
+                }
+                catch (error) {
+                    logger_1.logger.debug(`Got error deserializing error: ${error}`);
+                    throw new exception_1.ApiException(response.httpStatusCode, bodyText);
+                }
+                throw new exception_1.ApiException(response.httpStatusCode, body);
+            }
+            // Work around for missing responses in specification, e.g. for petstore.yaml
+            if (response.httpStatusCode >= 200 && response.httpStatusCode <= 299) {
+                const body = ObjectSerializer_1.ObjectSerializer.deserialize(ObjectSerializer_1.ObjectSerializer.parse(yield response.body.text(), contentType), "CustomForecastResponse", "");
+                return body;
+            }
+            const body = (yield response.body.text()) || "";
+            throw new exception_1.ApiException(response.httpStatusCode, 'Unknown API Status Code!\nBody: "' + body + '"');
+        });
+    }
     /**
      * Unwraps the actual response sent by the server from the response context and deserializes the response content
      * to the expected objects
@@ -4774,6 +4922,20 @@ class CloudCostManagementApi {
             });
         });
     }
+    /**
+     * Delete the custom forecast for a budget.
+     * @param param The request object
+     */
+    deleteCustomForecast(param, options) {
+        const requestContextPromise = this.requestFactory.deleteCustomForecast(param.budgetId, options);
+        return requestContextPromise.then((requestContext) => {
+            return this.configuration.httpApi
+                .send(requestContext)
+                .then((responseContext) => {
+                return this.responseProcessor.deleteCustomForecast(responseContext);
+            });
+        });
+    }
     /**
      * Delete a tag pipeline ruleset - Delete an existing tag pipeline ruleset by its ID
      * @param param The request object
@@ -4803,11 +4965,11 @@ class CloudCostManagementApi {
         });
     }
     /**
-     * Get a budget
+     * Get a budget by ID. Pass `actual=true` or `forecast=true` to include cost data in the response. Use `start` and `end` (millisecond epochs, both required) to set the cost window. When `forecast=true`, each entry also includes `ootb_forecast` (the ML forecast before overrides) and `custom_forecast` (`null` if no override is set, a number if one is).
      * @param param The request object
      */
     getBudget(param, options) {
-        const requestContextPromise = this.requestFactory.getBudget(param.budgetId, options);
+        const requestContextPromise = this.requestFactory.getBudget(param.budgetId, param.actual, param.forecast, param.start, param.end, options);
         return requestContextPromise.then((requestContext) => {
             return this.configuration.httpApi
                 .send(requestContext)
@@ -5508,6 +5670,21 @@ class CloudCostManagementApi {
             });
         });
     }
+    /**
+     * Create or replace the custom forecast for an existing budget.
+     * Pass an empty `entries` list to delete the custom forecast for the budget.
+     * @param param The request object
+     */
+    upsertCustomForecast(param, options) {
+        const requestContextPromise = this.requestFactory.upsertCustomForecast(param.body, options);
+        return requestContextPromise.then((requestContext) => {
+            return this.configuration.httpApi
+                .send(requestContext)
+                .then((responseContext) => {
+                return this.responseProcessor.upsertCustomForecast(responseContext);
+            });
+        });
+    }
     /**
      * Validate a budget configuration without creating or modifying it
      * @param param The request object