app.js

#!/usr/bin/env node

const fs = require('fs');
const childProcess = require('child_process');

const axios = require('axios');
const yargs = require('yargs');

const argv = yargs
  .options({
    n: {
      alias: 'nasa',
      describe: 'Choose NASA photo',
      type: 'boolean'
    },
    N: {
      alias: 'ng',
      describe: 'Choose National Geographic photo',
      type: 'boolean'
    },
    u: {
      alias: 'unsplash',
      describe: 'Choose Unsplash photo',
      type: 'boolean'
    },
    b: {
      alias: 'before',
      describe: 'Bing date option (# of days before)',
      type: 'number',
      default: 0,
      choices: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]
    },
    l: {
      alias: 'locale',
      describe: 'Bing locale option',
      type: 'string',
      default: 'en-US',
      choices: ['ar-XA', 'bg-BG', 'cs-CZ', 'da-DK', 'de-AT', 'de-CH', 'de-DE', 'el-GR', 'en-AU', 'en-CA', 'en-GB', 'en-ID', 'en-IE', 'en-IN', 'en-MY', 'en-NZ', 'en-PH', 'en-SG', 'en-US', 'en-XA', 'en-ZA', 'es-AR', 'es-CL', 'es-ES', 'es-MX', 'es-US', 'es-XL', 'et-EE', 'fi-FI', 'fr-BE', 'fr-CA', 'fr-CH', 'fr-FR', 'he-IL', 'hr-HR', 'hu-HU', 'it-IT', 'ja-JP', 'ko-KR', 'lt-LT', 'lv-LV', 'nb-NO', 'nl-BE', 'nl-NL', 'pl-PL', 'pt-BR', 'pt-PT', 'ro-RO', 'ru-RU', 'sk-SK', 'sl-SL', 'sv-SE', 'th-TH', 'tr-TR', 'uk-UA', 'zh-CN', 'zh-HK', 'zh-TW']
    },
    r: {
      alias: 'resolution',
      describe: 'Bing & Unsplash resolution option',
      type: 'string',
      default: '1920x1080',
      choices: ['800x600', '1024x768', '1280x720', '1280x768', '1366x768', '1920x1080', '1920x1200']
    },
    k: {
      alias: 'key',
      describe: 'NASA API key',
      type: 'string',
      default: 'DEMO_KEY'
    },
    d: {
      alias: 'date',
      describe: 'NASA & National Geographic date option (YYYY-mm-dd)',
      type: 'string',
      default: new Date().toISOString().slice(0, 10)
    },
    R: {
      alias: 'random',
      describe: 'NASA & National Geographic & Unsplash random option',
      type: 'boolean'
    },
    w: {
      alias: 'wallpaper',
      describe: 'Don\'t set downloaded photo as wallpaper (can be used for bulk downloading)',
      type: 'boolean'
    },
    o: {
      alias: 'option',
      describe: 'Linux background display option',
      type: 'string',
      default: 'zoom',
      choices: ['none', 'wallpaper', 'centered', 'scaled', 'stretched', 'zoom', 'spanned']
    }
  })
  .help()
  .alias('help', 'h')
  .version()
  .alias('version', 'v')
  .check(argv => {
    return argv.d.length === 10 && !isNaN(new Date(argv.d)) && new Date(argv.d).toISOString().slice(0, 10) === argv.d;
  })
  .argv;

const main = async () => {
  let searchUrl;
  if (argv.N) {
    if (argv.R) {
      const start = new Date(2009, 0, 1);
      argv.d = new Date(start.getTime() + Math.random() * (new Date().getTime() - start.getTime())).toISOString().slice(0, 10);
    }
    searchUrl = `https://www.nationalgeographic.com/photography/photo-of-the-day/_jcr_content/.gallery.${argv.d.slice(0, 7)}.json`;
  } else if (argv.n) {
    searchUrl = `https://api.nasa.gov/planetary/apod?api_key=${argv.k}&hd=True&${argv.R ? 'count=1' : `date=${argv.d}`}`;
  } else {
    let idx = 0;
    let n = 1;
    if (argv.b <= 7) {
      idx = argv.b;
      n = 1;
    } else {
      idx = 7;
      n = argv.b - 6;
    }
    searchUrl = `https://www.bing.com/HPImageArchive.aspx?format=js&idx=${idx}&n=${n}&mkt=${argv.l}`;
  }
  let photo;
  let photoUrl;
  let photoName;
  let photoDir;
  await axios(searchUrl)
    .then(res => {
      if (argv.u) {
        photoUrl = `https://source.unsplash.com/random/${argv.r}`;
        photoName = new Date().toISOString();
        if (argv.R) {
          photoName = photoName.replace(/:/g, '.');
        } else {
          photoUrl += '/daily';
          photoName = photoName.slice(0, 10);
        }
        photoName += '.jpg';
        photoDir = 'Unsplash';
      } else if (argv.N) {
        if (+argv.d.slice(8) <= res.data.items.length) {
          photo = res.data.items[res.data.items.length - +argv.d.slice(8)];
          if (photo.image) {
            photoUrl = photo.image.uri;
            photoName = `${photo.image.title}.jpg`;
          } else {
            photoUrl = photo.originalUrl;
            photoName = `${photo.title}.jpg`;
          }
          photoDir = 'National Geographic';
        } else {
          throw new Error(`There is no photo on ${argv.d}.`);
        }
      } else if (argv.n) {
        photo = argv.R ? res.data[0] : res.data;
        if (photo.hdurl) {
          photoUrl = photo.hdurl;
          photoName = photoUrl.slice(photoUrl.lastIndexOf('/') + 1);
          photoDir = 'NASA';
        } else {
          throw new Error(`There is no photo on ${argv.d}.`);
        }
      } else {
        photo = `${res.data.images[res.data.images.length - 1].urlbase}_${argv.r}.jpg`;
        photoUrl = `https://www.bing.com${photo}`;
        photoName = photo.slice(photo.indexOf('.') + 1);
        photoDir = 'Bing';
      }
    })
    .catch(err => {
      throw new Error(err);
    });
  let platform;
  let directory;
  if (process.platform === 'linux') {
    platform = 0;
    directory = '/home/' + process.env.USER;
  } else if (process.platform === 'darwin') {
    platform = 1;
    directory = '/Users/' + process.env.USER;
  } else if (process.platform === 'win32') {
    platform = 2;
    directory = '/Users/' + process.env.USERNAME;
  } else {
    throw new Error('Your platform is not supported yet.');
  }
  directory += '/Pictures/' + photoDir + '/';
  const fileName = directory + photoName;
  await fs.promises.mkdir(directory, { recursive: true })
    .catch(err => {
      throw new Error(err);
    });
  await axios({ url: photoUrl, responseType: 'stream', timeout: 20000 })
    .then(res => {
      res.data.pipe(fs.createWriteStream(fileName));
      console.log(`${photoName} is saved.`);
    })
    .catch(err => {
      throw new Error(err);
    });
  if (!argv.w) {
    let command;
    if (platform === 0) {
      command = `gsettings set org.gnome.desktop.background picture-options "${argv.o}" & gsettings set org.gnome.desktop.background picture-uri "file://${fileName}"`;
    } else if (platform === 1) {
      command = `osascript -e 'tell application "System Events" to tell every desktop to set picture to "${fileName}"'`;
    } else {
      throw new Error('Windows is not supported for wallpaper setting yet.');
    }
    childProcess.exec(command, (err, stdout, stderr) => {
      if (err) {
        throw new Error(err);
      }
    });
  }
};

main();