fluentd-server 0.1.0

data/lib/fluentd_server.rb

@@ -0,0 +1 @@
+require "fluentd_server/version"

data/lib/fluentd_server/cli.rb

@@ -0,0 +1,64 @@
+require "fileutils"
+require "dotenv"
+require "thor"
+require 'fluentd_server'
+
+class FluentdServer::CLI < Thor
+  BASE_DIR = File.join(Dir.pwd, "fluentd-server")
+  DATA_DIR = File.join(BASE_DIR, "data")
+  LOG_DIR = File.join(BASE_DIR, "log")
+  LOG_FILE = File.join(LOG_DIR, "application.log")
+  ENV_FILE = File.join(BASE_DIR, ".env")
+  PROCFILE = File.join(BASE_DIR, "Procfile")
+  CONFIGRU_FILE = File.join(BASE_DIR, "config.ru")
+  DB_DIR = File.join(BASE_DIR, "db")
+  CONFIG_DIR= File.join(BASE_DIR, "config")
+
+  DEFAULT_DOTENV =<<-EOS
+PORT=5126
+HOST=0.0.0.0
+DATABASE_URL=sqlite3:#{DATA_DIR}/fluentd_server.db
+LOG_PATH=#{LOG_FILE}
+LOG_LEVEL=warn
+EOS
+
+  DEFAULT_PROCFILE =<<-EOS
+web: unicorn -E production -p $PORT -o $HOST -c config/unicorn.conf
+EOS
+
+  default_command :start
+
+  desc "new", "Creates fluentd-server resource directory"
+  def new
+    FileUtils.mkdir_p(LOG_DIR)
+    File.write ENV_FILE, DEFAULT_DOTENV
+    File.write PROCFILE, DEFAULT_PROCFILE
+    FileUtils.cp(File.expand_path("../../../config.ru", __FILE__), CONFIGRU_FILE)
+    FileUtils.cp_r(File.expand_path("../../../db", __FILE__), DB_DIR)
+    FileUtils.cp_r(File.expand_path("../../../config", __FILE__), CONFIG_DIR)
+    puts 'fluentd-server new finished.'
+  end
+
+  desc "init", "Creates database schema"
+  def init
+    require 'fluentd_server/environments'
+    require 'rake'
+    require 'sinatra/activerecord/rake'
+    Rake::Task['db:migrate'].invoke
+    puts 'fluentd-server init finished.'
+  end
+
+  desc "start", "Sartup fluentd_server"
+  def start
+    Dotenv.load
+    require "foreman/cli"
+    Foreman::CLI.new.invoke(:start, [], {})
+  end
+
+  no_tasks do
+    def abort(msg)
+      $stderr.puts msg
+      exit 1
+    end
+  end
+end

data/lib/fluentd_server/config.rb

@@ -0,0 +1,27 @@
+require 'sinatra'
+require 'sinatra/activerecord'
+require 'fluentd_server'
+require 'dotenv'
+Dotenv.load
+
+module FluentdServer::Config
+  def self.database_url
+    ENV.fetch('DATABASE_URL', 'sqlite3:data/fluentd_server.db')
+  end
+
+  def self.log_path
+    ENV.fetch('LOG_PATH', 'STDOUT')
+  end
+
+  def self.log_level
+    ENV.fetch('LOG_LEVEL', 'debug')
+  end
+
+  def self.log_shift_age
+    ENV.fetch('LOG_SHIFT_AGE', '0')
+  end
+
+  def self.log_shift_size
+    ENV.fetch('LOG_SHIFT_SIZE', '1048576')
+  end
+end

data/lib/fluentd_server/decorator.rb

@@ -0,0 +1,21 @@
+require 'sinatra/decorator'
+
+class PostDecorator < Sinatra::Decorator::Base
+  include Rack::Utils
+
+  def success_message
+    'Success!'
+  end
+
+  def error_message
+    message = 'Failure! '
+    message += self.errors.map {|key, msg| escape_html("`#{key}` #{msg}") }.join('. ')
+    message
+  end
+
+  def render_body(locals)
+    namespace = OpenStruct.new(locals)
+    ERB.new(self.body, nil, "-").result(namespace.instance_eval { binding })
+  end
+end
+

data/lib/fluentd_server/environments.rb

