unison-watch/lib/unison/watcher.rb

require 'forwardable'

module Unison
  class Watcher < Qt::Application
    TRANSFER_LOG = '~/unison.log'
    SYNC_CHECK_TIME = 0.05

    extend Forwardable

    def initialize(profiles, *args)
      super(*args)

      @config = Unison::Config.ensure(File.expand_path('~/.unison/watch.yml'))
      @queue = Atomic.new([])

      @sync_now = false
      @exiting = false
      @active = true

      @config.on_update { @remote_sync_check = time_between_checks }
    end

    def time_between_checks
      @config.time_between_checks * 20
    end

    def <<(dirs)
      @queue.update { |q| q += [ dirs ].flatten ; q }
    end

    def processed_profiles
      @processed_profiles ||= Unison::Profile.process(@config.profiles)
    end

    def watch
      Unison::FilesystemWatcher.new(paths_to_watch, self)
    end

    def paths_to_watch
      processed_profiles.collect(&:paths_with_local_root).flatten
    end

    def fileview
      @fileview ||= Unison::UI::FileView.new(File.expand_path(TRANSFER_LOG))
    end

    def menu
      return @menu if @menu

      @menu = Unison::UI::Menu.new

      @menu.on(:sync_now) { @sync_now = true }
      @menu.on(:toggle_status) { toggle_status }
      @menu.on(:view_log) { fileview.show }
      @menu.on(:quit) { @exiting = true }
      @menu.on(:preferences) { preferences.show }
      @menu.generate

      @menu
    end

    def update_ui
      @icon.current_icon = @current_icon
      @menu.status_text = @current_text

      processEvents
    end

    def ui
      @icon = Unison::UI::Icon.new(menu, self, @config.profiles, File.join(Unison.root, 'assets'))
      @config.on_update { @icon.profiles = @config.profiles }

      @current_icon = 'idle'

      toggle_status true
      @prior_text = nil

      @icons = {}

      @remote_sync_check = time_between_checks

      while !@exiting
        check

        update_ui

        sleep SYNC_CHECK_TIME
      end
    end

    def start
      if !@config.active_profiles?
        preferences.show
      end

      watch
      ui
    end

    def preferences
      @preferences ||= Unison::UI::Preferences.new(@config)
    end

    def show_working
      index = 0
      while true do
        @current_icon = "working-#{index + 1}"
        fileview.read!

        update_ui

        break if @done or @exiting

        sleep 0.25
        index = (index + 1) % 2
      end

      @current_icon = current_icon
      fileview.read!
    end

    def check
      begin
        if @active && (@queue.value.length > 0 || @remote_sync_check == 0 || @sync_now)
          dir = nil

          @current_text = "Syncing..."

          @done = false

          Unison::Bridge.run(@config, TRANSFER_LOG) { @done = true }

          show_working

          @remote_sync_check = time_between_checks
          @sync_now = false
          @queue.update { [] }
        end
      rescue => e
        puts e.message
        puts e.backtrace.join("\n")
        exit 1
      end

      @remote_sync_check -= 1

      if @active
        @current_text = "Next check in #{sprintf("%.0d", SYNC_CHECK_TIME * @remote_sync_check)} secs."
      else
        @current_text = "Syncing paused."

        @remote_sync_check = time_between_checks
      end
    end

    def toggle_status(set = nil)
      @active = set || !@active

      @menu.active_status_text = @active ? "Pause syncing" : "Resume syncing"
      @current_icon = current_icon
    end

    def current_icon
      @active ? 'idle' : 'paused'
    end
  end
end