sqlpad/routes/download-results.js

Web app. Write SQL and visualize the results. Supports Postgres, MySQL, SQL Server, Crate, Vertica and SAP HANA.

const fs = require('fs');
const router = require('express').Router();
const Cache = require('../models/Cache.js');
const config = require('../lib/config');

router.get('/download-results/:cacheKey.csv', async function(req, res, next) {
  try {
    if (config.get('allowCsvDownload')) {
      const cache = await Cache.findOneByCacheKey(req.params.cacheKey);
      if (!cache) {
        return next(new Error('Cache not found'));
      }
      let filename = cache.queryName + '.csv';
      res.setHeader(
        'Content-disposition',
        'attachment; filename="' + encodeURIComponent(filename) + '"'
      );
      res.setHeader('Content-Type', 'text/csv');
      fs.createReadStream(cache.csvFilePath()).pipe(res);
    }
  } catch (error) {
    console.error(error);
    // TODO figure out what this sends and set manually
    return next(error);
  }
});

router.get('/download-results/:cacheKey.xlsx', async function(req, res, next) {
  try {
    if (config.get('allowCsvDownload')) {
      const cache = await Cache.findOneByCacheKey(req.params.cacheKey);
      if (!cache) {
        return next(new Error('Cache not found'));
      }
      let filename = cache.queryName + '.xlsx';
      res.setHeader(
        'Content-disposition',
        'attachment; filename="' + encodeURIComponent(filename) + '"'
      );
      res.setHeader(
        'Content-Type',
        'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet'
      );
      fs.createReadStream(cache.xlsxFilePath()).pipe(res);
    }
  } catch (error) {
    console.error(error);
    // TODO figure out what this sends and set manually
    return next(error);
  }
});

module.exports = router;