dag-amazing 0.1

data/lib/amazing/cli/helpers.rb

@@ -0,0 +1,159 @@
+# Copyright 2008 Dag Odenhall <dag.odenhall@gmail.com>
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+module Amazing
+  class CLI
+    module Helpers
+      private
+
+      def parse_options
+        @options.parse
+      end
+
+      def set_loglevel
+        begin
+          @log.level = Logger.const_get(@options[:loglevel].upcase)
+
+        rescue NameError
+          @log.error("Unsupported log level #{@options[:loglevel].inspect}")
+          @log.level = Logger::INFO
+        end
+      end
+
+      def stop_process(log=true)
+        Process.kill("SIGINT", File.read("#{ENV["HOME"]}/.amazing/pids/#{@display.display}.pid").to_i) 
+        @log.warn("Killed older process") if log
+      rescue
+      end
+
+      def load_scripts
+        scripts = @options[:include]
+
+        if @options[:autoinclude]
+          scripts << Dir["#{ENV["HOME"]}/.amazing/widgets/*"]
+        end
+
+        scripts.flatten.each do |script|
+          if File.exist?(script)
+            @log.debug("Loading script #{script.inspect}")
+
+            begin
+              Widgets.module_eval(File.read(script), script)
+
+            rescue SyntaxError => e
+              @log.error("Bad syntax in #{script} at line #{e.to_s.scan(/:(\d+)/)}")
+            end
+
+          else
+            @log.error("No such widget script #{script.inspect}")
+          end
+        end
+      end
+
+      def parse_config
+        @log.debug("Parsing configuration file")
+
+        begin
+          @config = Config.new(@options[:config])
+
+        rescue
+          @log.fatal("Unable to parse configuration file, exiting")
+
+          exit 1
+        end
+      end
+
+      def wait_for_sockets
+        @log.debug("Waiting for awesome control socket for display #{@display.display}")
+
+        begin
+          Timeout.timeout(30) do
+            sleep 1 until File.exist?("#{ENV["HOME"]}/.awesome_ctl.#{@display.display}")
+            @log.debug("Got socket for display #{@display.display}")
+          end
+
+        rescue Timeout::Error
+          @log.fatal("Socket for display #{@display.display} not created within 30 seconds, exiting")
+
+          exit 1
+        end
+      end
+
+      def save_pid
+        path = "#{ENV["HOME"]}/.amazing/pids"
+        FileUtils.makedirs(path)
+
+        File.open("#{path}/#{@display.display}.pid", "w+") do |f|
+          f.write($$)
+        end
+      end
+
+      def remove_pid
+        File.delete("#{ENV["HOME"]}/.amazing/pids/#{@display.display}.pid") rescue Errno::ENOENT
+      end
+
+      def set_traps
+        trap("SIGINT") do
+          @log.fatal("Received SIGINT, exiting")
+          remove_pid
+          exit
+        end
+      end
+
+      def update_non_interval
+        @threads << Thread.new do
+          @config[:awesome].each do |awesome|
+            awesome[:widgets].each do |widget|
+              next if widget[:interval]
+
+              @threads << Thread.new(awesome, widget) do |awesome, widget|
+                update_widget(awesome[:screen], awesome[:statusbar], widget)
+              end
+            end
+          end
+        end
+      end
+
+      def update_widget(screen, statusbar, widget, iteration=0)
+        threads = []
+        @log.debug("Updating widget #{widget[:identifier]} of type #{widget[:module]} on screen #{screen}")
+
+        begin
+          mod = Widgets.const_get(widget[:module]).new(widget.merge(:iteration => iteration))
+
+          if widget[:properties].empty?
+            threads << Thread.new(screen, statusbar, widget, mod) do |screen, statusbar, widget, mod|
+              @awesome.widget_tell(screen, statusbar, widget[:identifier], widget[:property], mod.formatize)
+            end
+          end
+
+          widget[:properties].each do |property, format|
+            threads << Thread.new(screen, statusbar, widget, property, mod, format) do |screen, statusbar, widget, property, mod, format|
+              @awesome.widget_tell(screen, statusbar, widget[:identifier], property, mod.formatize(format))
+            end
+          end
+
+        rescue WidgetError => e
+          @log.error(widget[:module]) { e.message }
+        end
+
+        threads.each {|t| t.join }
+      end
+
+      def join_threads
+        @threads.each {|t| t.join }
+      end
+    end
+  end
+end

