flueconf 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 9effe81389c67ea7f4b51c8683f99402fcc20cf5bc43a18f5200ea43a712b363
+  data.tar.gz: 697df4d11e65ecf85ec54427e5326e9231b6e75b320d2be589bd9ac3327bd872
+SHA512:
+  metadata.gz: eb218202d17afbd5afb401ab713e83cfe4ad748f6429af6d34089426cdc95cfc17c2bfc13e0080aa3de81bfe6a6dafd46cf8cc946d705142335c8ee89508de51
+  data.tar.gz: 212ef5b7c85e2fd53c4ddb8401c2623d353644d74fa19f4e9103b058ee09460ebf533d1baae430b94300ed091a46489f0a3dc40b6d365e2d92336cc7a87a842a

data/lib/flueconf.rb

@@ -0,0 +1,10 @@
+module Flueconf
+  def build(*args, &block)
+    Flueconf::Builder.new(*args, &block)
+  end
+  module_function :build
+end
+
+require 'flueconf/builder'
+require 'flueconf/serializer'
+require 'flueconf/version'

data/lib/flueconf/builder.rb

@@ -0,0 +1,94 @@
+require 'json'
+
+class Flueconf::Builder
+  def initialize(options = {}, &block)
+    @attributes = {}
+    @current = @attributes
+    @context = nil
+
+    build(&block) if block_given?
+  end
+
+  def build(&block)
+    return unless block_given?
+
+    @context = eval('self', block.binding)
+
+    instance_eval(&block)
+
+    self
+  end
+
+  def label(*args, &block)
+    method = block_given? ? :label : :@label
+    add(method, *args, &block)
+  end
+
+  %i(type id).each do |method|
+    define_method method do |*args, &block|
+      add(:"@#{method}", *args, &block)
+    end
+  end
+
+  %i(source match filter system).each do |method|
+    define_method method do |*args, &block|
+      add(method, *args, &block)
+    end
+  end
+
+  def method_missing(method, *args, &block)
+    if @context and @context.respond_to?(method)
+      @context.send(method, *args, &block)
+    else
+      add(method, *args, &block)
+    end
+  end
+
+  def add(*args, &block)
+    if block_given?
+      obj = {}
+      k = args.first.to_sym
+
+      @current[k] = [] unless @current[k].is_a? Array
+      @current[k] << obj
+
+      c = obj
+      args[1..-1].each do |k|
+        obj = {}
+        c[k.to_sym] = obj
+        c = obj
+      end
+
+      previous = @current
+      @current = obj
+      instance_eval(&block)
+      @current = previous
+    elsif args.length >= 1
+      method = args.shift.to_sym
+
+      value = case args.length
+              when 0
+                nil
+              when 1
+                args.first
+              else
+                args
+              end
+      value = case value
+              when Hash
+                value.map { |k, v| v.nil? ? nil : "#{k}:#{v}" }.compact.join(',')
+              when Array
+                value.compact.join(',')
+              else
+                value
+              end
+      @current[method] = value
+    else
+      raise ArgumentError, "One key and at least one of values are required: #{args.join(',')}"
+    end
+  end
+
+  def to_fluent(*args)
+    Flueconf::Serializer.serialize(@attributes, *args)
+  end
+end

data/lib/flueconf/serializer.rb

@@ -0,0 +1,44 @@
+require 'json'
+
+class Flueconf::Serializer
+  class << self
+    def serialize(obj, options = {})
+      indent = options[:indent]
+      indent = 2 unless indent.is_a? Integer and indent >= 0
+
+      dump(obj, [], indent, 0)
+    end
+
+    private
+
+    def dump(obj, keys = [], indent = 0, depth = 0)
+      prefix = ' ' * indent * depth
+      key = keys.empty? ? nil : keys.join(' ')
+
+      case obj
+      when Array
+        obj.map do |v|
+          send(__method__, v, [key], indent, depth)
+        end.compact.join("\n")
+      when Hash
+        compacted = keys.empty? || (!obj.empty? && obj.values.all? { |v| v.is_a? Hash })
+
+        str = obj.map do |k, v|
+          k = k.to_s.strip
+
+          send(__method__, v, (compacted ? (keys + [k]) : [k]), indent, (compacted ? depth : depth + 1))
+        end.compact.join("\n")
+
+        compacted ?  str : "#{prefix}<#{key}>\n" + (str.empty? ? '' : "#{str}\n") + "#{prefix}</#{keys.first}>"
+      when Numeric, TrueClass, FalseClass
+        (key ? "#{prefix}#{key} " : '') + obj.inspect
+      when NilClass
+        key ? "#{prefix}#{key}" : nil
+      when Regexp
+        (key ? "#{prefix}#{key} " : '') + obj.to_s
+      else
+        (key ? "#{prefix}#{key} " : '') + (/["#]/.match?(obj) ? obj.to_s.to_json : obj.to_s)
+      end
+    end
+  end
+end

data/lib/flueconf/version.rb

@@ -0,0 +1,8 @@
+module Flueconf
+  VERSION = '0.0.1'.freeze
+
+  def version
+    VERSION
+  end
+  module_function :version
+end

data/spec/flueconf_spec.rb

@@ -0,0 +1,340 @@
+describe Flueconf do
+  it 'can handle primitive types' do
+    fluent = Flueconf.build do
+      it_is_integer 1
+      it_is_negtive_integer -1
+      it_is_float 1.23
+      it_is_negtive_float -1.23
+      it_is_string 'foo'
+      it_is_special_string '#{Socket.gethostname}'
+      it_is_null nil
+      it_is_true true
+      it_is_false false
+      it_is_regexp /^.*$/
+      it_is_array [1, -2, 3.3, -4.4, 'bar', nil, true, false]
+      it_is_empty_array []
+      it_is_hash({ foo: 123, bar: 'aaa', null: nil })
+      it_is_empty_hash({})
+    end.to_fluent
+
+    expect(fluent).to eq <<~'EOF'.chomp
+      it_is_integer 1
+      it_is_negtive_integer -1
+      it_is_float 1.23
+      it_is_negtive_float -1.23
+      it_is_string foo
+      it_is_special_string "#{Socket.gethostname}"
+      it_is_null
+      it_is_true true
+      it_is_false false
+      it_is_regexp (?-mix:^.*$)
+      it_is_array 1,-2,3.3,-4.4,bar,true,false
+      it_is_empty_array 
+      it_is_hash foo:123,bar:aaa
+      it_is_empty_hash 
+    EOF
+  end
+
+  it 'can handle array object' do
+    fluent = Flueconf.build do
+      it_is_array_object do
+        foo 'bar'
+      end
+    end.to_fluent
+
+    expect(fluent).to eq <<~'EOF'.chomp
+      <it_is_array_object>
+        foo bar
+      </it_is_array_object>
+    EOF
+  end
+
+  it 'can handle empty array object' do
+    fluent = Flueconf.build do
+      it_is_array_object do
+      end
+    end.to_fluent
+
+    expect(fluent).to eq <<~'EOF'.chomp
+      <it_is_array_object>
+      </it_is_array_object>
+    EOF
+  end
+
+  it 'can handle object' do
+    fluent = Flueconf.build do
+      it_is_object 'obj' do
+        foo 'bar'
+      end
+    end.to_fluent
+
+    expect(fluent).to eq <<~'EOF'.chomp
+      <it_is_object obj>
+        foo bar
+      </it_is_object>
+    EOF
+  end
+
+  it 'can handle empty object' do
+    fluent = Flueconf.build do
+      it_is_object 'obj' do
+      end
+    end.to_fluent
+
+    expect(fluent).to eq <<~'EOF'.chomp
+      <it_is_object obj>
+      </it_is_object>
+    EOF
+  end
+
+  it 'can handle object with multiple keys' do
+    fluent = Flueconf.build do
+      it_is_object 'o', 'b', 'j' do
+        foo 'bar'
+      end
+    end.to_fluent
+
+    expect(fluent).to eq <<~'EOF'.chomp
+      <it_is_object o b j>
+        foo bar
+      </it_is_object>
+    EOF
+  end
+
+  it 'can handle nested data types' do
+    fluent = Flueconf.build do
+      foo 'bar'
+      obj 'nested', 'obj' do
+        obj 'obj', 'in', 'obj' do
+          float 3.14
+          arr do
+            int 123
+          end
+          arr do
+            bool true
+            arr_in_arr do
+              bool false
+            end
+          end
+        end
+      end
+    end.to_fluent
+
+    expect(fluent).to eq <<~'EOF'.chomp
+      foo bar
+      <obj nested obj>
+        <obj obj in obj>
+          float 3.14
+          <arr>
+            int 123
+          </arr>
+          <arr>
+            bool true
+            <arr_in_arr>
+              bool false
+            </arr_in_arr>
+          </arr>
+        </obj>
+      </obj>
+    EOF
+  end
+
+  it 'can handle multiple builds' do
+    builder = Flueconf.build do
+      obj do
+        first 1
+      end
+    end
+
+    fluent = builder.build do
+      second 2
+    end.to_fluent
+
+    expect(fluent).to eq <<~'EOF'.chomp
+      <obj>
+        first 1
+      </obj>
+      second 2
+    EOF
+  end
+
+  it 'can serialize with different indent' do
+    fluent = Flueconf.build do
+      foo 'bar'
+      obj 'nested', 'obj' do
+        obj 'obj', 'in', 'obj' do
+          float 3.14
+          arr do
+            int 123
+          end
+          arr do
+            bool true
+          end
+        end
+      end
+    end.to_fluent(indent: 4)
+
+    expect(fluent).to eq <<~'EOF'.chomp
+      foo bar
+      <obj nested obj>
+          <obj obj in obj>
+              float 3.14
+              <arr>
+                  int 123
+              </arr>
+              <arr>
+                  bool true
+              </arr>
+          </obj>
+      </obj>
+    EOF
+  end
+
+  it 'can handle method in different context' do
+    def bar
+      'barbar'
+    end
+
+    fluent = Flueconf.build do
+      foo bar
+    end.to_fluent
+
+    expect(fluent).to eq <<~'EOF'.chomp
+      foo barbar
+    EOF
+  end
+
+  it 'can handle predefined types' do
+    fluent = Flueconf.build do
+      system do
+        dir_permission '0755'
+      end
+      source do
+        type 'forward'
+        id 'in_forward'
+        label '@outside'
+        port 24224
+      end
+      source do
+        type 'tail'
+        id 'in_tail_httpd_access'
+        path '/var/log/httpd-access.log'
+        pos_file '/var/log/td-agent/httpd-access.log.pos'
+        tag 'apache.access'
+        parse do
+          type 'apache2'
+        end
+      end
+      filter 'fluent.**' do
+        type 'record_transformer'
+        record do
+          level '${tag_parts[1]}'
+        end
+      end
+      filter '**' do
+        type 'record_transformer'
+        record do
+          hostname '#{Socket.gethostname}'
+        end
+      end
+      match 'fluent.**' do
+        type 'rewrite_tag_filter'
+        rule do
+          key 'message'
+          pattern /.*/
+          tag 'fluent'
+        end
+      end
+      match '**' do
+        type 'forward'
+        server do
+          host '192.0.2.1'
+          weight 100
+        end
+        server do
+          host '192.0.2.2'
+          weight 50
+        end
+      end
+      label '@outside' do
+        match '**' do
+          type 'forward'
+          server do
+            host '192.0.2.1'
+            weight 100
+          end
+          server do
+            host '192.0.2.2'
+            weight 50
+          end
+        end
+      end
+    end.to_fluent
+
+    expect(fluent).to eq <<~'EOF'.chomp
+      <system>
+        dir_permission 0755
+      </system>
+      <source>
+        @type forward
+        @id in_forward
+        @label @outside
+        port 24224
+      </source>
+      <source>
+        @type tail
+        @id in_tail_httpd_access
+        path /var/log/httpd-access.log
+        pos_file /var/log/td-agent/httpd-access.log.pos
+        tag apache.access
+        <parse>
+          @type apache2
+        </parse>
+      </source>
+      <filter fluent.**>
+        @type record_transformer
+        <record>
+          level ${tag_parts[1]}
+        </record>
+      </filter>
+      <filter **>
+        @type record_transformer
+        <record>
+          hostname "#{Socket.gethostname}"
+        </record>
+      </filter>
+      <match fluent.**>
+        @type rewrite_tag_filter
+        <rule>
+          key message
+          pattern (?-mix:.*)
+          tag fluent
+        </rule>
+      </match>
+      <match **>
+        @type forward
+        <server>
+          host 192.0.2.1
+          weight 100
+        </server>
+        <server>
+          host 192.0.2.2
+          weight 50
+        </server>
+      </match>
+      <label @outside>
+        <match **>
+          @type forward
+          <server>
+            host 192.0.2.1
+            weight 100
+          </server>
+          <server>
+            host 192.0.2.2
+            weight 50
+          </server>
+        </match>
+      </label>
+    EOF
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1 @@
+require 'flueconf'

metadata

@@ -0,0 +1,94 @@
+--- !ruby/object:Gem::Specification
+name: flueconf
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Rianol Jou
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2019-08-17 00:00:00.000000000 Z
+dependencies:
+- !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: '12.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '12.3'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.8'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.8'
+description: "    Config fluentd in ruby. And featuring all programming features (variables,
+  iterators, functions, regexp, etc) in ruby.\n"
+email:
+- rianol.jou@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/flueconf.rb
+- lib/flueconf/builder.rb
+- lib/flueconf/serializer.rb
+- lib/flueconf/version.rb
+- spec/flueconf_spec.rb
+- spec/spec_helper.rb
+homepage: https://github.com/RiANOl/flueconf
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '2.4'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.0.3
+signing_key: 
+specification_version: 4
+summary: Config fluentd in ruby.
+test_files:
+- spec/flueconf_spec.rb
+- spec/spec_helper.rb