class Middleman::CoreExtensions::FileWatcher

def after_configuration
  @watcher.update_config(
    disable_watcher: app.config[:watcher_disable],
    force_polling: app.config[:watcher_force_polling],
    latency: app.config[:watcher_latency],
    wait_for_delay: app.config[:watcher_wait_for_delay]
  )
  if @original_source_dir != app.config[:source]
    @watcher.update_path(app.config[:source])
  end
  @sources.start!
  @sources.poll_once!
end

def before_configuration
  @sources.poll_once!
end

def initialize(app, config={}, &block)
  super
  # Setup source collection.
  @sources = ::Middleman::Sources.new(app)
  # Add default ignores.
  IGNORES.each do |key, value|
    @sources.ignore key, :all, value
  end
  # Watch current source.
  start_watching(app.config[:source])
end

def start_watching(dir)
  @original_source_dir = dir
  @watcher = @sources.watch :source, path: File.join(app.root, dir)
end