-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
improve client library, apis and error handling
- Loading branch information
1 parent
27105e1
commit d3904cd
Showing
7 changed files
with
255 additions
and
124 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file was deleted.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,175 @@ | ||
| import { DEFAULT_HEADERS } from "../../drivers/constants" | ||
| import { SQLiteCloudError } from "../../drivers/types" | ||
| import { getAPIUrl } from "../utils" | ||
| import { Fetch, fetchWithAuth } from "../utils/fetch" | ||
|
|
||
| interface StorageResponse { | ||
| data: any | ||
| error: any | ||
| } | ||
|
|
||
| interface Storage { | ||
| createBucket(bucket: string): Promise<StorageResponse> | ||
| getBucket(bucket: string): Promise<StorageResponse> | ||
| deleteBucket(bucket: string): Promise<StorageResponse> | ||
| listBuckets(): Promise<StorageResponse> | ||
| upload(bucket: string, pathname: string, file: File | Buffer | Blob | string, options: { contentType: string }): Promise<StorageResponse> | ||
| download(bucket: string, pathname: string): Promise<StorageResponse> | ||
| remove(bucket: string, pathName: string): Promise<StorageResponse> | ||
| list(bucket: string): Promise<StorageResponse> | ||
| } | ||
|
|
||
| export class StorageClient implements Storage { | ||
| protected filesUrl: string | ||
| protected webliteSQLUrl: string | ||
| protected headers: Record<string, string> | ||
| protected fetch: Fetch | ||
|
|
||
| constructor( | ||
| connectionString: string, | ||
| options: { | ||
| customFetch?: Fetch, | ||
| headers?: Record<string, string> | ||
| } = {}) { | ||
| this.filesUrl = getAPIUrl(connectionString, 'files') | ||
| this.webliteSQLUrl = getAPIUrl(connectionString, 'weblite/sql') | ||
| this.fetch = options.customFetch || fetchWithAuth(connectionString) | ||
| this.headers = options.headers ? { ...DEFAULT_HEADERS, ...options.headers } : { ...DEFAULT_HEADERS } | ||
| } | ||
|
|
||
| async createBucket(bucket: string) { | ||
| const sql = `USE DATABASE files; INSERT INTO files (Bucket) VALUES ('${bucket}');` | ||
|
|
||
| try { | ||
| const response = await this.fetch(this.webliteSQLUrl, { | ||
| method: 'POST', | ||
| body: JSON.stringify({ sql }), | ||
| headers: this.headers | ||
| }) | ||
|
|
||
| if (!response.ok) { | ||
| throw new SQLiteCloudError(`Failed to create bucket: ${response.statusText}`) | ||
| } | ||
|
|
||
| return { data: await response.json(), error: null } | ||
| } catch (error) { | ||
| return { data: null, error } | ||
| } | ||
| } | ||
|
|
||
| async getBucket(bucket: string) { | ||
| const url = `${this.filesUrl}/${bucket}` | ||
| const response = await this.fetch(url, { method: 'GET', headers: this.headers }) | ||
| if (!response.ok) { | ||
| throw new SQLiteCloudError(`Failed to get bucket: ${response.statusText}`) | ||
| } | ||
|
|
||
| return { data: await response.json(), error: null } | ||
| } | ||
|
|
||
| async deleteBucket(bucket: string) { | ||
| const url = `${this.filesUrl}/${bucket}` | ||
| try { | ||
| const response = await this.fetch(url, { method: 'DELETE', headers: this.headers }) | ||
| if (!response.ok) { | ||
| throw new SQLiteCloudError(`Failed to delete bucket: ${response.statusText}`) | ||
| } | ||
| return { data: await response.json(), error: null } | ||
| } catch (error) { | ||
| return { data: null, error } | ||
| } | ||
| } | ||
|
|
||
| async listBuckets() { | ||
| const sql = `USE DATABASE files.sqlite; SELECT * FROM files;` | ||
| try { | ||
| const response = await this.fetch(this.webliteSQLUrl, { | ||
| method: 'POST', | ||
| body: JSON.stringify({ sql }), | ||
| headers: this.headers | ||
| }) | ||
| if (!response.ok) { | ||
| throw new SQLiteCloudError(`Failed to list buckets: ${response.statusText}`) | ||
| } | ||
| return { data: await response.json(), error: null } | ||
| } catch (error) { | ||
| return { | ||
| data: null, | ||
| error | ||
| } | ||
| } | ||
| } | ||
|
|
||
| async upload(bucket: string, pathname: string, file: File | Buffer | Blob | string, options: { contentType: string }) { | ||
| const url = `${this.filesUrl}/${bucket}/${pathname}`; | ||
| const headers = { | ||
| 'Content-Type': options?.contentType || 'application/octet-stream' | ||
| } | ||
| try { | ||
| const response = await this.fetch(url, { method: 'POST', body: file, headers }) | ||
| if (!response.ok) { | ||
| throw new SQLiteCloudError(`Failed to upload file: ${response.statusText}`) | ||
| } | ||
| return { data: await response.json(), error: null } | ||
| } catch (error) { | ||
| return { data: null, error } | ||
| } | ||
| } | ||
|
|
||
| async download(bucket: string, pathname: string) { | ||
| const url = `${this.filesUrl}/${bucket}/${pathname}`; | ||
| try { | ||
| const response = await this.fetch(url, { method: 'GET' }) | ||
| if (!response.ok) { | ||
| throw new SQLiteCloudError(`Failed to download file: ${response.statusText}`) | ||
| } | ||
| let responseType = (response.headers.get('Content-Type') ?? 'text/plain').split(';')[0].trim() | ||
| let data: any | ||
| if (responseType === 'application/json') { | ||
| data = await response.json() | ||
| } else if (responseType === 'application/octet-stream') { | ||
| data = await response.blob() | ||
| } else if (responseType === 'text/event-stream') { | ||
| data = response | ||
| } else if (responseType === 'multipart/form-data') { | ||
| data = await response.formData() | ||
| } else { | ||
| // default to text | ||
| data = await response.text() | ||
| } | ||
| return { data, error: null } | ||
| } catch (error) { | ||
| return { data: null, error } | ||
| } | ||
| } | ||
|
|
||
| async remove(bucket: string, pathName: string) { | ||
| const url = `${this.filesUrl}/${bucket}/${pathName}` | ||
| try { | ||
| const response = await this.fetch(url, { method: 'DELETE' }) | ||
| if (!response.ok) { | ||
| throw new SQLiteCloudError(`Failed to remove file: ${response.statusText}`) | ||
| } | ||
| return { data: response.json(), error: null } | ||
| } catch (error) { | ||
| return { data: null, error } | ||
| } | ||
| } | ||
|
|
||
| async list(bucket: string) { | ||
| const sql = `USE DATABASE files.sqlite; SELECT * FROM files WHERE bucket = '${bucket}'` | ||
| try { | ||
| const response = await this.fetch(this.webliteSQLUrl, { | ||
| method: 'POST', | ||
| body: JSON.stringify({ sql }), | ||
| headers: this.headers | ||
| }) | ||
| if (!response.ok) { | ||
| throw new SQLiteCloudError(`Failed to list files: ${response.statusText}`) | ||
| } | ||
| return { data: await response.json(), error: null } | ||
| } catch (error) { | ||
| return { data: null, error } | ||
| } | ||
| } | ||
| } |
Oops, something went wrong.