@@ -0,0 +1,34 @@
+require 'sinatra'
+require 'sinatra/activerecord'
+require 'fluentd_server/config'
+require 'fluentd_server/logger'
+
+configure do
+  set :show_exceptions, true
+  ActiveRecord::Base.logger = FluentdServer.logger
+  I18n.enforce_available_locales = false
+end
+
+configure :production, :development do
+  if FluentdServer::Config.database_url.start_with?('sqlite')
+    set :database, FluentdServer::Config.database_url 
+  else
+    # DATABASE_URL => "postgres://randuser:randpass@randhost:randport/randdb" on heroku
+    db = URI.parse(FluentdServer::Config.database_url)
+    ActiveRecord::Base.establish_connection(
+      :adapter  => db.scheme == 'postgres' ? 'postgresql' : db.scheme,
+      :host     => db.host,
+      :username => db.user,
+      :password => db.password,
+      :database => db.path[1..-1],
+      :encoding => 'utf8'
+    )
+  end
+end
+
+configure :test do
+  ActiveRecord::Base.establish_connection(
+    :adapter  => 'sqlite3',
+    :database => ':memory'
+  )
+end

data/lib/fluentd_server/logger.rb

@@ -0,0 +1,90 @@
+require 'logger'
+require 'fluentd_server'
+require "fluentd_server/config"
+
+module FluentdServer
+  module Logger
+    def self.included(klass)
+      # To define logger *class* method
+      klass.extend(self)
+    end
+
+    # for test
+    def logger=(logger)
+      FluentdServer.logger = logger
+    end
+
+    def logger
+      FluentdServer.logger
+    end
+  end
+
+  # for test
+  def self.logger=(logger)
+    @logger = logger
+  end
+
+  def self.logger
+    return @logger if @logger
+
+    log_path = FluentdServer::Logger::Config.log_path
+    log_level = FluentdServer::Logger::Config.log_level
+    # NOTE: Please note that ruby 2.0.0's Logger has a problem on log rotation.
+    # Update to ruby 2.1.0. See https://github.com/ruby/ruby/pull/428 for details.
+    log_shift_age = FluentdServer::Logger::Config.log_shift_age
+    log_shift_size = FluentdServer::Logger::Config.log_shift_size
+    @logger = ::Logger.new(log_path, log_shift_age, log_shift_size)
+    @logger.level = log_level
+    @logger
+  end
+
+  class Logger::Config
+    def self.log_path(log_path = FluentdServer::Config.log_path)
+      case log_path
+      when 'STDOUT'
+        $stdout
+      when 'STDERR'
+        $stderr
+      else
+        log_path
+      end
+    end
+
+    def self.log_level(log_level = FluentdServer::Config.log_level)
+      case log_level
+      when 'debug'
+        ::Logger::DEBUG
+      when 'info'
+        ::Logger::INFO
+      when 'warn'
+        ::Logger::WARN
+      when 'error'
+        ::Logger::ERROR
+      when 'fatal'
+        ::Logger::FATAL
+      else
+        raise ArgumentError, "invalid log_level #{log_level}"
+      end
+    end
+
+    def self.log_shift_age(log_shift_age = FluentdServer::Config.log_shift_age)
+      case log_shift_age
+      when /\d+/
+        log_shift_age.to_i
+      when 'daily'
+        log_shift_age
+      when 'weekly'
+        log_shift_age
+      when 'monthly'
+        log_shift_age
+      else
+        raise ArgumentError, "invalid log_shift_age #{log_shift_age}"
+      end
+    end
+
+    def self.log_shift_size(log_shift_size = FluentdServer::Config.log_shift_size)
+      log_shift_size.to_i
+    end
+  end
+
+end

data/lib/fluentd_server/model.rb

@@ -0,0 +1,10 @@
+require 'sinatra/activerecord'
+require 'sinatra/decorator'
+require 'fluentd_server/environments'
+
+class Post < ActiveRecord::Base
+  include Sinatra::Decorator::Decoratable
+
+  validates :name, presence: true
+  validates :body, presence: true
+end

data/lib/fluentd_server/version.rb

@@ -0,0 +1,3 @@
+module FluentdServer
+  VERSION = "0.1.0"
+end

data/lib/fluentd_server/web.rb

