lbrt 0.1.0

data/lib/lbrt/space/dsl/context/space.rb

@@ -0,0 +1,19 @@
+class Lbrt::Space::DSL::Context::Space
+  def initialize(name_or_id, &block)
+    @name_or_id = name_or_id
+    @result = {'charts' => {}}
+    instance_eval(&block)
+  end
+
+  attr_reader :result
+
+  private
+
+  def chart(chart_name_or_id, &block)
+    if @result[chart_name_or_id]
+      raise "Space `#{@name_or_id}` > Chart `#{chart_name_or_id}` is already defined"
+    end
+
+    @result['charts'][chart_name_or_id] = Lbrt::Space::DSL::Context::Space::Chart.new(@name_or_id, chart_name_or_id, &block).result
+  end
+end

data/lib/lbrt/space/dsl/context/space/chart.rb

@@ -0,0 +1,44 @@
+class Lbrt::Space::DSL::Context::Space::Chart
+  REQUIRED_ATTRIBUTES = %w(
+    type
+  )
+
+  def initialize(space_name_or_id, name_or_id, &block)
+    @space_name_or_id = space_name_or_id
+    @name_or_id = name_or_id
+    @result = {'streams' => []}
+    instance_eval(&block)
+  end
+
+  def result
+    REQUIRED_ATTRIBUTES.each do |name|
+      unless @result.has_key?(name)
+        raise "Space `#{@space_name_or_id}` > Chart `#{@name_or_id}` > `#{name}` is not defined"
+      end
+    end
+
+    @result
+  end
+
+  private
+
+  def type(value)
+    @result['type'] = value.to_s
+  end
+
+  def stream(&block)
+    @result['streams'] << Lbrt::Space::DSL::Context::Space::Chart::Stream.new(@space_name_or_id, @name_or_id, &block).result
+  end
+
+  def max(value)
+    @result['max'] = value.to_f
+  end
+
+  def min(value)
+    @result['min'] = value.to_f
+  end
+
+  def label(value)
+    @result['label'] = value.to_s
+  end
+end

data/lib/lbrt/space/dsl/context/space/chart/stream.rb

@@ -0,0 +1,48 @@
+class Lbrt::Space::DSL::Context::Space::Chart::Stream
+  REQUIRED_ATTRIBUTES = %w(
+    metric
+    type
+    source
+    group_function
+    summary_function
+  )
+
+  def initialize(space_name_or_id, chart_name_or_id, &block)
+    @space_name_or_id = space_name_or_id
+    @chart_name_or_id = chart_name_or_id
+    @result = {}
+    instance_eval(&block)
+  end
+
+  def result
+    REQUIRED_ATTRIBUTES.each do |name|
+      unless @result.has_key?(name)
+        raise "Space `#{@space_name_or_id}` > Chart `#{@chart_name_or_id}` > Stream > `#{name}` is not defined"
+      end
+    end
+
+    @result
+  end
+
+  private
+
+  def metric(value)
+    @result['metric'] = value.to_s
+  end
+
+  def type(value)
+    @result['type'] = value.to_s
+  end
+
+  def source(value)
+    @result['source'] = value.to_s
+  end
+
+  def group_function(value)
+    @result['group_function'] = value.to_s
+  end
+
+  def summary_function(value)
+    @result['summary_function'] = value.to_s
+  end
+end

data/lib/lbrt/space/dsl/converter.rb

@@ -0,0 +1,107 @@
+class Lbrt::Space::DSL::Converter
+  def self.convert(exported, options = {})
+    self.new(exported, options).convert
+  end
+
+  def initialize(exported, options = {})
+    @exported = exported
+    @options = options
+  end
+
+  def convert
+    output_spaces(@exported)
+  end
+
+  private
+
+
+  def output_spaces(space_by_name_or_id)
+    spaces = []
+
+    space_by_name_or_id.sort_by {|k, v| k.to_s }.map do |name_or_id, attrs|
+      next unless target_matched?(name_or_id)
+      spaces << output_space(name_or_id, attrs)
+    end
+
+    spaces.join("\n")
+  end
+
+  def output_space(name_or_id, attrs)
+    chart_by_name_or_id = attrs.fetch('charts')
+
+    <<-EOS
+space #{name_or_id.inspect} do
+  #{output_charts(chart_by_name_or_id)}
+end
+    EOS
+  end
+
+  def output_charts(chart_by_name_or_id)
+    if chart_by_name_or_id.empty?
+      '# no chart'
+    else
+      # Don't sort!
+      chart_by_name_or_id.map {|name_or_id, attrs|
+        output_chart(name_or_id, attrs)
+      }.join("\n").strip
+    end
+  end
+
+  def output_chart(name_or_id, attrs)
+    type = attrs.fetch('type')
+    streams = attrs.fetch('streams')
+
+    out = <<-EOS
+  chart #{name_or_id.inspect} do
+    type #{type.inspect}
+    #{output_streams(streams)}
+    EOS
+
+    %w(max min label).each do |key|
+      next unless attrs.has_key?(key)
+      value = attrs[key]
+
+      out << <<-EOS
+    #{key} #{value.inspect}
+      EOS
+    end
+
+    out << <<-EOS
+  end
+    EOS
+  end
+
+  def output_streams(streams)
+    if streams.empty?
+      '# no stream'
+    else
+      streams.map {|i| output_stream(i) }.join.strip
+    end
+  end
+
+  def output_stream(stream)
+    metric = stream.fetch('metric')
+    type = stream.fetch('type')
+    source = stream.fetch('source')
+    group_function = stream.fetch('group_function')
+    summary_function = stream.fetch('summary_function')
+
+    <<-EOS
+    stream do
+      metric #{metric.inspect}
+      type #{type.inspect}
+      source #{source.inspect}
+      group_function #{group_function.inspect}
+      summary_function #{summary_function.inspect}
+    end
+    EOS
+  end
+
+  def target_matched?(str)
+    if @options[:target]
+      str =~ @options[:target]
+    else
+      true
+    end
+  end
+end

data/lib/lbrt/space/exporter.rb

@@ -0,0 +1,56 @@
+class Lbrt::Space::Exporter
+  DEFAULT_CONCURRENCY = 32
+
+  class << self
+    def export(client, options = {})
+      self.new(client, options).export
+    end
+  end # of class methods
+
+  def initialize(client, options = {})
+    @client = client
+    @options = options
+  end
+
+  def export
+    spaces = @client.spaces.get
+    normalize_spaces(spaces)
+  end
+
+  private
+
+  def normalize_spaces(spaces)
+    space_by_name_or_id = {}
+    concurrency = @options[:export_concurrency] || DEFAULT_CONCURRENCY
+
+    Parallel.each(spaces, :in_threads => concurrency) do |spc|
+      space_id = spc.fetch('id')
+      name_or_id = spc.fetch('name') || space_id
+      charts = @client.spaces(space_id).charts.get
+
+      if space_by_name_or_id[name_or_id]
+        raise "Duplicate space name exists: #{name}"
+      end
+
+      space_by_name_or_id[name_or_id] = {
+        'id' => space_id,
+        'charts' => normalize_charts(charts),
+      }
+    end
+
+    space_by_name_or_id
+  end
+
+  def normalize_charts(charts)
+    chart_by_name_or_id = {}
+
+    charts.each do |chrt|
+      name_or_id = chrt.delete('name')
+      name_or_id = chrt['id'] if name_or_id.empty?
+      chrt.delete('space_id')
+      chart_by_name_or_id[name_or_id] = chrt
+    end
+
+    chart_by_name_or_id
+  end
+end

data/lib/lbrt/utils.rb

@@ -0,0 +1,64 @@
+class Lbrt::Utils
+  class << self
+    def unbrace(str)
+      str.sub(/\A\s*\{/, '').sub(/\}\s*\z/, '').strip
+    end
+
+    def matched?(str, target)
+      str = str.to_s
+
+      if target
+        str =~ target
+      else
+        true
+      end
+    end
+  end # of class methods
+
+  module ContextHelper
+    def require(file)
+      file = (file =~ %r|\A/|) ? file : File.expand_path(File.join(File.dirname(@path), file))
+
+      if File.exist?(file)
+        instance_eval(File.read(file), file)
+      elsif File.exist?(file + '.rb')
+        instance_eval(File.read(file + '.rb'), file + '.rb')
+      else
+        Kernel.require(file)
+      end
+    end
+  end
+
+  module CLIHelper
+    REGEXP_OPTIONS = [
+      :target
+    ]
+
+    def client(klass)
+      librato = Librato::Client.new(
+        :user => options.delete(:user),
+        :token => options.delete(:token),
+        :debug => options[:debug]
+      )
+
+      String.colorize = options[:color]
+
+
+      options.keys.each do |key|
+        if key.to_s =~ /-/
+          value = options.delete(key)
+          key = key.to_s.gsub('-', '_').to_sym
+          options[key] = value
+        end
+      end
+
+      REGEXP_OPTIONS.each do |key|
+        if options[key]
+          options[key] = Regexp.new(options[key])
+        end
+      end
+
+      klass.new(librato, options)
+    end
+  end
+end

data/lib/lbrt/version.rb

@@ -0,0 +1,3 @@
+module Lbrt
+  VERSION = '0.1.0'
+end

metadata

@@ -0,0 +1,202 @@
+--- !ruby/object:Gem::Specification
+name: lbrt
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Genki Sugawara
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2015-08-16 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: diffy
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: librato-client
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: parallel
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: term-ansicolor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 3.0.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 3.0.0
+description: A tool to manage Librato. It defines the state of Librato using DSL,
+  and updates Librato according to DSL.
+email:
+- sgwr_dts@yahoo.co.jp
+executables:
+- lbrt
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- ".travis.yml"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- exe/lbrt
+- lbrt.gemspec
+- lib/lbrt.rb
+- lib/lbrt/alert.rb
+- lib/lbrt/alert/dsl.rb
+- lib/lbrt/alert/dsl/context.rb
+- lib/lbrt/alert/dsl/context/alert.rb
+- lib/lbrt/alert/dsl/context/alert/condition.rb
+- lib/lbrt/alert/dsl/converter.rb
+- lib/lbrt/alert/exporter.rb
+- lib/lbrt/cli.rb
+- lib/lbrt/cli/alert.rb
+- lib/lbrt/cli/app.rb
+- lib/lbrt/cli/service.rb
+- lib/lbrt/cli/space.rb
+- lib/lbrt/driver.rb
+- lib/lbrt/ext/string_ext.rb
+- lib/lbrt/logger.rb
+- lib/lbrt/service.rb
+- lib/lbrt/service/dsl.rb
+- lib/lbrt/service/dsl/context.rb
+- lib/lbrt/service/dsl/context/service.rb
+- lib/lbrt/service/dsl/converter.rb
+- lib/lbrt/service/exporter.rb
+- lib/lbrt/space.rb
+- lib/lbrt/space/dsl.rb
+- lib/lbrt/space/dsl/context.rb
+- lib/lbrt/space/dsl/context/space.rb
+- lib/lbrt/space/dsl/context/space/chart.rb
+- lib/lbrt/space/dsl/context/space/chart/stream.rb
+- lib/lbrt/space/dsl/converter.rb
+- lib/lbrt/space/exporter.rb
+- lib/lbrt/utils.rb
+- lib/lbrt/version.rb
+homepage: https://github.com/winebarrel/lbrt
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.5
+signing_key: 
+specification_version: 4
+summary: A tool to manage Librato. It defines the state of Librato using DSL, and
+  updates Librato according to DSL.
+test_files: []