totem 0.0.1

data/.gitignore

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

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in totem.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2013 Chad Remesch
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,33 @@
+# Totem
+
+Totem is a framework for writing actor model based servers using the Ruby language.
+Currently it is in an experimental state, but you are welcome to play with it.
+
+Features:
+- Ruby on Rails inspired folder structure.
+- Includes Active Record and Active Support.
+- Generators for database migrations.
+- Integrated console.
+- Designed for MRI and JRuby.
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'totem'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install totem
+
+## 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,13 @@
+require 'bundler/gem_tasks'
+
+desc 'Start an IRB console with Workers loaded'
+task :console do
+  $LOAD_PATH.unshift(File.join(File.dirname(__FILE__), 'lib'))
+
+  require 'totem'
+  require 'irb'
+
+  ARGV.clear
+
+  IRB.start
+end

data/lib/totem.rb

@@ -0,0 +1,147 @@
+require 'tribe'
+require 'tribe_em'
+
+require 'totem/version'
+require 'totem/tasks/console'
+require 'totem/tasks/database'
+require 'totem/tasks/generator'
+require 'totem/actable'
+require 'totem/connection'
+require 'totem/tcp_server'
+
+module Totem
+  def self.initialize(root)
+    raise 'Already initialized.' if @setup
+
+    @setup = true
+    @root = root
+
+    Bundler.require(Totem.env.to_sym)
+    Time.zone = 'UTC'
+    db_connect
+    load_app
+
+    return true
+  end
+
+  def self.root
+    return @root
+  end
+
+  def self.env
+    return (@env ||= (ENV['GAME_SRV_ENV'] || 'development'))
+  end
+
+  def self.settings
+    return @settings if @settings
+
+    begin
+      @settings = Fiona::Settings.new do |s|
+        eval(File.read(File.join(root, 'config', 'settings.rb')))
+      end
+    rescue Exception => e
+      puts "Failed to initialize settings: #{e.message}\n#{e.backtrace.join("\n")}"
+    end
+
+    return @settings
+  end
+
+  def self.db_config
+    return (@db_config ||= YAML.load_file(File.join(root, 'config', 'database.yml'))[env])
+  end
+
+  def self.db_connect
+    return false if db_connected?
+
+    begin
+      ActiveRecord::Base.establish_connection(Totem.db_config)
+    rescue Exception => e
+      puts "Failed to establish DB connection: #{e.message}\n#{e.backtrace.join("\n")}"
+      return false
+    end
+
+    return true
+  end
+
+  def self.db_disconnect
+    return false unless db_connected?
+
+    ActiveRecord::Base.connection_pool.disconnect!
+
+    return true
+  end
+
+  def self.db_reconnect
+    db_disconnect
+    db_connect
+
+    return true
+  end
+
+  def self.db_connected?
+    return !!ActiveRecord::Base.connected?
+  end
+
+  def self.load_app
+    Dir.chdir("#{Totem.root}/app")
+    load("loader.rb")
+    Dir.chdir(Totem.root)
+  end
+
+  def self.component
+    return @component || 'default'
+  end
+
+  def self.component=(val)
+    return @component = val
+  end
+
+  def self.instance
+    return @instance || 0
+  end
+
+  def self.instance=(val)
+    return @instance = val
+  end
+
+  def self.logger=(val)
+    return @logger = val
+  end
+
+  def self.logger
+    return @logger if @logger
+
+    case env
+    when 'development'
+      log_to_stdout
+    when 'production'
+      log_to_file
+    else
+      log_to_stdout
+    end
+
+    return @logger
+  end
+
+  def self.log_to_stdout
+    init_logger($stdout)
+
+    return nil
+  end
+
+  def self.log_to_file
+    init_logger(File.join(root, 'log', "#{env}_#{component}_#{instance}.log"))
+
+    return nil
+  end
+
+  def self.init_logger(output)
+    @logger = Logger.new(output)
+
+    @logger.formatter = proc do |severity, datetime, progname, msg|
+      "#{datetime} :: #{msg}\n"
+    end
+
+    return nil
+  end
+end

data/lib/totem/actable.rb

@@ -0,0 +1,9 @@
+module Totem
+  module Actable
+    def process_event(event)
+      ActiveRecord::Base.connection_pool.with_connection do
+        super
+      end
+    end
+  end
+end

data/lib/totem/connection.rb

@@ -0,0 +1,13 @@
+module Totem
+  class Connection < Tribe::EM::Connection
+    include Totem::Actable
+
+    private
+
+    def initialize(options = {})
+      options[:logger] = Totem.logger
+
+      super
+    end
+  end
+end

data/lib/totem/tasks/console.rb

@@ -0,0 +1,10 @@
+module Totem
+  module Tasks
+    class Console
+      def start
+        ARGV.clear
+        IRB.start
+      end
+    end
+  end
+end

data/lib/totem/tasks/database.rb