@@ -0,0 +1,101 @@
+require 'sinatra'
+require 'sinatra/flash'
+require 'sinatra/redirect_with_flash'
+require 'slim'
+
+require 'fluentd_server'
+require 'fluentd_server/config'
+require 'fluentd_server/environments'
+require 'fluentd_server/model'
+require 'fluentd_server/decorator'
+require 'fluentd_server/logger'
+require 'fluentd_server/web_helper'
+
+class FluentdServer::Web < Sinatra::Base
+  include FluentdServer::Logger
+  helpers FluentdServer::WebHelper
+
+  set :dump_errors, true
+  set :public_folder, File.join(__dir__, '..', '..', 'public')
+  set :views,         File.join(__dir__, '..', '..', 'views')
+
+  enable :sessions
+  register Sinatra::Flash
+  helpers Sinatra::RedirectWithFlash
+  register Sinatra::ActiveRecordExtension
+
+  # get ALL posts
+  get "/" do
+    @posts = Post.order("name ASC")
+    slim :"posts/index"
+  end
+
+  # create new post
+  get "/posts/create" do
+    @tab = 'create'
+    @title = 'Create'
+    @post = Post.new
+    slim :"posts/create"
+  end
+
+  post "/posts" do
+    @post = Post.new(params[:post])
+    if @post.save
+      redirect "posts/#{@post.id}/edit", :notice => @post.decorate.success_message
+    else
+      redirect "posts/create", :error => @post.decorate.error_message
+    end
+  end
+
+  # edit post
+  get "/posts/:id/edit" do
+    @title = 'Edit'
+    @post = Post.find(params[:id])
+    return 404 unless @post
+    slim :"posts/edit"
+  end
+
+  post "/posts/:id" do
+    @post = Post.find(params[:id])
+    return 404 unless @post
+    if @post.update(params[:post])
+      redirect "/posts/#{@post.id}/edit", :notice => @post.decorate.success_message
+    else
+      redirect "/posts/#{@post.id}/edit", :error => @post.decorate.error_message
+    end
+  end
+
+  # delete post
+  get "/posts/:name/delete" do
+    @post = Post.find_by(name: params[:name])
+    if @post.destroy
+      redirect "/", :notice => @post.decorate.success_message
+    else
+      redirect "/", :error => @post.decorate.error_message
+    end
+  end
+
+  # get ALL posts in json
+  get "/json/list" do
+    @posts = Post.order("id ASC")
+    content_type :json
+    @posts.to_json
+  end
+
+  # get post in json
+  get "/json/:name" do
+    @post = Post.find_by(name: params[:name])
+    return 404 unless @post
+    content_type :json
+    @post.to_json
+  end
+
+  # render api
+  get "/api/:name" do
+    @post = Post.find_by(name: params[:name])
+    return 404 unless @post
+    query_params = parse_query(request.query_string)
+    content_type :text
+    @post.decorate.render_body(query_params)
+  end
+end

data/lib/fluentd_server/web_helper.rb

@@ -0,0 +1,91 @@
+require 'cgi'
+require 'fluentd_server'
+
+module FluentdServer::WebHelper
+  include Rack::Utils
+  alias_method :h, :escape_html
+
+  # override RackUtil.parse_query
+  # @param qs query string
+  # @param d delimiter
+  # @return
+  def parse_query(qs, d=nil)
+    params = {}
+    (qs || '').split(d ? /[#{d}] */n : DEFAULT_SEP).each do |p|
+      k, v = p.split('=', 2).map { |x| unescape(x) }
+      if k.ends_with?('[]')
+        k1 = k[0..-3]
+        if params[k1] and params[k1].class == Array
+          params[k1] << v
+        else
+          params[k1] = [v]
+        end
+      elsif k.ends_with?(']') and md = k.match(/^([^\[]+)\[([^\]]+)\]$/)
+        k1, k2 = md[1], md[2]
+        if params[k1] and params[k1].class == Hash
+          params[k1][k2] = v
+        else
+          params[k1] = { k2 => v }
+        end
+      else
+        params[k] = v
+      end
+    end
+    params
+  end
+
+  def url_for(url_fragment, mode=nil, options = nil)
+    if mode.is_a? Hash
+      options = mode
+      mode = nil
+    end
+
+    if mode.nil?
+      mode = :path_only
+    end
+
+    mode = mode.to_sym unless mode.is_a? Symbol
+    optstring = nil
+
+    if options.is_a? Hash
+      optstring = '?' + options.map { |k,v| "#{k}=#{URI.escape(v.to_s, /[^#{URI::PATTERN::UNRESERVED}]/)}" }.join('&')
+    end
+
+    case mode
+    when :path_only
+      base = request.script_name
+    when :full
+      scheme = request.scheme
+      if (scheme == 'http' && request.port == 80 ||
+          scheme == 'https' && request.port == 443)
+        port = ""
+      else
+        port = ":#{request.port}"
+      end
+      base = "#{scheme}://#{request.host}#{port}#{request.script_name}"
+    else
+      raise TypeError, "Unknown url_for mode #{mode.inspect}"
+    end
+    "#{base}#{url_fragment}#{optstring}"
+  end
+
+  def escape_url(str)
+    CGI.escape(str)
+  end
+
+  def title
+    @title || 'Welcome'
+  end
+
+  def tab
+    @tab || 'welcome'
+  end
+
+  def active(_tab)
+    if tab == _tab
+      'active'
+    else
+      ''
+    end
+  end
+end