fluent-plugin-config_pit 0.0.1

data/.gitignore

@@ -0,0 +1,18 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+vendor/

data/Changes

@@ -0,0 +1,4 @@
+Revision history for fluent-plugin-config_pit
+
+0.01    Wed Feb  19
+        - original version

data/Gemfile

@@ -0,0 +1,4 @@
+source :rubygems
+
+# Specify your gem's dependencies in fluent-plugin-config-pit.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,13 @@
+Copyright (c) 2013- Naoya Ito
+
+   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.

data/README.md

@@ -0,0 +1,50 @@
+# fluent-plugin-config_pit
+
+## ConfigPitInput, ConfigPitOutput
+
+Fluentd Input/Output plugin which provides pit directive in configuration.
+
+## Configuration
+
+Fore both of input and output (for source and match), you can use 'pit' directive like below:
+
+    <match my.tag>
+      type config_pit
+      <pit aws>
+        type s3
+        ...
+        aws_key_id  $pit[aws_access_key]
+        aws_seckey  $pit[aws_secret_access_key]
+        ...
+      </pit>
+    </match>
+
+Variables '$pit[key]' will be replaced with values saved in ~/.pit directory by pit (https://github.com/cho45/pit). So you can safely treat API credentials or something like that.
+
+## Copyright
+
+* Copyright (c) 2013- Naoya Ito (@naoya_ito)
+* License
+  * Apache License, Version 2.0
+  
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'fluent-plugin-config-pit'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install fluent-plugin-config-pit  
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1,9 @@
+require "bundler/gem_tasks"
+require 'rake/testtask'
+
+Rake::TestTask.new do |t|
+  t.libs << "test"
+  t.pattern = 'test/**/test_*.rb'
+end
+
+task :default => :test

data/fluent-plugin-config_pit.gemspec

@@ -0,0 +1,23 @@
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |gem|
+  gem.name          = "fluent-plugin-config_pit"
+  gem.version       = "0.0.1"
+  gem.authors       = ["Naoya Ito"]
+  gem.email         = ["i.naoya@gmail.com"]
+  gem.description   = %q{This plugin provides directive for pit}
+  gem.summary       = %q{Fluentd plugin to enable pit in configuration}
+  gem.homepage      = "https://github.com/naoay/fluent-plugin-config_pit"
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.require_paths = ["lib"]
+
+  gem.add_development_dependency "rake"
+  gem.add_development_dependency "fluentd"
+  gem.add_development_dependency "pit"
+
+  gem.add_runtime_dependency "fluentd"
+  gem.add_runtime_dependency "pit"
+end

data/lib/fluent/plugin/in_config_pit.rb

@@ -0,0 +1,36 @@
+require_relative 'pit'
+
+class Fluent::ConfigPitInput < Fluent::Input
+  attr_accessor :plugin
+
+  Fluent::Plugin.register_input('config_pit', self)
+
+  def mark_used(conf)
+    conf.used = conf.keys
+    conf.elements.each{|e| mark_used(e)}
+  end
+
+  def configure(conf)
+    super
+
+    configs = conf.elements.select {|e| e.name == 'pit' }
+    if configs.size != 1
+      raise Fluent::ConfigError, "config_pit needs just one <pit ...> ... </pit> section"
+    end
+
+    Fluent::Config::Pit.extract(configs.first).tap do |c|
+      @plugin = Fluent::Plugin.new_input(c['type'])
+      @plugin.configure(c)
+    end
+
+    mark_used(configs.first)
+  end
+
+  def start
+    @plugin.start
+  end
+
+  def shutdown
+    @plugin.shutdown
+  end
+end

data/lib/fluent/plugin/out_config_pit.rb

@@ -0,0 +1,40 @@
+require_relative 'pit'
+
+class Fluent::ConfigPitOutput < Fluent::Output
+  attr_accessor :plugin
+
+  Fluent::Plugin.register_output('config_pit', self)
+
+  def mark_used(conf)
+    conf.used = conf.keys
+    conf.elements.each{|e| mark_used(e)}
+  end
+
+  def configure(conf)
+    super
+
+    configs = conf.elements.select {|e| e.name == 'pit' }
+    if configs.size != 1
+      raise Fluent::ConfigError, "config_pit needs just one <pit ...> ... </pit> section"
+    end
+
+    Fluent::Config::Pit.extract(configs.first).tap do |c|
+      @plugin = Fluent::Plugin.new_output(c['type'])
+      @plugin.configure(c)
+    end
+
+    mark_used(configs.first)
+  end
+
+  def start
+    @plugin.start
+  end
+
+  def shutdown
+    @plugin.shutdown
+  end
+
+  def emit(tag, es, chain)
+    @plugin.emit(tag, es, chain)
+  end
+end

data/lib/fluent/plugin/pit.rb

@@ -0,0 +1,14 @@
+require 'pit'
+
+module Fluent::Config::Pit
+  def self.extract(conf)
+    pit = Pit.get(conf.arg)
+    ex = Fluent::Config::Element.new('', '', conf, [])
+    ex.each do |k, v|
+      if v =~ /^\$pit\[(.+)\]$/
+        ex[k] = pit[$1]
+      end
+    end
+    return ex
+  end
+end

data/test/helper.rb

@@ -0,0 +1,29 @@
+require 'tmpdir'
+require 'pathname'
+
+class Pathname
+  @@tempname_number = 0
+  def self.tempname(base=$0, dir=Dir.tmpdir)
+    @@tempname_number += 1
+    path = new(dir) + "#{File.basename(base)}.#{$$}.#{@@tempname_number}"
+    at_exit do
+      path.rmtree if path.exist?
+    end
+    path
+  end
+end
+
+dir = Pathname.tempname
+dir.mkpath
+ENV["HOME"] = dir.to_s
+
+require 'pit'
+require 'test/unit'
+
+require 'fluent/test'
+require 'fluent/config'
+
+require 'fluent/plugin/pit'
+require 'fluent/plugin/in_config_pit'
+require 'fluent/plugin/out_config_pit'
+require 'plugins'

data/test/plugin/test_in_config_pit.rb

@@ -0,0 +1,28 @@
+require 'helper'
+
+class ConfigPitInputTest < Test::Unit::TestCase
+  def setup
+    Pit.set("test", :data => {"user" => "fiorung", "password" => "xenoblade" })
+    Fluent::Test.setup
+  end
+
+  def create_driver(conf, pit_id = 'test')
+    Fluent::Test::InputTestDriver.new(Fluent::ConfigPitInput).configure(conf)
+  end
+
+  def test_configure
+    driver = create_driver %[
+      <pit test>
+        type  config_pit_test
+        tag   pit_test
+        user  $pit[user]
+        password $pit[password]
+      </pit>
+    ]
+
+    assert_equal 'Fluent::ConfigPitTestInput', driver.instance.plugin.class.to_s
+    assert_equal 'pit_test', driver.instance.plugin.tag
+    assert_equal 'fiorung', driver.instance.plugin.user
+    assert_equal 'xenoblade', driver.instance.plugin.password
+  end
+end

data/test/plugin/test_out_config_pit.rb

@@ -0,0 +1,28 @@
+require 'helper'
+
+class ConfigPitOutputTest < Test::Unit::TestCase
+  def setup
+    Pit.set("test", :data => {"user" => "fiorung", "password" => "xenoblade" })
+    Fluent::Test.setup
+  end
+
+  def create_driver(conf, pit_id = 'test')
+    Fluent::Test::OutputTestDriver.new(Fluent::ConfigPitOutput).configure(conf)
+  end
+
+  def test_configure
+    driver = create_driver %[
+      <pit test>
+        type  config_pit_test
+        tag   pit_test
+        user  $pit[user]
+        password $pit[password]
+      </pit>
+    ]
+
+    assert_equal 'Fluent::ConfigPitTestOutput', driver.instance.plugin.class.to_s
+    assert_equal 'pit_test', driver.instance.plugin.tag
+    assert_equal 'fiorung', driver.instance.plugin.user
+    assert_equal 'xenoblade', driver.instance.plugin.password
+  end
+end

data/test/plugin/test_pit.rb

@@ -0,0 +1,21 @@
+require 'helper'
+
+class ConfigPitTest < Test::Unit::TestCase
+  def setup
+    Pit.set("test", :data => {"user" => "fiorung", "password" => "xenoblade" })
+    @module = Fluent::Config::Pit
+  end
+
+  def test_extract
+    attrs = {
+      'tag' => 'mytag',
+      'user' => '$pit[user]',
+      'password' => '$pit[password]'
+    }
+    conf = @module.extract Fluent::Config::Element.new('pit', 'test', attrs, [])
+
+    assert_equal conf['tag'], 'mytag'
+    assert_equal conf['user'], 'fiorung'
+    assert_equal conf['password'], 'xenoblade'
+  end
+end

data/test/plugins.rb

@@ -0,0 +1,13 @@
+class Fluent::ConfigPitTestInput < Fluent::Input
+  Fluent::Plugin.register_input('config_pit_test', self)
+  config_param :tag, :string
+  config_param :user, :string
+  config_param :password, :string
+end
+
+class Fluent::ConfigPitTestOutput < Fluent::Output
+  Fluent::Plugin.register_output('config_pit_test', self)
+  config_param :tag, :string
+  config_param :user, :string
+  config_param :password, :string
+end

metadata

@@ -0,0 +1,145 @@
+--- !ruby/object:Gem::Specification
+name: fluent-plugin-config_pit
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Naoya Ito
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-02-19 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: fluentd
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pit
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: fluentd
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pit
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: This plugin provides directive for pit
+email:
+- i.naoya@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Changes
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- fluent-plugin-config_pit.gemspec
+- lib/fluent/plugin/in_config_pit.rb
+- lib/fluent/plugin/out_config_pit.rb
+- lib/fluent/plugin/pit.rb
+- test/helper.rb
+- test/plugin/test_in_config_pit.rb
+- test/plugin/test_out_config_pit.rb
+- test/plugin/test_pit.rb
+- test/plugins.rb
+homepage: https://github.com/naoay/fluent-plugin-config_pit
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Fluentd plugin to enable pit in configuration
+test_files:
+- test/helper.rb
+- test/plugin/test_in_config_pit.rb
+- test/plugin/test_out_config_pit.rb
+- test/plugin/test_pit.rb
+- test/plugins.rb