strapi/packages/strapi/lib/core/admin.js

'use strict';

// Dependencies.
const path = require('path');
const fs = require('fs');
const URL = require('url');
const cheerio = require('cheerio');
const _ = require('lodash');

module.exports = function() {
  return new Promise((resolve, reject) => {
    try {
      const environment = this.config.environment;

      if (environment === 'test') {
        return resolve();
      }

      const sourcePath = path.resolve(this.config.appPath, 'admin', 'admin', 'build', 'index.html');

      fs.access(path.resolve(this.config.appPath, 'admin', 'admin'), err => {
        if (err && err.code !== 'ENOENT') {
          return reject(err);
        }

        // No admin.
        if (err && err.code === 'ENOENT') {
          return resolve();
        }

        // Try to access to path.
        fs.access(sourcePath, err => {
          if (err && err.code !== 'ENOENT') {
            return reject(err);
          } else if (err && err.code === 'ENOENT') {
            return resolve();
          }

          fs.readFile(sourcePath, (err, html) => {
            const $ = cheerio.load(html.toString());

            $('script').each(function() {
              if ($(this).attr('src')) {
                const parse = path.parse($(this).attr('src'));
                const url = URL.parse(_.get(strapi.config.currentEnvironment.server, 'admin.build.host', _.get(strapi.config.currentEnvironment.server, 'admin.path', '/admin')));

                $(this).attr('src', `${url.pathname.replace(/\/$/, '')}/${parse.base}`);
              }
            });

            // Remove previous and use build configurations.
            if (environment === 'production') {
              $('body').removeAttr('front');
              $('body').removeAttr('back');
            } else {
              // Update attribute with the current server configurations.
              $('body').attr('front', `${_.get(strapi.config.currentEnvironment.server, 'admin.path', '/admin')}`);
              $('body').attr('back', `/`);
            }

            if (!strapi.config.currentEnvironment.server.production) {
              fs.writeFile(sourcePath, $.html(), (err) => {
                if (err) {
                  return reject(err);
                }

                resolve();
              });
            } else {
              resolve();
            }
          });
        });
      });
    } catch (e) {
      reject(e);
    }
  });
};