const fs = require('fs'); const readline = require('readline'); const {google} = require('googleapis'); var FILE = require('./file'); var GOOGLEAPI = function() {}; var auth = null; GOOGLEAPI.prototype.init = function() { // If modifying these scopes, delete token.json. const SCOPES = [ 'https://www.googleapis.com/auth/drive', 'https://www.googleapis.com/auth/drive.file', 'https://www.googleapis.com/auth/drive.appdata', 'https://www.googleapis.com/auth/drive.apps.readonly', 'https://www.googleapis.com/auth/drive.readonly', 'https://www.googleapis.com/auth/drive.metadata.readonly', 'https://www.googleapis.com/auth/drive.metadata', 'https://www.googleapis.com/auth/drive.photos.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) console.error(err); // console.log('Token stored to', TOKEN_PATH); }); callback(oAuth2Client); }); }); } /** * Lists the names and IDs of up to 10 files. * @param {google.auth.OAuth2} auth An authorized OAuth2 client. */ function listFiles(a) { auth = a; // console.log('auth', auth); const drive = google.drive({version: 'v3', auth}); drive.files.list( { pageSize: 10, fields: 'nextPageToken, files(id, name, mimeType, size)' }, (err, res) => { if (err) return console.log('The API returned an error: ' + err); const files = res.data.files; if (files.length) { console.log('Files:'); files.map(file => { console.log(`${file.name} (${file.id})`); }); } else { console.log('No files found.'); } } ); } }; GOOGLEAPI.prototype.listGoogleDrive = function(callback) { console.log('listGoogleDrive'); const drive = google.drive({version: 'v3', auth}); drive.files.list( { pageSize: 10, fields: 'nextPageToken, files(id, name, mimeType, size)' }, (err, res) => { if (err) return console.log('The API returned an error: ' + err); var files = res.data.files; // console.log('Files:', files); files.map(f => { f.filename = f.name; return f; }); files = files.filter(f => !f.mimeType.includes('vnd.google-apps.document')) callback(files); } ); }; GOOGLEAPI.prototype.uploadGoogleDrive = function( file_name, type, file_url, callback ) { const drive = google.drive({version: 'v3', auth}); var fileMetadata = { name: file_name }; var media = { mimeType: type, body: fs.createReadStream(file_url) }; console.log('uploadGoogleDrive...'); drive.files.create( { resource: fileMetadata, media: media, fields: 'id' }, function(err, file) { if (err) { // Handle error console.error('uploadGoogleDrive', err); } else { console.log('uploadGoogleDrive => file', file); callback(file.data.id); } } ); }; GOOGLEAPI.prototype.downloadGoogleDrive = function(file, callback) { console.log('downloadGoogleDrive => file', file); let drive = google.drive({version: 'v3', auth}); let file_url = 'files\\' + file.filename; var dest = fs.createWriteStream(file_url); drive.files.get( {fileId: file.id, alt: 'media'}, {responseType: 'stream'}, function(err, res) { if (err) console.error('drive.files.get', err); res.data .on('end', () => { console.log('Done'); callback(file_url); }) .on('error', err => { console.log('Error', err); }) .pipe(dest); } ); }; module.exports = new GOOGLEAPI();