servicio_social_api/server/helper/drive.js

require('../config/config');
const fs = require('fs');
const { google } = require('googleapis');
const drive = google.drive('v3');
const credenciales = JSON.parse(fs.readFileSync('./cred.json', 'utf8'));
const jwtClient = new google.auth.JWT(
  credenciales.client_email,
  null,
  credenciales.private_key,
  ['https://www.googleapis.com/auth/drive']
);

const Servicio = require(`../db/tablas/Servicio`);
const Usuario = require(`../db/tablas/Usuario`);

const list = (pageToken) => {
  return new Promise((resolve, reject) => {
    jwtClient.authorize((err, tokens) => {
      if (err) reject('Error de autorización: ' + err);
      drive.files.list(
        {
          auth: jwtClient,
          corpora: 'user',
          supportsTeamDrives: true,
          includeTeamDriveItems: true,
          spaces: 'drive',
          pageSize: 1000,
          pageToken,
          // fields: '*',
          orderBy: 'name',
        },
        (err, resp) => {
          if (err) reject('Ha habido un error en la petición: ' + err);
          resolve(resp.data);
        }
      );
    });
  });
};

// Borrar archivo
const deleteFile = (fileId) => {
  if (fileId === process.env.CARPETA)
    throw new Error('No se puede eliminar este archivo/carpeta.');
  return new Promise((resolve, reject) => {
    jwtClient.authorize(async (err, tokens) => {
      if (err) reject(err);
      drive.files.delete(
        {
          fileId,
          auth: jwtClient,
        },
        (err, resp) => {
          if (err) reject(err);
          resolve(resp);
        }
      );
    });
  });
};

// SUBIR ARCHIVO
const uploadFile = (path, name, mimeType, parent) => {
  return new Promise((resolve, reject) => {
    const requestBody = {
      name,
      parents: [parent],
    };
    const media = {
      mimeType,
      body: fs.createReadStream(`${path}`),
    };

    jwtClient.authorize(function (err, tokens) {
      if (err) reject('Error de autorización: ' + err);
      drive.files.create(
        {
          media,
          requestBody,
          fields: 'id',
          auth: jwtClient,
        },
        (err, file) => {
          if (err) reject(err);
          fs.unlinkSync(path);
          resolve(file.data.id);
        }
      );
    });
  });
};

// CREAR CARPETA
const mkDir = (name) => {
  return new Promise((resolve, reject) => {
    const requestBody = {
      name,
      mimeType: 'application/vnd.google-apps.folder',
      parents: [process.env.CARPETA],
    };

    jwtClient.authorize((err, tokens) => {
      if (err) reject('Error de autorización: ' + err);
      drive.files.create(
        {
          fields: 'id',
          auth: jwtClient,
          requestBody,
        },
        (err, res) => {
          if (err) reject(err);
          resolve(res.data.id);
        }
      );
    });
  });
};

// Update File
const update = (fileId, addParents, removeParents) => {
  return new Promise((resolve, reject) => {
    jwtClient.authorize((err, tokens) => {
      if (err) reject('Error de autorización: ' + err);
      drive.files.update(
        { auth: jwtClient, fileId, addParents, removeParents },
        (err, resp) => {
          if (err) reject('Ha habido un error en la petición: ' + err);
          resolve(resp.data);
        }
      );
    });
  });
};

const folder = async (numeroCuenta) => {
  let pageToken = '';

  do {
    let res = await list(pageToken);

    for (let i = 0; i < res.files.length; i++)
      if (
        res.files[i].name === numeroCuenta &&
        res.files[i].mimeType === 'application/vnd.google-apps.folder'
      )
        return res.files[i].id;
    pageToken = res.nextPageToken ? res.nextPageToken : '';
  } while (pageToken);
  return mkDir(numeroCuenta);
};

// const borrarTodo = () => {
//   return list().then(async (res) => {
//     for (let i = 0; i < res.files.length; i++)
//       if (res.files[i].id !== process.env.CARPETA)
//         await deleteFile(res.files[i].id);
//   });
// };

const findAll = async () => {
  const folders = [];
  const pdfs = [];
  let pageToken = '';

  do {
    await list(pageToken).then((res) => {
      for (let i = 0; i < res.files.length; i++) {
        if (res.files[i].id !== process.env.CARPETA) {
          if (res.files[i].mimeType === 'application/vnd.google-apps.folder')
            folders.push(res.files[i].id);
          if (res.files[i].mimeType === 'application/pdf')
            pdfs.push(res.files[i].id);
        }
      }
      pageToken = res.nextPageToken ? res.nextPageToken : '';
    });
  } while (pageToken);
  return { folders, pdfs };
};

const mkDirInsideOtherDir = (name, ruta) => {
  return new Promise((resolve, reject) => {
    const requestBody = {
      name,
      mimeType: 'application/vnd.google-apps.folder',
      parents: [ruta],
    };

    jwtClient.authorize((err, tokens) => {
      if (err) reject('Error de autorización: ' + err);
      drive.files.create(
        {
          fields: 'id',
          auth: jwtClient,
          requestBody,
        },
        (err, res) => {
          if (err) reject(err);
          resolve(res.data.id);
        }
      );
    });
  });
};

const scriptCambioDeCarpeta = async () => {
  console.log('Entrando a scriptCambioDeCarpeta');

  const servicios = await Servicio.findAll({
    where: { idStatus: [1, 2, 3, 4, 5, 7, 8, 9] },
  });

  const idUsuarios = servicios.map((servicio) => servicio.idUsuario); //aqui se traen todos los idUsuario de los servicios entre 1-5

  const usuarios = await Usuario.findAll({
    where: { idUsuario: idUsuarios },
  });

  const parentFolderId = await mkDir('Servicios [1 - 5]'); //se crea la carpeta donde se van a guardar los alumnos y se guarda el id de la carpeta

  for (let i = 0; i < usuarios.length; i++) {
    //por cada cuenta
    const carpetaCreada = await mkDirInsideOtherDir(
      `${usuarios[i].usuario}`,
      parentFolderId
    ); //se crea una crapeta dentro de la carpeta cuyo id guardamos antes y guardamos el id de la carpeta creada

    await Servicio.update(
      { carpeta: carpetaCreada },
      { where: { idUsuario: usuarios[i].idUsuario } }
    ); // se actualiza el id de la carpeta por la que se acaba de crear
    console.log(
      i,
      ' - ',
      '[',
      usuarios[i].idUsuario,
      ']',
      '[',
      usuarios[i].usuario,
      '] : [',
      carpetaCreada,
      ']'
    );
  }
  console.log('Saliendo de scriptCambioDeCarpeta');
};

//scriptCambioDeCarpeta();

module.exports = {
  folder,
  uploadFile,
  deleteFile,
  mkDir,
};