|
|
|
let API_KEY = ''; |
|
let CLIENT_ID = ''; |
|
|
|
|
|
async function getCredentials() { |
|
const response = await fetch('/api/config'); |
|
if (!response.ok) { |
|
throw new Error('Failed to fetch Google Drive credentials'); |
|
} |
|
const config = await response.json(); |
|
API_KEY = config.google_drive?.api_key; |
|
CLIENT_ID = config.google_drive?.client_id; |
|
|
|
if (!API_KEY || !CLIENT_ID) { |
|
throw new Error('Google Drive API credentials not configured'); |
|
} |
|
} |
|
const SCOPE = [ |
|
'https://www.googleapis.com/auth/drive.readonly', |
|
'https://www.googleapis.com/auth/drive.file' |
|
]; |
|
|
|
|
|
const validateCredentials = () => { |
|
if (!API_KEY || !CLIENT_ID) { |
|
throw new Error('Google Drive API credentials not configured'); |
|
} |
|
if (API_KEY === '' || CLIENT_ID === '') { |
|
throw new Error('Please configure valid Google Drive API credentials'); |
|
} |
|
}; |
|
|
|
let pickerApiLoaded = false; |
|
let oauthToken: string | null = null; |
|
let initialized = false; |
|
|
|
export const loadGoogleDriveApi = () => { |
|
return new Promise((resolve, reject) => { |
|
if (typeof gapi === 'undefined') { |
|
const script = document.createElement('script'); |
|
script.src = 'https://apis.google.com/js/api.js'; |
|
script.onload = () => { |
|
gapi.load('picker', () => { |
|
pickerApiLoaded = true; |
|
resolve(true); |
|
}); |
|
}; |
|
script.onerror = reject; |
|
document.body.appendChild(script); |
|
} else { |
|
gapi.load('picker', () => { |
|
pickerApiLoaded = true; |
|
resolve(true); |
|
}); |
|
} |
|
}); |
|
}; |
|
|
|
export const loadGoogleAuthApi = () => { |
|
return new Promise((resolve, reject) => { |
|
if (typeof google === 'undefined') { |
|
const script = document.createElement('script'); |
|
script.src = 'https://accounts.google.com/gsi/client'; |
|
script.onload = resolve; |
|
script.onerror = reject; |
|
document.body.appendChild(script); |
|
} else { |
|
resolve(true); |
|
} |
|
}); |
|
}; |
|
|
|
export const getAuthToken = async () => { |
|
if (!oauthToken) { |
|
return new Promise((resolve, reject) => { |
|
const tokenClient = google.accounts.oauth2.initTokenClient({ |
|
client_id: CLIENT_ID, |
|
scope: SCOPE.join(' '), |
|
callback: (response: any) => { |
|
if (response.access_token) { |
|
oauthToken = response.access_token; |
|
resolve(oauthToken); |
|
} else { |
|
reject(new Error('Failed to get access token')); |
|
} |
|
}, |
|
error_callback: (error: any) => { |
|
reject(new Error(error.message || 'OAuth error occurred')); |
|
} |
|
}); |
|
tokenClient.requestAccessToken(); |
|
}); |
|
} |
|
return oauthToken; |
|
}; |
|
|
|
const initialize = async () => { |
|
if (!initialized) { |
|
await getCredentials(); |
|
validateCredentials(); |
|
await Promise.all([loadGoogleDriveApi(), loadGoogleAuthApi()]); |
|
initialized = true; |
|
} |
|
}; |
|
|
|
export const createPicker = () => { |
|
return new Promise(async (resolve, reject) => { |
|
try { |
|
console.log('Initializing Google Drive Picker...'); |
|
await initialize(); |
|
console.log('Getting auth token...'); |
|
const token = await getAuthToken(); |
|
if (!token) { |
|
console.error('Failed to get OAuth token'); |
|
throw new Error('Unable to get OAuth token'); |
|
} |
|
console.log('Auth token obtained successfully'); |
|
|
|
const picker = new google.picker.PickerBuilder() |
|
.enableFeature(google.picker.Feature.NAV_HIDDEN) |
|
.enableFeature(google.picker.Feature.MULTISELECT_ENABLED) |
|
.addView( |
|
new google.picker.DocsView() |
|
.setIncludeFolders(false) |
|
.setSelectFolderEnabled(false) |
|
.setMimeTypes( |
|
'application/pdf,text/plain,application/vnd.openxmlformats-officedocument.wordprocessingml.document,application/vnd.google-apps.document,application/vnd.google-apps.spreadsheet,application/vnd.google-apps.presentation' |
|
) |
|
) |
|
.setOAuthToken(token) |
|
.setDeveloperKey(API_KEY) |
|
|
|
.setCallback(async (data: any) => { |
|
if (data[google.picker.Response.ACTION] === google.picker.Action.PICKED) { |
|
try { |
|
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]; |
|
|
|
if (!fileId || !fileName) { |
|
throw new Error('Required file details missing'); |
|
} |
|
|
|
|
|
const mimeType = doc[google.picker.Document.MIME_TYPE]; |
|
|
|
let downloadUrl; |
|
let exportFormat; |
|
|
|
if (mimeType.includes('google-apps')) { |
|
|
|
if (mimeType.includes('document')) { |
|
exportFormat = 'text/plain'; |
|
} else if (mimeType.includes('spreadsheet')) { |
|
exportFormat = 'text/csv'; |
|
} else if (mimeType.includes('presentation')) { |
|
exportFormat = 'text/plain'; |
|
} else { |
|
exportFormat = 'application/pdf'; |
|
} |
|
downloadUrl = `https://www.googleapis.com/drive/v3/files/${fileId}/export?mimeType=${encodeURIComponent(exportFormat)}`; |
|
} else { |
|
|
|
downloadUrl = `https://www.googleapis.com/drive/v3/files/${fileId}?alt=media`; |
|
} |
|
|
|
const response = await fetch(downloadUrl, { |
|
headers: { |
|
Authorization: `Bearer ${token}`, |
|
Accept: '*/*' |
|
} |
|
}); |
|
|
|
if (!response.ok) { |
|
const errorText = await response.text(); |
|
console.error('Download failed:', { |
|
status: response.status, |
|
statusText: response.statusText, |
|
error: errorText |
|
}); |
|
throw new Error(`Failed to download file (${response.status}): ${errorText}`); |
|
} |
|
|
|
const blob = await response.blob(); |
|
const result = { |
|
id: fileId, |
|
name: fileName, |
|
url: downloadUrl, |
|
blob: blob, |
|
headers: { |
|
Authorization: `Bearer ${token}`, |
|
Accept: '*/*' |
|
} |
|
}; |
|
resolve(result); |
|
} catch (error) { |
|
reject(error); |
|
} |
|
} else if (data[google.picker.Response.ACTION] === google.picker.Action.CANCEL) { |
|
resolve(null); |
|
} |
|
}) |
|
.build(); |
|
picker.setVisible(true); |
|
} catch (error) { |
|
console.error('Google Drive Picker error:', error); |
|
reject(error); |
|
} |
|
}); |
|
}; |
|
|