atom/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()
    }
  }
}