Created
December 28, 2016 17:32
-
-
Save joelongstreet/f30ec0028390195671118444cbd760d0 to your computer and use it in GitHub Desktop.
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
#!/usr/bin/env node | |
'use strict'; | |
const chalk = require('chalk'); | |
const del = require('del'); | |
const fs = require('fs'); | |
const google = require('googleapis'); | |
const googleAuth = require('google-auth-library'); | |
const _ = require('lodash'); | |
const async = require('async'); | |
const readline = require('readline'); | |
const IMAGE_DIR = '/home/osmc/Pictures/'; | |
// const IMAGE_DIR = '/Users/wpracht/Desktop/Junk\ Drawer/framepics/'; | |
// If modifying these scopes, delete your previously saved credentials | |
// at ~/.credentials/drive-nodejs-quickstart.json | |
const SCOPES = [ | |
'https://www.googleapis.com/auth/drive' | |
]; | |
const TOKEN_DIR = (process.env.HOME || process.env.HOMEPATH || | |
process.env.USERPROFILE) + '/.credentials/'; | |
const TOKEN_PATH = TOKEN_DIR + 'drive-nodejs-quickstart.json'; | |
// Load client secrets from a local file. | |
fs.readFile('client_secret.json', function processClientSecrets(err, content) { | |
if (err) { | |
console.err('Error loading client secret file: ' + err); | |
return; | |
} | |
// Authorize a client with the loaded credentials, then call the | |
// Drive API. | |
authorize(JSON.parse(content), downloadImages); | |
}); | |
/** | |
* Beautiful logging. | |
* | |
* @param {any} message | |
*/ | |
function _message(message, type) { | |
if(type === 'success') { | |
console.log(chalk.green(message)); | |
} else { | |
console.log(chalk.blue(message)); | |
} | |
} | |
/** | |
* 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 clientSecret = credentials.installed.client_secret; | |
const clientId = credentials.installed.client_id; | |
const redirectUrl = credentials.installed.redirect_uris[0]; | |
const auth = new googleAuth(); | |
const oauth2Client = new auth.OAuth2(clientId, clientSecret, redirectUrl); | |
// Check if we have previously stored a token. | |
fs.readFile(TOKEN_PATH, function(err, token) { | |
if (err) { | |
getNewToken(oauth2Client, callback); | |
} else { | |
oauth2Client.credentials = 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 to call with the authorized | |
* client. | |
*/ | |
function getNewToken(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: ', function(code) { | |
rl.close(); | |
oauth2Client.getToken(code, function(err, token) { | |
if (err) { | |
console.log('Error while trying to retrieve access token', err); | |
return; | |
} | |
oauth2Client.credentials = token; | |
storeToken(token); | |
callback(oauth2Client); | |
}); | |
}); | |
} | |
/** | |
* Store token to disk be used in later program executions. | |
* | |
* @param {Object} token The token to store to disk. | |
*/ | |
function storeToken(token) { | |
try { | |
fs.mkdirSync(TOKEN_DIR); | |
} catch (err) { | |
if (err.code !== 'EEXIST') { | |
throw err; | |
} | |
} | |
fs.writeFile(TOKEN_PATH, JSON.stringify(token)); | |
console.log('Token stored to ' + TOKEN_PATH); | |
} | |
/** | |
* Remove existing images. | |
*/ | |
function removeImages() { | |
_message('Removing existing images...'); | |
del.sync([IMAGE_DIR + '**/*.{jpeg,jpg,JPG}'], { | |
force: true | |
}); | |
} | |
/** | |
* Download all of the shared images. | |
* | |
* @param {google.auth.OAuth2} auth An authorized OAuth2 client. | |
*/ | |
function downloadImages(auth) { | |
function downloadIt(file){ | |
} | |
const gDrive = google.drive({ | |
version: 'v3', | |
auth: auth | |
}); | |
gDrive.files.list({ | |
q: 'sharedWithMe = true and mimeType = "image/jpeg"' | |
}, (err, resp) => { | |
if(err) { | |
console.log('The API returned an error: ' + err); | |
return; | |
} | |
if(!resp.files.length) { | |
console.error('No files found.'); | |
} else { | |
// Remove existing images. | |
// removeImages(); | |
async.each(resp.files, (file, cb) => { | |
if(fs.existsSync(IMAGE_DIR + file.name)) { | |
return; | |
} | |
gDrive.files.get({ | |
fileId: file.id | |
}) | |
.on('end', () => { | |
console.log(chalk.green(file.name + ' successfully downloaded.')); | |
cb(); | |
}) | |
.on('error', (err) => { | |
console.log(err); | |
cb(); | |
}) | |
.pipe(fs.createWriteStream(IMAGE_DIR + file.name)); | |
}); | |
} | |
}); | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment