automux 0.0.1

data/lib/automux/core/tmux/pane.rb

@@ -0,0 +1,19 @@
+module Automux
+  module Core
+    module Tmux
+      class Pane < Base
+        attr_reader :command, :window
+        private :window
+
+        def initialize(window, command)
+          @window = window
+          @command = command
+        end
+
+        def index
+          window.panes.find_index(self)
+        end
+      end
+    end
+  end
+end

data/lib/automux/core/tmux/session.rb

@@ -0,0 +1,137 @@
+module Automux
+  module Core
+    module Tmux
+      class Session < Base
+        include Support::HooksHelper
+        include Support::OptionsHelper
+
+        attr_reader :data, :name, :root, :data_windows, :flags, :base_index
+        dup_attr_reader :windows, :hooks, :options
+        private :data, :data_windows
+
+        def initialize(blueprint_data)
+          @data = blueprint_data
+          @name = data['name']
+          @root = data['root'] || '.'
+          @data_windows = data['windows'] || []
+          @data_hooks = data['hooks'] || []
+          @windows = []
+          @flags = data['flags']
+          @hooks = []
+          @data_options = data['options'] || []
+          @options = []
+        end
+
+        def start_server
+          %[tmux start-server]
+        end
+
+        def new_session
+          %[#{ tmux_with_flags } new-session -d -s #{ name }]
+        end
+
+        def set_option(option)
+          %[tmux set-option #{ option.name } '#{ option.value }']
+        end
+
+        def new_window(window)
+          %[tmux new-window -t #{ name }:#{ window.index } 2> /dev/null]
+        end
+
+        def rename_window(window, window_name = window.name)
+          %[tmux rename-window -t #{ name }:#{ window.index } #{ window_name }]
+        end
+
+        def send_keys(identifier, command)
+          window = get_window(identifier)
+          %[tmux send-keys -t #{ name }:#{ window.index } "#{ command }" C-m]
+        end
+
+        def attach_session
+          %[#{ tmux_with_flags } attach-session -t #{ name }]
+        end
+
+        def select_layout(window)
+          %[tmux select-layout -t #{ name }:#{ window.index } #{ window.layout }]
+        end
+
+        def create_pane
+          %[tmux split-window]
+        end
+
+        def get_binding
+          binding
+        end
+
+        def setup
+          setup_options
+          setup_base_index
+          setup_windows
+          setup_hooks
+        end
+
+        def window_indexes
+          @windows.map(&:index).compact
+        end
+
+        def next_available_window_index
+          n = base_index + number_of_windows
+          (Array(base_index..n) - window_indexes).first
+        end
+
+        private ###
+
+        def tmux_with_flags
+          %[tmux #{ flags }].strip
+        end
+
+        def number_of_windows
+          @windows.length
+        end
+
+        def add_window(window)
+          @windows << window
+        end
+
+        # When multiple windows have been assigned the same index by the user, the additional window indexes will be removed.
+        def remove_duplicate_indexes(original_data)
+          non_indexed_data = original_data.select { |h| h['index'].to_s.empty? }
+          indexed_data = original_data - non_indexed_data
+
+          uniq_indexed_data = indexed_data.uniq { |h| h['index'] }
+          conflicting_data = indexed_data - uniq_indexed_data
+          removed_index_data = conflicting_data.each { |h| h.delete('index') }
+
+          non_indexed_data + uniq_indexed_data + removed_index_data
+        end
+
+        def add_windows(windows_data)
+          windows_data.each do |window_data|
+            window = Automux::Core::Tmux::Window.new(self, window_data)
+            add_window(window) if window.opted_in?
+          end
+        end
+
+        def setup_windows
+          windows_data = remove_duplicate_indexes(data_windows)
+          add_windows(windows_data)
+          @windows.each(&:update_index)
+          @windows.each(&:setup)
+        end
+
+        def setup_base_index
+          @base_index = 0
+          if option = options.find { |option| option.name == 'base-index' }
+            @base_index = option.value.to_i
+          end
+        end
+
+        def get_window(identifier)
+          return identifier if identifier.is_a?(Window)
+
+          @windows.find { |window| [window.index, window.name].include?(identifier) }
+        end
+      end
+    end
+  end
+end

data/lib/automux/core/tmux/window.rb

@@ -0,0 +1,76 @@
+module Automux
+  module Core
+    module Tmux
+      class Window < Base
+        include Support::HooksHelper
+        include Support::OptionsHelper
+
+        attr_reader :data, :session, :index, :root
+        dup_attr_reader :panes, :hooks, :options
+        private :data, :session
+
+        def initialize(session, data)
+          @session = session
+          @data = data
+          @opt = data['opt']
+          @index = data['index']
+          @root = data['root']
+          @data_hooks = data['hooks'] || []
+          @hooks = []
+          @data_options = data['options'] || []
+          @options = []
+          @panes = []
+        end
+
+        def set_option(option)
+          %[tmux set-window-option -t #{ session.name }:#{ index } #{ option.name } '#{ option.value }']
+        end
+
+        def name
+          data['name']
+        end
+
+        def setup
+          setup_options
+          setup_panes
+          setup_hooks
+        end
+
+        def has_panes?
+          !@panes.nil?
+        end
+
+        def layout
+          data['layout']
+        end
+
+        def update_index
+          @index ||= session.next_available_window_index
+        end
+
+        def opted_in?
+          return true if @opt.nil?
+
+          @opt
+        end
+
+        def change_root_command
+          %[cd #{ root }]
+        end
+
+        def get_binding
+          binding
+        end
+
+        private ###
+
+        def setup_panes
+          [data['panes']].flatten.each do |command|
+            pane = Automux::Core::Tmux::Pane.new(self, command)
+            @panes << pane
+          end
+        end
+      end
+    end
+  end
+end

data/lib/automux/initializers.rb

@@ -0,0 +1,2 @@
+require 'automux/initializers/custom_hooks'
+require 'automux/initializers/setup_caches'

data/lib/automux/initializers/custom_hooks.rb

@@ -0,0 +1,4 @@
+# Call after_inherited hooks for all controller classes inheriting from Base.
+Automux::Controller.base_inheriting_classes.each do |class_name|
+  Automux::Controller::Base.after_inherited(Automux::Controller.const_get(class_name))
+end

data/lib/automux/initializers/setup_caches.rb

@@ -0,0 +1,3 @@
+Automux::Cache.module_constants.each do |constant_name|
+  Automux::Cache.const_get(constant_name).setup
+end

data/lib/automux/library.rb

@@ -0,0 +1,5 @@
+require 'automux/library/mini_erb'
+require 'automux/library/yaml_parser'
+
+module Automux::Library
+end

data/lib/automux/library/mini_erb.rb

@@ -0,0 +1,22 @@
+require 'erb'
+
+module Automux
+  module Library
+    class MiniErb < ERB
+
+      def initialize(template)
+        modified_template = erb_parseable_template(template)
+        super(modified_template, nil, '%<>')
+      end
+
+      private ###
+
+      # - [1, 2, 3]      =>     % [1, 2, 3]
+      # = [1, 2, 3]      =>     <%= [1, 2, 3] %>
+      def erb_parseable_template(string)
+        string.gsub(/^\s*-(.+)$/, '%\1').
+               gsub(/^\s*=(.+)$/, '<%=\1 %> ')
+      end
+    end
+  end
+end

data/lib/automux/library/yaml_parser.rb

@@ -0,0 +1,41 @@
+require 'yaml'
+require 'optparse'
+
+module Automux
+  module Library
+    class YamlParser
+      class << self
+        def load_file(file)
+          data_string = File.read(file)
+          opts_replaced_string = replace_opts_with_user_input(data_string)
+          YAML.load(opts_replaced_string)
+        end
+
+        private ###
+
+        # Scan blueprint for patterns like "-\w" or '-\w' to get a options list.
+        # The options list is used by OptionsParser to read the options from commandline.
+        def replace_opts_with_user_input(string)
+          blueprint_opts = string.scan(/['|"]-(\w:?)['|"]/m).flatten
+          user_input_values = get_options(blueprint_opts)
+          replace_opts(string, user_input_values)
+        end
+
+        # Replace patterns like "-\w" with matching commandline option.
+        def replace_opts(string, values)
+          values.each do |k, v|
+            # Interpolate #{ v } - to substitute literal booleans.
+            string.gsub!(/['|"]-(#{ k }):?['|"]/m, "#{ v }")
+          end
+
+          string
+        end
+
+        def get_options(opts)
+          ARGV.shift while ARGV[0].to_s.match(/^\w/)
+          ARGV.getopts(*opts)
+        end
+      end
+    end
+  end
+end

data/lib/automux/paths.rb

@@ -0,0 +1,41 @@
+module Automux
+  module Paths
+    extend self
+
+    def default_blueprint
+      File.join(data, 'blueprints', 'default.yml')
+    end
+
+    def default_recipe
+      File.join(data, 'recipes', 'default.sh.erb')
+    end
+
+    def data
+      File.join(root, 'data/automux')
+    end
+
+    def root
+      File.expand_path('../../../', __FILE__)
+    end
+
+    def views
+      'lib/automux/views'
+    end
+
+    def user_assets
+      File.join(ENV['HOME'], '.automux')
+    end
+
+    %w(blueprints recipes).each do |name|
+      define_method(name) do
+        Dir[File.join(user_assets, name, '*')]
+      end
+    end
+
+    %w(blueprints recipes).each do |name|
+      define_method("#{ name }_container") do
+        File.join(user_assets, name)
+      end
+    end
+  end
+end

data/lib/automux/views/blueprints/copy.sh.erb

@@ -0,0 +1,2 @@
+cp <%= source_blueprint.path %> <%= path %>
+echo <%= File.basename(source_blueprint.path) %> copied

data/lib/automux/views/blueprints/create.sh.erb

@@ -0,0 +1,2 @@
+cp <%= source_blueprint.path %> <%= path %>
+$EDITOR <%= path %>

data/lib/automux/views/blueprints/edit.sh.erb

@@ -0,0 +1 @@
+$EDITOR <%= path %>

data/lib/automux/views/messages/error.sh.erb

@@ -0,0 +1,3 @@
+printf "\033[0;31m"
+echo <%= message %>
+printf "\033[0m"

data/lib/automux/views/setup/clone_defaults.sh.erb

@@ -0,0 +1,21 @@
+setup() {
+  mkdir -p <%= Automux::Paths.blueprints_container %>
+  mkdir -p <%= Automux::Paths.recipes_container %>
+
+  cp <%= Automux::Paths.default_blueprint %> <%= Automux::Paths.blueprints_container %>
+  cp <%= Automux::Paths.default_recipe %> <%= Automux::Paths.recipes_container %>
+
+  echo default files copied
+}
+
+if [ -d <%= Automux::Paths.blueprints_container %> ]; then
+  read -r -p "This will replace the default blueprint and recipe under ~/.automux. Continue? [Y/n] " response
+
+  case $response in
+    [yY][eE][sS]|[yY])
+      setup
+    ;;
+  esac
+else
+  setup
+fi

data/test/session_test.rb

@@ -0,0 +1,16 @@
+require 'support/test_helper'
+
+describe 'Session' do
+  it "should assign window indexes" do
+
+    windows_data = [nil, nil, 1, 0].map do |i|
+      HashFactory.build(:window, index: i)
+    end
+    data = HashFactory.build(:session, windows: windows_data)
+
+    session = Automux::Core::Tmux::Session.new(data)
+    session.setup_windows_and_hooks
+
+    session.window_indexes.must_equal [2, 3, 1, 0]
+  end
+end

data/test/support/factories.rb

@@ -0,0 +1,13 @@
+HashFactory.define :session do
+  {
+    project_name: :test,
+    project_root: '~/',
+  }
+end
+
+HashFactory.define :window do
+  {
+    name: "name",
+    panes: ['ls']
+  }
+end

data/test/support/hash_factory.rb

@@ -0,0 +1,19 @@
+class HashFactory
+  @@factories = {}
+
+  class << self
+    def define(name, &block)
+      @@factories[name.to_s] = block
+    end
+
+    def build(name, attrs = {})
+      defaults = stringify(@@factories[name.to_s].call)
+      attributes = stringify(attrs)
+      defaults.merge(attributes)
+    end
+
+    def stringify(data)
+      Hash[data.map { |k, v| [k.to_s, v] }]
+    end
+  end
+end