swee 0.0.1

data/lib/swee/exception.rb

@@ -0,0 +1,2 @@
+class OptionsError < ArgumentError
+end

data/lib/swee/helper.rb

@@ -0,0 +1,29 @@
+require "erb"
+# require "cgi"
+
+module Swee
+
+  module Helper
+    def raw text
+      text.html_safe
+    end
+
+    def image_tag
+
+    end
+
+    def form_for
+
+    end
+
+    def tag(name, options = nil, open = false, escape = true)
+
+      "<#{name}#{tag_options(options, escape) if options}#{open ? ">" : " />"}".html_safe
+    end
+
+    def parse_tag_options options
+
+    end
+
+  end
+end

data/lib/swee/installer.rb

@@ -0,0 +1,58 @@
+module Swee
+  module Installer
+    module_function
+    def run proj_name
+
+      # 打印安装信息
+      puts "--------------- Swee 安装 -----------------"
+      proj_dir = "#{Dir.pwd}/#{proj_name}"
+
+      # 创建项目目录
+      if Dir.exist?(proj_dir)
+        puts "当前路径已存在#{proj_name}目录"
+        return
+      end
+      puts "#{proj_name}目录创建成功"
+      Dir.mkdir(proj_dir)
+
+      # 安装目录
+      app_path = proj_dir
+      ["public","controllers","logs","models","tmp","views","views/home"].each do |dir|
+        _dir = app_path + "/#{dir}"
+        if Dir.exist?(_dir)
+          puts "#{_dir}目录已存在,略过..."
+        else
+          Dir.mkdir(_dir)
+          puts "#{_dir}目录成功创建"
+        end
+      end
+
+      # 安装 文件
+      ["config.rb","routes.rb","public/404.html","controllers/HomeController.rb", "views/home/index.erb"].each do |_file|
+        file = app_path + "/#{_file}"
+
+        if File.exist?(file)
+          puts "#{file} 已经安装过了"
+        else
+          f = File.open(file,"w")
+          config_file = File.expand_path("../template/#{_file}",File.dirname(__FILE__))
+          f.write(File.read(config_file))
+          f.close
+          puts "#{file} 已经安装"
+        end
+      end
+
+      # 打印结束信息
+      puts "恭喜,全部安装完成"
+      puts "您可以先输入: swee s 来启动服务器"
+      puts "请打开config.rb 和 routes.rb 配置所需的参数"
+      puts "更多参数请输入: swee --help"
+      puts ">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>"
+      puts "我们先启动一个欢迎页面"
+      puts "浏览器输入 http://localhost:3000 来访问欢迎页面吧"
+      puts "按 Ctrl + C 关闭服务"
+      puts ">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>"
+      `cd #{proj_name} && swee s`
+    end
+  end
+end

data/lib/swee/lodder.rb

@@ -0,0 +1,92 @@
+module Swee
+  module Lodder
+    @@mtime_files_cache = {}
+
+    CACHE_FILE_MTIME_DIR = ["models","controllers"]
+    EXTENSION_NAMES = [".rb",".erb",".haml",".slim","ymal"]
+
+    module_function
+
+    def cache_file_mtime
+      CACHE_FILE_MTIME_DIR.each { |d| Lodder.search_app_file(d) }
+    end
+
+    def mtime_files
+      @@mtime_files_cache
+    end
+
+    # 递归寻找目录下所有文件
+    # 保存为如下结构(用于 代码修改 reload)
+    # filename => mtime
+    def search_app_file dir
+      app_path = ENV["app_path"]
+      Dir.glob("#{app_path}/#{dir}/*") do |file|
+        if EXTENSION_NAMES.include? File.extname(file).downcase
+          @@mtime_files_cache[file] = File.mtime(file)
+        else
+          if File.directory?(file)
+            search_app_file(dir + "/" + file.split("/").last)
+          end
+        end
+      end
+    end
+
+    def base_require
+      require_relative './support'
+      require_relative './config'
+      # require_relative './application'
+
+      # patches
+      require_relative './patches/logger.rb'
+
+      # app
+      require_relative './routes'
+      require_relative './helper'
+      require_relative './controller_filter'
+      require_relative './controller'
+      require_relative './view'
+      require_relative './app_executor'
+
+      # middlewaves
+      require_relative './middlewaves/content_length'
+      require_relative './middlewaves/common_logger'
+      require_relative './middlewaves/reloader'
+
+      # thin
+      require_relative './thin/headers'
+      require_relative './thin/request'
+      require_relative './thin/response'
+
+      # server
+      require_relative './swee_logger'
+      require_relative './daemonize'
+      require_relative './connection'
+      require_relative './server'
+      require_relative './exception'
+
+    end
+
+    # 用户配置
+    def app_require
+      # require_relative './application'
+      begin
+        require File.expand_path('config', ENV["app_path"])
+      rescue LoadError
+        raise "未找到config.rb" 
+      end
+    end
+
+    # 条件读取
+    def conditional_require
+      # 存在 AR 读取
+      unless Gem.find_files("active_record").empty? 
+        require 'active_record'
+      end
+    end
+
+    def all
+      conditional_require
+      base_require
+    end
+  end
+end

