pulsar/packages/dev-live-reload/lib/ui-watcher.js

const { CompositeDisposable } = require('atom');

const BaseThemeWatcher = require('./base-theme-watcher');
const PackageWatcher = require('./package-watcher');

module.exports = class UIWatcher {
  constructor() {
    this.subscriptions = new CompositeDisposable();
    this.reloadAll = this.reloadAll.bind(this);
    this.watchers = [];
    this.baseTheme = this.createWatcher(new BaseThemeWatcher());
    this.watchPackages();
  }

  watchPackages() {
    this.watchedThemes = new Map();
    this.watchedPackages = new Map();
    for (const theme of atom.themes.getActiveThemes()) {
      this.watchTheme(theme);
    }
    for (const pack of atom.packages.getActivePackages()) {
      this.watchPackage(pack);
    }
    this.watchForPackageChanges();
  }

  watchForPackageChanges() {
    this.subscriptions.add(
      atom.themes.onDidChangeActiveThemes(() => {
        // We need to destroy all theme watchers as all theme packages are destroyed
        // when a theme changes.
        for (const theme of this.watchedThemes.values()) {
          theme.destroy();
        }

        this.watchedThemes.clear();

        // Rewatch everything!
        for (const theme of atom.themes.getActiveThemes()) {
          this.watchTheme(theme);
        }
      })
    );

    this.subscriptions.add(
      atom.packages.onDidActivatePackage(pack => this.watchPackage(pack))
    );

    this.subscriptions.add(
      atom.packages.onDidDeactivatePackage(pack => {
        // This only handles packages - onDidChangeActiveThemes handles themes
        const watcher = this.watchedPackages.get(pack.name);
        if (watcher) watcher.destroy();
        this.watchedPackages.delete(pack.name);
      })
    );
  }

  watchTheme(theme) {
    if (PackageWatcher.supportsPackage(theme, 'theme')) {
      this.watchedThemes.set(
        theme.name,
        this.createWatcher(new PackageWatcher(theme))
      );
    }
  }

  watchPackage(pack) {
    if (PackageWatcher.supportsPackage(pack, 'atom')) {
      this.watchedPackages.set(
        pack.name,
        this.createWatcher(new PackageWatcher(pack))
      );
    }
  }

  createWatcher(watcher) {
    watcher.onDidChangeGlobals(() => {
      console.log('Global changed, reloading all styles');
      this.reloadAll();
    });
    watcher.onDidDestroy(() =>
      this.watchers.splice(this.watchers.indexOf(watcher), 1)
    );
    this.watchers.push(watcher);
    return watcher;
  }

  reloadAll() {
    this.baseTheme.loadAllStylesheets();
    for (const pack of atom.packages.getActivePackages()) {
      if (PackageWatcher.supportsPackage(pack, 'atom')) {
        pack.reloadStylesheets();
      }
    }

    for (const theme of atom.themes.getActiveThemes()) {
      if (PackageWatcher.supportsPackage(theme, 'theme')) {
        theme.reloadStylesheets();
      }
    }
  }

  destroy() {
    this.subscriptions.dispose();
    this.baseTheme.destroy();
    for (const pack of this.watchedPackages.values()) {
      pack.destroy();
    }
    for (const theme of this.watchedThemes.values()) {
      theme.destroy();
    }
  }
};