import json
import os

from parable_types_web_api import PrepareArtifactUploadInput
from parable_web_api_sdk import ClientConfig, WebApiSDK

sdk = WebApiSDK(
    ClientConfig(
        base_url=os.environ["PARABLE_API_URL"],
        auth_token=os.environ["PARABLE_API_TOKEN"],
        default_headers={"X-Tenant": os.environ["PARABLE_WORKSPACE"]},
    )
)

input_data = PrepareArtifactUploadInput.model_validate(
    json.loads("{\"requestId\":\"550e8400-e29b-41d4-a716-446655440000\",\"fileName\":\"example\",\"fileSize\":1,\"contentType\":\"example\"}"),
)
result = sdk.artifacts.prepare_artifact_upload(input_data)

sdk, err := webapisdk.New(webapisdk.SDKConfig{
	BaseURL: os.Getenv("PARABLE_API_URL"),
	Auth:    &webapisdk.AuthConfig{Token: os.Getenv("PARABLE_API_TOKEN")},
	RequestInterceptor: func(ctx context.Context, req *http.Request) error {
		req.Header.Set("X-Tenant", os.Getenv("PARABLE_WORKSPACE"))
		return nil
	},
})
if err != nil {
	return err
}

var input webapitypes.PrepareArtifactUploadInput
if err := json.Unmarshal([]byte(`{"requestId":"550e8400-e29b-41d4-a716-446655440000","fileName":"example","fileSize":1,"contentType":"example"}`), &input); err != nil {
	return err
}
result, err := sdk.ArtifactsNamespace.PrepareArtifactUpload(ctx, input)

{
  "data": {
    "artifact": {
      "createdAt": "2023-11-07T05:31:56Z",
      "createdBy": "<string>",
      "fileName": "<string>",
      "fileSize": 4503599627370495,
      "fileType": "csv",
      "id": "<string>",
      "schemaInferenceStatus": "not_requested",
      "updatedAt": "2023-11-07T05:31:56Z",
      "updatedBy": "<string>",
      "uploadStatus": "pending",
      "uploadedBy": {
        "email": "jsmith@example.com",
        "id": "<string>",
        "name": "<string>"
      },
      "analysisWork": {
        "artifactId": "<string>",
        "state": "PENDING",
        "workId": "<string>",
        "errorCode": "INVALID_SOURCE",
        "errorMessage": "<string>",
        "tapId": "<string>"
      },
      "attributedTapId": "<string>",
      "dataSchema": {},
      "path": "<string>",
      "publishWork": {
        "artifactId": "<string>",
        "state": "PENDING",
        "workId": "<string>",
        "errorCode": "INVALID_SOURCE",
        "errorMessage": "<string>",
        "tapId": "<string>"
      }
    },
    "expiresAt": "2023-11-07T05:31:56Z",
    "requiredHeaders": [
      {
        "name": "<string>",
        "value": "<string>"
      }
    ],
    "uploadUrl": "<string>"
  },
  "meta": {
    "requestId": "<string>",
    "totalCount": 123
  },
  "links": {
    "first": "<string>",
    "last": "<string>",
    "next": "<string>",
    "prev": "<string>"
  }
}

{
  "detail": "<string>",
  "status": 400,
  "title": "Bad Request",
  "type": "about:blank",
  "code": "WA-VL-001",
  "details": "<unknown>",
  "errors": "<unknown>",
  "requestId": "<string>"
}

{
  "detail": "<string>",
  "status": 400,
  "title": "Bad Request",
  "type": "about:blank",
  "code": "WA-VL-001",
  "details": "<unknown>",
  "errors": "<unknown>",
  "requestId": "<string>"
}

{
  "detail": "<string>",
  "status": 400,
  "title": "Bad Request",
  "type": "about:blank",
  "code": "WA-VL-001",
  "details": "<unknown>",
  "errors": "<unknown>",
  "requestId": "<string>"
}

{
  "detail": "<string>",
  "status": 400,
  "title": "Bad Request",
  "type": "about:blank",
  "code": "WA-VL-001",
  "details": "<unknown>",
  "errors": "<unknown>",
  "requestId": "<string>"
}