UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-03-05 03:48:55 +00:00
Code Issues Actions10 Packages Projects Releases Wiki Activity

fix: Resolve Google Drive picker promise with file data

Browse Source
This commit is contained in:
Taylor Wilsdon (aider) 2024-12-15 16:31:30 -05:00
parent 90e70608b9
commit 7bc1876e37
1 changed files with 37 additions and 29 deletions

66
src/lib/utils/google-drive-picker.ts
View File

@ -47,34 +47,42 @@ export const getAuthToken = async () => {
return oauthToken; return oauthToken;
}; };
export const createPicker = async () => { export const createPicker = () => {
if (!pickerApiLoaded) { return new Promise(async (resolve, reject) => {
await loadGoogleDriveApi(); try {
} if (!pickerApiLoaded) {
await loadGoogleDriveApi();
const token = await getAuthToken();
if (!token) {
throw new Error('Unable to get OAuth token');
}
const picker = new google.picker.PickerBuilder()
.addView(google.picker.ViewId.DOCS)
.setOAuthToken(token)
.setDeveloperKey(API_KEY)
.setCallback((data: any) => {
if (data[google.picker.Response.ACTION] === google.picker.Action.PICKED) {
const doc = data[google.picker.Response.DOCUMENTS][0];
const fileId = doc[google.picker.Document.ID];
const fileName = doc[google.picker.Document.NAME];
const fileUrl = doc[google.picker.Document.URL];
return {
id: fileId,
name: fileName,
url: fileUrl
};
} }
})
.build(); const token = await getAuthToken();
picker.setVisible(true); if (!token) {
throw new Error('Unable to get OAuth token');
}
const picker = new google.picker.PickerBuilder()
.addView(google.picker.ViewId.DOCS)
.setOAuthToken(token)
.setDeveloperKey(API_KEY)
.setCallback((data: any) => {
if (data[google.picker.Response.ACTION] === google.picker.Action.PICKED) {
const doc = data[google.picker.Response.DOCUMENTS][0];
const fileId = doc[google.picker.Document.ID];
const fileName = doc[google.picker.Document.NAME];
const fileUrl = doc[google.picker.Document.URL];
resolve({
id: fileId,
name: fileName,
url: fileUrl
});
} else if (data[google.picker.Response.ACTION] === google.picker.Action.CANCEL) {
resolve(null);
}
})
.build();
picker.setVisible(true);
} catch (error) {
reject(error);
}
});
}; };