data/lib/swee/middlewaves/common_logger.rb

@@ -0,0 +1,29 @@
+module Rack
+  class CommonLogger
+    def initialize(app, logger=nil)
+      @app = app
+      @logger = logger
+    end
+
+    private
+
+    def log(env, status, header, began_at)
+      now = Time.now
+      length = extract_content_length(header)
+
+      msg = FORMAT % [
+        env['HTTP_X_FORWARDED_FOR'] || env["REMOTE_ADDR"] || "-",
+        env["REMOTE_USER"] || "-",
+        now.strftime("%d/%b/%Y:%H:%M:%S %z"),
+        env['REQUEST_METHOD'],
+        env['PATH_INFO'],
+        env['QUERY_STRING'].empty? ? "" : "?"+env['QUERY_STRING'],
+        env["SERVER_PROTOCOL"],
+        status.to_s[0..3],
+        length,
+        now - began_at ]
+
+      @logger << msg
+    end
+  end
+end

data/lib/swee/middlewaves/content_length.rb

@@ -0,0 +1,21 @@
+module Swee
+
+  class ContentLength
+    def initialize(app)
+      @app = app
+    end
+
+    def call(env)
+      status, headers, body = @app.call(env)
+      headers.merge!( { "Content-Length" => calc_length(body) } )
+      [status, headers, body]
+    end
+
+    private
+    def calc_length(body)
+      length = 0
+      body.each { |part| length += part.bytesize }
+      length
+    end
+  end
+end

data/lib/swee/middlewaves/reloader.rb

@@ -0,0 +1,24 @@
+module Swee
+
+  class Reloader
+    def initialize(app,logger)
+      @app = app
+      @logger = logger
+      app_path = ENV["app_path"]
+    end
+
+    def call(env)
+      # 快速遍历 文件 mtime cache表
+      # mtime不一致则重新 load
+      Lodder.mtime_files.each_pair do |file,omtime|
+        mtime = File.mtime(file)
+        if mtime != omtime
+          load file
+          Lodder.mtime_files[file] = mtime
+          @logger << "#{file}文件已重载!"
+        end
+      end
+      return @app.call(env)
+    end
+  end
+end

data/lib/swee/patches/logger.rb

@@ -0,0 +1,15 @@
+require "logger" if !defined? Logger
+
+class Logger
+
+  # 判断当前日志是否输出到stdout
+  def io?
+    instance_variable_get(:@logdev).dev === STDOUT
+  end
+
+
+  # 判断当前日志输出到logfile
+  def file?
+    instance_variable_get(:@logdev).dev.is_a?(File)
+  end
+end

data/lib/swee/routes.rb

@@ -0,0 +1,46 @@
+module Swee
+  class Routes
+    @@tables = {}
+    class RouteStruct
+      attr_reader :controller, :action, :request_methods
+      def initialize c,a,m
+        @controller,@action,@request_methods = c,a,m
+      end
+
+      def controller_name
+        "#{controller[0].upcase+controller[1..controller.size-1]}Controller"
+      end
+
+      def create_controller_instance
+        eval "#{controller_name}.new"
+      end
+    end
+    class << self
+
+      def get *args
+        self._parse "get",*args
+      end
+
+      def post *args
+        self._parse "post",*args
+      end
+
+      def match *args
+        self._parse "match",*args
+      end
+
+      def tables
+        @@tables
+      end
+
+      def _parse _m,*args
+        _path_info = args[0].to_s
+        if args[1] =~ /^(.*)#(.*)$/
+          @@tables[_path_info] = RouteStruct.new $1,$2, _m == "match" ? args[2][:via].map(&:to_sym) : [_m.to_sym]
+        else
+          raise "routes error!"
+        end
+      end
+    end
+  end
+end