data/lib/amazing/cli/initializers.rb

@@ -0,0 +1,60 @@
+# Copyright 2008 Dag Odenhall <dag.odenhall@gmail.com>
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+module Amazing
+  class CLI
+    module Initializers
+      private
+
+      def initialize_threads
+        @threads = []
+      end
+
+      def initialize_encoding
+        $KCODE = "utf-8"
+      end
+
+      def initialize_logger
+        @log = Logger.new(STDOUT)
+      end
+
+      def initialize_options
+        @options = Options.new(@args)
+      end
+
+      def initialize_display
+        begin
+          @display = X11::DisplayName.new
+
+        rescue X11::EmptyDisplayName => e
+          @log.warn("#{e.message}, falling back on :0")
+          @display = X11::DisplayName.new(":0")
+
+        rescue X11::InvalidDisplayName => e
+          @log.fatal("#{e.message}, exiting")
+
+          exit 1
+        end
+      end
+
+      def initialize_awesome
+        @awesome = Awesome.new(@display.display)
+      end
+
+      def initialize_exit
+        at_exit { Thread.list.each {|t| t.exit } }
+      end
+    end
+  end
+end

data/lib/amazing/config.rb

@@ -0,0 +1,66 @@
+require 'amazing/string'
+require 'amazing/config/dsl'
+require 'amazing/config/yaml'
+
+module Amazing
+  class Config < Hash
+    def initialize(config)
+      @config = config
+      case File.extname(@config)[1..-1].to_sym
+      when :rb
+        from_dsl
+      when :yml, :yaml
+        from_yaml
+      end
+    end
+
+    private
+
+    def from_dsl
+      dsl = Dsl.new(@config)
+      self[:awesome] = []
+      dsl.awesome_statusbars.each do |awesome|
+        self[:awesome] << {}
+        self[:awesome][-1][:screen] = awesome.options[:screen] || 0
+        self[:awesome][-1][:statusbar] = awesome.options[:statusbar] || "mystatusbar"
+        self[:awesome][-1][:widgets] = []
+        awesome.widgets.each do |widget|
+          self[:awesome][-1][:widgets] << {}
+          self[:awesome][-1][:widgets][-1][:identifier] = widget.identifier
+          self[:awesome][-1][:widgets][-1][:properties] = widget.properties
+          self[:awesome][-1][:widgets][-1].merge!(widget.options)
+          self[:awesome][-1][:widgets][-1][:module] ||= widget.identifier
+          self[:awesome][-1][:widgets][-1][:property] ||= "text"
+          self[:awesome][-1][:widgets][-1][:module] = self[:awesome][-1][:widgets][-1][:module].to_s.camel_case
+        end
+      end
+    end
+
+    def from_yaml
+      yaml = Yaml.new(@config)
+      self[:awesome] = []
+      yaml["awesome"].each do |awesome|
+        self[:awesome] << {}
+        self[:awesome][-1][:screen] = awesome["screen"] || 0
+        self[:awesome][-1][:statusbar] = awesome["statusbar"] || "mystatusbar"
+        self[:awesome][-1][:widgets] = []
+        awesome["widgets"].each do |widget|
+          self[:awesome][-1][:widgets] << {}
+          case widget
+          when Hash
+            self[:awesome][-1][:widgets][-1][:identifier] = widget.keys[0]
+            widget.values[0].each do |key, value|
+              self[:awesome][-1][:widgets][-1][key.to_sym] = value
+            end
+            self[:awesome][-1][:widgets][-1][:module] ||= widget.keys[0]
+          when String
+            self[:awesome][-1][:widgets][-1][:identifier] = widget
+            self[:awesome][-1][:widgets][-1][:module] = widget
+          end
+          self[:awesome][-1][:widgets][-1][:property] ||= "text"
+          self[:awesome][-1][:widgets][-1][:module] = self[:awesome][-1][:widgets][-1][:module].to_s.camel_case
+        end
+      end
+    end
+  end
+end

data/lib/amazing/config/dsl.rb

@@ -0,0 +1,32 @@
+require 'amazing/config/dsl/awesome'
+require 'amazing/numeric'
+
+module Amazing
+  class Config < Hash
+    class Dsl
+      attr_reader :awesome_statusbars
+
+      def initialize(config=nil, &block)
+        @awesome_statusbars = []
+        if config
+          config = File.expand_path(config)
+          @relative_path = File.dirname(config)
+          import(config)
+        end
+        import(&block)
+      end
+
+      def import(config=nil, &block)
+        if config
+          config = "#@relative_path/#{config}" if config[0] != ?/
+          instance_eval(File.read(config))
+        end
+        instance_eval(&block) if block
+      end
+
+      def awesome(opts={}, &block)
+        @awesome_statusbars << Awesome.new(opts, &block)
+      end
+    end
+  end
+end

data/lib/amazing/config/dsl/awesome.rb

@@ -0,0 +1,26 @@
+require 'amazing/config/dsl/awesome/widget'
+
+module Amazing
+  class Config < Hash
+    class Dsl
+      class Awesome
+        attr_reader :options
+        attr_reader :widgets
+
+        def initialize(opts={}, &block)
+          @options = opts
+          @widgets = []
+          instance_eval(&block)
+        end
+
+        def set(opts)
+          @options.merge!(opts)
+        end
+
+        def widget(identifier, opts={}, &block)
+          @widgets << Widget.new(identifier, opts, &block)
+        end
+      end
+    end
+  end
+end

data/lib/amazing/config/dsl/awesome/widget.rb

@@ -0,0 +1,27 @@
+module Amazing
+  class Config < Hash
+    class Dsl
+      class Awesome
+        class Widget
+          attr_reader :options
+          attr_reader :identifier
+          attr_reader :properties
+
+          def initialize(identifier, opts={}, &block)
+            @identifier, @options = identifier, opts
+            @properties = {}
+            instance_eval(&block) if block
+          end
+
+          def set(opts)
+            @options.merge!(opts)
+          end
+
+          def property(name, &block)
+            @properties[name] = block
+          end
+        end
+      end
+    end
+  end
+end

data/lib/amazing/config/yaml.rb

@@ -0,0 +1,11 @@
+require 'yaml'
+
+module Amazing
+  class Config < Hash
+    class Yaml < Hash
+      def initialize(config)
+        merge!(YAML.load_file(config))
+      end
+    end
+  end
+end

data/lib/amazing/helpers/pango_markup.rb

@@ -0,0 +1,26 @@
+module Amazing
+  module Helpers
+    module PangoMarkup
+      def span(text, opts)
+        attrs = opts.map {|key, value| "#{key}=#{value.to_s.inspect}" }.join(" ")
+        "<span #{attrs}>#{text}</span>"
+      end
+
+      def background(color, text)
+        span(text, :background => color)
+      end
+
+      def foreground(color, text)
+        span(text, :foreground => color)
+      end
+
+      def underline(text, style=:single)
+        span(text, :underline => style)
+      end
+
+      def bold(text, level=:bold)
+        span(text, :weight => level)
+      end
+    end
+  end
+end

data/lib/amazing/numeric.rb

@@ -0,0 +1,50 @@
+# Copyright 2008 Dag Odenhall <dag.odenhall@gmail.com>
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+class Numeric
+  def seconds
+    self
+  end
+  alias :second :seconds
+
+  def minutes
+    self * 60
+  end
+  alias :minute :minutes
+
+  def hours
+    minutes * 60
+  end
+  alias :hour :hours
+
+  def days
+    hours * 24
+  end
+  alias :day :days
+
+  def weeks
+    days * 7
+  end
+  alias :week :weeks
+
+  def months
+    weeks * 4
+  end
+  alias :month :months
+
+  def years
+    months * 12
+  end
+  alias :year :years
+end