@@ -0,0 +1,68 @@
+module Totem
+  module Tasks
+    class Database
+      def initialize
+        @config = Totem.db_config.clone
+      end
+
+      def create
+        options = { :charset => 'utf8', :collation => 'utf8_unicode_ci' }
+
+        create_db = lambda do |config|
+          ActiveRecord::Base.establish_connection(config.merge('database' => nil))
+          ActiveRecord::Base.connection.create_database(config['database'], options)
+          ActiveRecord::Base.establish_connection(config)
+        end
+
+        begin
+          create_db.call(@config)
+        rescue Mysql::Error => sqlerr
+          if sqlerr.errno == 1405
+            print "#{sqlerr.error}. \nPlease provide the root password for your mysql installation\n>"
+            root_password = $stdin.gets.strip
+
+            grant_statement = <<-SQL
+              GRANT ALL PRIVILEGES ON #{config['database']}.*
+                TO '#{config['username']}'@'localhost'
+                IDENTIFIED BY '#{config['password']}' WITH GRANT OPTION;
+            SQL
+
+            create_db.call(@config.merge('database' => nil, 'username' => 'root', 'password' => root_password))
+          else
+            $stderr.puts sqlerr.error
+            $stderr.puts "Couldn't create database for #{config.inspect}, charset: utf8, collation: utf8_unicode_ci"
+            $stderr.puts "(if you set the charset manually, make sure you have a matching collation)" if config['charset']
+          end
+        end
+
+        Totem.db_reconnect
+
+        return true
+      end
+
+      def drop
+        ActiveRecord::Base.connection.drop_database(@config['database'])
+
+        return true
+      end
+
+      def migrate
+        ActiveRecord::Migration.verbose = true
+        ActiveRecord::Migrator.migrate('db/migrate', ENV['VERSION'] ? ENV['VERSION'].to_i : nil)
+
+        Totem.db_reconnect
+
+        return true
+      end
+
+      def rollback
+        step = ENV['STEP'] ? ENV['STEP'].to_i : 1
+        ActiveRecord::Migrator.rollback('db/migrate', step)
+
+        Totem.db_reconnect
+
+        return true
+      end
+    end
+  end
+end

data/lib/totem/tasks/generator.rb

@@ -0,0 +1,29 @@
+module Totem
+  module Tasks
+    class Generator
+      def migration(name)
+        tstamp =  Time.now.utc.strftime("%Y%m%d%H%M%S")
+        fname = "#{tstamp}_#{name}.rb"
+        path = File.join(Totem.root, 'db', 'migrate', fname)
+
+        name || raise('You must specify a name')
+
+        puts "Creating migration: #{path}"
+
+        if File.exists?(path)
+          puts 'ERROR: File already exists.'
+          return
+        end
+
+        content = <<-EOS.unindent
+          class #{name.camelize} < ActiveRecord::Migration
+          end
+        EOS
+
+        File.open(path, 'w') { |f| f.write(content) }
+
+        return nil
+      end
+    end
+  end
+end

data/lib/totem/tcp_server.rb

@@ -0,0 +1,13 @@
+module Totem
+  class TcpServer < Tribe::EM::TcpServer
+    include Totem::Actable
+
+    private
+
+    def initialize(ip, port, conn_class, options = {})
+      options[:logger] = Totem.logger
+
+      super
+    end
+  end
+end

data/lib/totem/version.rb

@@ -0,0 +1,3 @@
+module Totem
+  VERSION = '0.0.1'
+end

data/totem.gemspec

@@ -0,0 +1,22 @@
+# -*- encoding: utf-8 -*-
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'totem/version'
+
+Gem::Specification.new do |gem|
+  gem.name          = 'totem'
+  gem.version       = Totem::VERSION
+  gem.authors       = ['Chad Remesch']
+  gem.email         = ['chad@remesch.com']
+  gem.description   = %q{The Totem framework.}
+  gem.summary       = %q{An actor model based framework for rapid server server side development.}
+  gem.homepage      = 'https://github.com/chadrem/totem'
+
+  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_dependency('tribe', '0.0.8')
+  gem.add_dependency('tribe_em', '0.0.2')
+end

metadata

@@ -0,0 +1,91 @@
+--- !ruby/object:Gem::Specification
+name: totem
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Chad Remesch
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-01-30 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: tribe
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.0.8
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.0.8
+- !ruby/object:Gem::Dependency
+  name: tribe_em
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.0.2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.0.2
+description: The Totem framework.
+email:
+- chad@remesch.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- lib/totem.rb
+- lib/totem/actable.rb
+- lib/totem/connection.rb
+- lib/totem/tasks/console.rb
+- lib/totem/tasks/database.rb
+- lib/totem/tasks/generator.rb
+- lib/totem/tcp_server.rb
+- lib/totem/version.rb
+- totem.gemspec
+homepage: https://github.com/chadrem/totem
+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.24
+signing_key: 
+specification_version: 3
+summary: An actor model based framework for rapid server server side development.
+test_files: []