data/lib/swee/server.rb

@@ -0,0 +1,298 @@
+module Swee
+  class Server
+    include Daemonize
+
+    attr_reader :code_reload
+
+    def initialize
+      @config = Swee.config
+      @options = @config.server
+
+      @signature = nil
+
+      @handle_request_mode = @options[:handle_request_mode]
+      
+      @restart_mode = @options[:restart_mode]
+      @pid_file   = File.expand_path(@options[:pid_file],ENV["app_path"])
+      @touch_file = File.expand_path(@options[:touch_file],ENV["app_path"])
+
+      @logger                         = nil
+      @log_file_options               = @options[:log_file]
+      @log_file                       = File.expand_path(@log_file_options.first,ENV["app_path"])
+      @log_file_options[0]            = @log_file
+
+      @logger_level                   = @options[:logger_level]
+
+      @code_reload                    = @options[:code_reload]
+
+      # 最大连接数 和 稳定连接处
+      @maximum_connections            = @options[:max_connections]
+      @maximum_persistent_connections = @options[:max_connections]
+
+      # 监听端口
+      @listen = @options[:listen]
+
+      # 实际连接 __id__ => connection 
+      @connections                    = {}
+
+      # 超时时间
+      @timeout                        = 30
+
+      # 活动连接数
+      @persistent_connection_count    = 0
+
+      # touch 模式重启时间
+      @restart_time = nil
+
+      # 信号队列 暂时只处理 :USR2
+      @signal_queue = []
+
+      # 守护进程
+      @daemonize = @options[:run_background]
+
+      # todo: 性能监控
+      # @performance_monitoring = @options[:performance_monitoring]
+    end
+
+    # 获取 Logger STDOUT 和 File
+    def logger
+      @logger
+    end
+
+    # 删除pid文件 用于重启 和 exit
+    def remove_pid_file
+      File.delete(@pid_file) if @pid_file && File.exists?(@pid_file)
+    end
+
+    # 读取pid文件 用于 USR2 信号获得后和Process.pid 比对
+    def read_pid_file
+      File.read(@pid_file)
+    end
+
+    # 写入Process.pid到 pid 文件
+    def write_pid_file
+      open(@pid_file,"w") { |f| f.write(Process.pid) }
+      File.chmod(0644, @pid_file)
+    end
+
+    # 是否运行标志
+    def running?
+      @running
+    end
+
+    # 强制停止
+    def stop!
+      @running  = false
+      disconnect
+      EventMachine.stop
+      @connections.each_value { |connection| connection.close_connection }
+    end
+
+    # 放入 注册的 exit_at 函数中
+    # 重启 停止服务 -> 删除pid文件 -> 退出进程
+    #
+    def restart
+      stop!
+      remove_pid_file
+      exit
+    end
+
+    # 处理日志 stdout 和 file 两种
+    def handle_logger
+      # 创建日志文件
+      if !File.exist?(@log_file)
+        FileUtils.mkdir_p File.dirname(@log_file)
+      end
+      logs = []
+      logs << Logger.new(*@log_file_options)
+      unless @daemonize
+        logs << Logger.new(STDOUT)
+      end
+
+      # todo: 日志等级, datetime, 格式配置
+      # logs.each do |_logger|
+      #   _logger.level = @logger_level
+
+      #   _logger.datetime_format = '%Y-%m-%d %H:%M:%S'
+      #   _logger.formatter = proc do |severity, datetime, progname, msg|
+      #     "Started GET #{msg} at #{datetime}"
+      #   end
+      # end
+
+      @logger = SweeLogger.new
+
+      logs.each { |_logger| @logger.addlog _logger }
+    end
+
+    # 处理重启
+    def handle_restart
+      send "handle_" + @restart_mode.to_s + "_restart"
+    end
+
+    # 处理touch重启方式
+    def handle_touch_restart
+      @restart_time = File.mtime(@touch_file)
+    end
+
+    # 处理pid重启方式
+    def handle_pid_restart
+      write_pid_file
+    end
+
+    # 判断是否为 touch 重启方式
+    def touch_mode?
+      @restart_mode == :touch
+    end
+
+    # 判断是否为 USR2 pid 重启方式 
+    def pid_mode?
+      @restart_mode == :pid
+    end
+
+    # 判断 pid 文件是否为 当前进程ID
+    def current_pid? pid
+      read_pid_file == Process.pid.to_s
+    end
+
+    # 追踪重启 EM Timer 方式
+    # 每秒执行一次
+    # touch: 每秒读取touch文件 mtime
+    # pid: trap usr2 信号放入 信号队列, 然后每秒追踪队列变化获取信号
+    def trace_restart
+      if touch_mode?
+        EM.add_periodic_timer(1) {
+          mtime = File.mtime(@touch_file)
+          if mtime != @restart_time
+            @restart_time = mtime
+            puts "重启了"
+            restart
+            # 放弃 next_tick 原因：会在下次有请求时处理重启
+            # EM.next_tick { restart }
+          end
+        }
+      end
+
+      if pid_mode?
+        EM.add_periodic_timer(1) {
+          signal = @signal_queue.shift
+          case signal
+          when nil
+          when :QUIT # graceful shutdown
+          when :TERM, :INT # immediate shutdown
+            stop!
+          when :USR1 # rotate logs
+          when :USR2 # exec binary, stay alive in case something went wrong
+            _restart = current_pid?
+          when :WINCH
+          when :TTIN
+          when :TTOU
+          when :HUP
+          end
+          restart if _restart
+        }
+        trap(:USR2) {
+          # 收到 USR2信号加入 信号队列
+          @signal_queue << :USR2
+        }
+      end
+    end
+
+    # 处理守护进程(后台运行)
+    # 创建子进程
+    # 进程命名
+    # 注册守护进程重启
+    def handle_daemonize
+      # 先删除pid文件
+      remove_pid_file
+
+      # todo 关闭io 管道
+      # rd, wr = IO.pipe
+      # grandparent = $$
+
+      # 创建子进程
+      safefork && exit
+
+      # 设定进程名称
+      $0 = "swee"
+      
+      # 重新创建pid文件
+      write_pid_file
+
+      # 注册守护进程重启
+      at_exit{
+        # 简单的用异常判断 系统退出，并非其他异常 获得重启
+        if $!.class == SystemExit
+          @logger << "Swee服务器重启!"
+          remove_pid_file
+          ::Swee::Engine.restart_server!
+        end
+      }
+    end
+
+    # 创建日志 和 pid 文件
+    def create_touch_and_pid_file
+      [@pid_file,@touch_file].each do |_file|
+        if !File.exist?(_file)
+          FileUtils.mkdir_p File.dirname(_file)
+          open(_file,"w")
+          File.chmod(0644, _file)
+        end
+      end
+    end
+
+    # 启动服务器
+    def run!
+      # 创建日志和pid文件
+      create_touch_and_pid_file
+      # 处理日志
+      handle_logger
+      # 处理重启
+      handle_restart
+      # EM默认配置
+      EventMachine.threadpool_size = 20
+      EventMachine.epoll
+      EventMachine.set_descriptor_table_size(@maximum_connections)
+
+      @logger << "正在启动swee服务器"
+
+      EventMachine::run {
+        @signature = connection
+        @running = true
+
+        # EM启动后处理后台守护进程运行
+        handle_daemonize if @daemonize
+
+        # 追踪重启文件
+        trace_restart
+
+        puts "swee 服务器已启动, 端口:#{@listen}"
+      }
+    end
+
+    # 连接
+    def connection
+      EventMachine::start_server "0.0.0.0", @listen, Connection do |_connection|
+        _connection.server                  = self
+
+        # 记录活动连接数
+        if @persistent_connection_count < @maximum_persistent_connections
+          @persistent_connection_count += 1
+        end
+        @connections[_connection.__id__] = _connection
+      end
+    end
+
+    # 断开
+    def disconnect
+      EventMachine.stop_server(@signature)
+    end
+
+    # 完成连接
+    def connection_finished(connection)
+      @persistent_connection_count -= 1
+      @connections.delete(connection.__id__)
+      
+      # TODO: 停止或重启
+    end
+  end
+end