|
'use strict' |
|
|
|
if (process.argv.length != 4) { |
|
console.log(`Usage: ${process.argv[1]} [Share Drive ID] [Parent Folder ID]`) |
|
return 1 |
|
} |
|
|
|
const shareDriveId = process.argv[2] |
|
const parentFolderId = process.argv[3] |
|
|
|
const fs = require('fs') |
|
const readline = require('readline') |
|
const { google } = require('googleapis') |
|
|
|
// If modifying these scopes, delete token.json. |
|
const SCOPES = ['https://www.googleapis.com/auth/drive.metadata.readonly'] |
|
// The file token.json stores the user's access and refresh tokens, and is |
|
// created automatically when the authorization flow completes for the first |
|
// time. |
|
const TOKEN_PATH = 'token.json' |
|
|
|
// Load client secrets from a local file. |
|
fs.readFile('credentials.json', (err, content) => { |
|
if (err) return console.log('Error loading client secret file:', err) |
|
// Authorize a client with credentials, then call the Google Drive API. |
|
authorize(JSON.parse(content), listFiles) |
|
}) |
|
|
|
/** |
|
* Create an OAuth2 client with the given credentials, and then execute the |
|
* given callback function. |
|
* @param {Object} credentials The authorization client credentials. |
|
* @param {function} callback The callback to call with the authorized client. |
|
*/ |
|
function authorize(credentials, callback) { |
|
const { client_secret, client_id, redirect_uris } = credentials.installed |
|
const oAuth2Client = new google.auth.OAuth2( |
|
client_id, |
|
client_secret, |
|
redirect_uris[0], |
|
) |
|
|
|
// Check if we have previously stored a token. |
|
fs.readFile(TOKEN_PATH, (err, token) => { |
|
if (err) return getAccessToken(oAuth2Client, callback) |
|
oAuth2Client.setCredentials(JSON.parse(token)) |
|
callback(oAuth2Client) |
|
}) |
|
} |
|
|
|
/** |
|
* Get and store new token after prompting for user authorization, and then |
|
* execute the given callback with the authorized OAuth2 client. |
|
* @param {google.auth.OAuth2} oAuth2Client The OAuth2 client to get token for. |
|
* @param {getEventsCallback} callback The callback for the authorized client. |
|
*/ |
|
function getAccessToken(oAuth2Client, callback) { |
|
const authUrl = oAuth2Client.generateAuthUrl({ |
|
access_type: 'offline', |
|
scope: SCOPES, |
|
}) |
|
console.log('Authorize this app by visiting this url:', authUrl) |
|
const rl = readline.createInterface({ |
|
input: process.stdin, |
|
output: process.stdout, |
|
}) |
|
rl.question('Enter the code from that page here: ', (code) => { |
|
rl.close() |
|
oAuth2Client.getToken(code, (err, token) => { |
|
if (err) return console.error('Error retrieving access token', err) |
|
oAuth2Client.setCredentials(token) |
|
// Store the token to disk for later program executions |
|
fs.writeFile(TOKEN_PATH, JSON.stringify(token), (err) => { |
|
if (err) return console.error(err) |
|
console.log('Token stored to', TOKEN_PATH) |
|
}) |
|
callback(oAuth2Client) |
|
}) |
|
}) |
|
} |
|
|
|
function getFilesFromDrive(drive, parentId) { |
|
return new Promise((resolve) => { |
|
drive.files.list( |
|
{ |
|
pageSize: 100, |
|
fields: 'nextPageToken, files(id, name, mimeType)', |
|
spaces: 'drive', |
|
corpora: 'drive', |
|
driveId: shareDriveId, |
|
includeItemsFromAllDrives: true, |
|
supportsAllDrives: true, |
|
q: `"${parentId}" in parents`, |
|
}, |
|
(err, res) => { |
|
proccessFiles(err, res, drive).then((results) => { |
|
// files.splice(0, 0, ...results) |
|
resolve(results) |
|
}) |
|
}, |
|
) |
|
}) |
|
} |
|
|
|
async function proccessFiles(err, res, drive) { |
|
if (err) return console.log('The API returned an error: ' + err) |
|
|
|
const files = res.data.files.filter( |
|
(f) => f.mimeType !== 'application/vnd.google-apps.folder', |
|
) |
|
const folders = res.data.files.filter( |
|
(f) => f.mimeType == 'application/vnd.google-apps.folder', |
|
) |
|
for (const f of folders) { |
|
const inFiles = (await getFilesFromDrive(drive, f.id)).map((i) => { |
|
i.name = `${f.name}/${i.name}` |
|
return i |
|
}) |
|
files.splice(0, 0, ...inFiles) |
|
} |
|
|
|
return files |
|
} |
|
|
|
function listFiles(auth) { |
|
const drive = google.drive({ version: 'v3', auth }) |
|
getFilesFromDrive(drive, parentFolderId).then((files) => { |
|
if (files) { |
|
files.map((f) => { |
|
console.log(f.name) |
|
}) |
|
} |
|
}) |
|
} |