const needle = require('needle');
const fs = require('fs');
const path = require('path');
const [, scriptName, filename, uploadURL] = process.argv;
if (!filename || !uploadURL) {
console.error(`Uso: node ${path.basename(scriptName)} nombre_archivo url_carga`);
process.exit(-1);
}
async function uploadFile(file, url) {
// rangeEnd es el índice del último byte del archivo, es decir, el número de bytes del archivo
const rangeEnd = (await fs.promises.stat(file)).size;
let options = {
headers: {
'Content-Range': `bytes */${rangeEnd}`,
},
};
const response = await needle('put', url, null, options);
switch (response.statusCode) {
case 200:
case 201:
console.log('Carga completada');
return;
case 308:
return resumeUpload(response, file, url);
default:
console.log('Se recibió un código de respuesta inesperado: ', response.statusCode);
return;
}
}
async function resumeUpload(response, file, url) {
console.log('Carga no completada, reanudando');
if (response.headers.range) {
let resumeOffset = Number(response.headers.range.split('-')[1]) + 1;
let options = {
headers: {
'Content-Range': `bytes ${resumeOffset}-${rangeEnd-1}/${rangeEnd}`,
'Content-Length': `${rangeEnd-resumeOffset}`,
},
};
let readStream = fs.createReadStream(file, {start: resumeOffset});
return needle('put', url, readStream, options);
} else {
console.log('Iniciando carga');
let options = {
headers: {
'Content-Type': 'text/plain'
}
};
let readStream = fs.createReadStream(file);
return needle('put', url, readStream, options);
}
}
// Solicitar URL de sesión reanudable
async function requestResumableSession(url) {
const options = {
headers: {
'Content-Type': 'text/plain',
'Content-Length': '0',
'x-goog-resumable': 'start',
},
};
const res = await needle('post', url, null, options);
if (res.statusCode === 201) {
const resumableSessionURL = res.headers['location'];
console.log('Iniciando carga en: ', resumableSessionURL);
await uploadFile(filename, resumableSessionURL);
} else {
console.log('No se pudo crear la URI de sesión reanudable');
}
}
requestResumableSession(uploadURL).then(result => console.log('Carga completada'));
