Contents

require 'base64'
require 'fileutils'
require 'uri'

begin
  require 'sequel'
rescue LoadError
  puts 'sequel gem is required to use file store'
end


module Jackfs
  class DbAdapter
    TEMP_PATH = File.join('tmp','fs_cache')

    attr_accessor :connection, :table_name, :dbase, :app_root, :app_env

    def initialize(app_root, app_env)
      @app_root = app_root
      @app_env = app_env

      FileUtils.mkdir_p temp_file_path

      yml = YAML.load_file(config_file)[@app_env.to_s]
      @connection = uri_escape(yml["connection"])
      @table_name = yml["table_name"]

      # Clean up temp files
      FileUtils.remove_file(File.join(temp_file_path,'/*'), true)
    end

    def store(f, name)
      body = Base64.encode64(f.read)
      data.insert(
        :name       => name,
        :body       => body,
        :created_at => Time.now,
        :updated_at => Time.now
      )
      name
    end

    def get(name)
      record = data.where(:name => name).order(:name).first
      unique_name = UUIDTools::UUID.random_create.to_s
      # Write Body to generated tmp file
      f = temp_file_path + unique_name
      open(File.join(f), 'wb') { |file| file.write Base64.decode64(record[:body]) }
      f
    end

    def data
      db = Sequel.connect(@connection)
      make_table(db) unless db.tables.include?(@table_name.to_sym)
      db[@table_name.to_sym]
    end

    def make_table(db)
      db.create_table @table_name do
        primary_key :id
        string :name
        text :body
        timestamp :created_at
        timestamp :updated_at
      end
    rescue
      # table exists
    end

    def temp_file_path
      File.join(@app_root, TEMP_PATH)
    end

    def config_file
      File.join(@app_root, Jackfs::FileStore::CONFIG_FILE)
    end

    def uri_escape(uri)
      begin
        URI.escape(uri)
      rescue Exception => e
        $stdout.puts "Error encountered when parsing #{uri} - #{e.message}"
      end
    end    

  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
jackfs-0.1.1	lib/jackfs/adapters/db_adapter.rb