Skip to content

Instantly share code, notes, and snippets.

@tacck
Last active December 5, 2024 15:58
Show Gist options
  • Save tacck/2ae10129342d29622d2792a5c05d5c32 to your computer and use it in GitHub Desktop.
Save tacck/2ae10129342d29622d2792a5c05d5c32 to your computer and use it in GitHub Desktop.
Show file list in Google Share Drive's folder. base on https://developers.google.com/drive/api/v3/quickstart/nodejs
'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)
})
}
})
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment