|
17 | 17 | import com.skyflow.generated.rest.resources.records.requests.RecordServiceBatchOperationBody; |
18 | 18 | import com.skyflow.generated.rest.resources.records.requests.RecordServiceInsertRecordBody; |
19 | 19 | import com.skyflow.generated.rest.resources.records.requests.RecordServiceUpdateRecordBody; |
| 20 | +import com.skyflow.generated.rest.resources.records.requests.UploadFileV2Request; |
20 | 21 | import com.skyflow.generated.rest.resources.strings.StringsClient; |
21 | 22 | import com.skyflow.generated.rest.resources.strings.requests.DeidentifyStringRequest; |
22 | 23 | import com.skyflow.generated.rest.resources.strings.requests.ReidentifyStringRequest; |
|
32 | 33 | import com.skyflow.utils.Utils; |
33 | 34 | import com.skyflow.utils.logger.LogUtil; |
34 | 35 | import com.skyflow.utils.validations.Validations; |
| 36 | +import com.skyflow.vault.data.FileUploadRequest; |
35 | 37 | import com.skyflow.vault.data.InsertRequest; |
36 | 38 | import com.skyflow.vault.data.UpdateRequest; |
37 | 39 | import com.skyflow.vault.detect.*; |
|
44 | 46 | import io.github.cdimascio.dotenv.Dotenv; |
45 | 47 | import io.github.cdimascio.dotenv.DotenvException; |
46 | 48 |
|
| 49 | +import java.io.File; |
| 50 | +import java.io.IOException; |
| 51 | +import java.nio.file.Files; |
| 52 | +import java.nio.file.Paths; |
47 | 53 | import java.util.*; |
48 | 54 | import java.util.stream.Collectors; |
49 | 55 |
|
@@ -209,6 +215,22 @@ protected V1TokenizePayload getTokenizePayload(TokenizeRequest request) { |
209 | 215 | return payloadBuilder.build(); |
210 | 216 | } |
211 | 217 |
|
| 218 | + protected File getFileForFileUpload(FileUploadRequest fileUploadRequest) throws IOException { |
| 219 | + if (fileUploadRequest.getFilePath() != null) { |
| 220 | + return new File(fileUploadRequest.getFilePath()); |
| 221 | + } |
| 222 | + else if (fileUploadRequest.getBase64() != null) { |
| 223 | + byte[] decodedBytes = Base64.getDecoder().decode(fileUploadRequest.getBase64()); |
| 224 | + File file = new File(fileUploadRequest.getFileName()); |
| 225 | + Files.write(file.toPath(), decodedBytes); |
| 226 | + return file; |
| 227 | + } |
| 228 | + else if (fileUploadRequest.getFileObject() != null) { |
| 229 | + return fileUploadRequest.getFileObject(); |
| 230 | + } |
| 231 | + return null; |
| 232 | + } |
| 233 | + |
212 | 234 | protected void setBearerToken() throws SkyflowException { |
213 | 235 | prioritiseCredentials(); |
214 | 236 | Validations.validateCredentials(this.finalCredentials); |
|
0 commit comments