pione 0.1.0

data/lib/pione/command/pione-relay-client-db.rb

@@ -0,0 +1,80 @@
+module Pione
+  module Command
+    class PioneRelayClientDB < BasicCommand
+      set_program_name("pione-relay-client-db")
+
+      set_program_message <<TXT
+Add, delete, or show clients in this relay server.
+TXT
+
+      define_option("-a", "--add", "add a client") do
+        @action = :add
+      end
+
+      define_option("-d", "--delete", "delete a client") do
+        @action = :delete
+      end
+
+      define_option("-l", "--list", "list clients") do
+        @action = :list
+      end
+
+      define_option("-u name", "--user name", "user name") do |name|
+        @name = name
+      end
+
+      define_option("-p password", "--password password", "password") do |password|
+        @password = password
+      end
+
+      define_option("-f path", "--file path", "client db path") do |path|
+        Global.relay_client_db_path = path
+      end
+
+      def initialize
+        @action = nil
+        @name = nil
+        @password = nil
+      end
+
+      def validate_options
+        abort("error: -a, -d, or -l") unless @action
+      end
+
+      def prepare
+        # set client db
+        Global.relay_client_db = RelayClientDB.new(Global.relay_client_db_path)
+
+        # set name
+        if [:add, :delete].include?(@action)
+          @name = HighLine.new.ask("Name: ") unless @name
+        end
+
+        # set password
+        if @action == :add
+          unless @password
+            @password = HighLine.new.ask("Password: "){|q| q.echo = "*"}
+            password = HighLine.new.ask("Re-enter password: "){|q| q.echo = "*"}
+            abort("error: password mismatch") unless @password == password
+          end
+        end
+      end
+
+      def start
+        case @action
+        when :add
+          Global.relay_client_db.add(@name, @password)
+          Global.relay_client_db.save
+        when :delete
+          Global.relay_client_db.delete(@name)
+          Global.relay_client_db.save
+        when :list
+          names = Global.relay_client_db.names
+          puts "%s clients in %s" % [names.size, Global.relay_client_db_path]
+          names.each {|user| puts user}
+        end
+        terminate
+      end
+    end
+  end
+end

data/lib/pione/command/pione-relay.rb

@@ -0,0 +1,47 @@
+module Pione
+  module Command
+    class PioneRelay < FrontOwnerCommand
+      set_program_name("pione-relay") do
+        "--relay-port %s" % [@relay_port]
+      end
+
+      set_program_message <<TXT
+Runs relay process for connecting between clients and brokers.
+TXT
+
+      define_option("--realm name", "set relay realm name for client authentification") do |name|
+        Global.relay_realm = name
+      end
+
+      define_option("--relay-port port", "set relay port") do |port|
+        @relay_port = port
+      end
+
+      def initialize
+        @relay_port = Global.relay_port
+      end
+
+      def validate_options
+        abort("error: no realm name") if Global.relay_realm.nil? or Global.relay_realm.empty?
+        abort("error: no relay port") unless @relay_port
+      end
+
+      def create_front
+        Front::RelayFront.new(self)
+      end
+
+      def start
+        # wake up tuple space provider process
+        Pione::TupleSpaceProvider.instance
+
+        puts DRb.front.uri
+        DRb::DRbServer.new(
+          "relay://:%s" % @relay_port,
+          nil,
+          {:SSLCertName => Global.relay_ssl_certname}
+        )
+        DRb.thread.join
+      end
+    end
+  end
+end

data/lib/pione/command/pione-syntax-checker.rb

@@ -0,0 +1,103 @@
+module Pione
+  module Command
+    class PioneSyntaxChecker < BasicCommand
+      set_program_name("pione-syntax-checker")
+
+      set_program_message <<TXT
+Prints syntax tree of PIONE notation.
+TXT
+
+      define_option('-e', '--expr=expr', 'expression string') do |e|
+        @expr = e
+        @readline_mode = false
+      end
+
+      define_option('-t', '--transformer', 'show transformer result') do
+        @transform = true
+      end
+
+      def initialize
+        @readline_mode = true
+      end
+
+      def prepare
+        require 'pp'
+        @history = File.join(Global.dot_pione_dir, "pione-history")
+        trap_int
+      end
+
+      def start
+        if @readline_mode
+          require 'readline'
+          restore_history
+
+          # start loop
+          while buf = Readline.readline(Terminal.red("> "), true)
+            if /[^\s]/.match(buf)
+              # don't record if previous line is the same
+              if Readline::HISTORY.size > 1 && Readline::HISTORY[-2] == buf
+                Readline::HISTORY.pop
+              end
+              # print parsing result
+              print_result(buf)
+            else
+              # don't record if it is an empty line
+              Readline::HISTORY.pop
+            end
+          end
+        else
+          # print parsing result
+          print_result(@expr)
+        end
+      end
+
+      private
+
+      # Prints parsing result of the string
+      def print_result(str)
+        begin
+          puts Terminal.green("syntax:")
+          stree = DocumentParser.new.parse(str).first
+          pp stree
+          if @transform
+            puts Terminal.green("model:")
+            pp DocumentTransformer.new.apply(stree)
+          end
+        rescue Pione::Parser::ParserError, Parslet::UnconsumedInput, Parslet::ParseFailed => e
+          msg = "Pione syntax error: %s (%s)" % [e.message, e.class.name]
+          @readline_mode ? puts(msg) : abort(msg)
+        rescue Pione::Model::PioneModelTypeError,
+          Pione::Model::VariableBindingError => e
+          msg = "Pione model error: %s (%s)" % [e.message, e.class.name]
+          @readline_mode ? puts(msg) : abort(msg)
+        end
+      end
+
+      # Makes trap Ctr+C
+      def trap_int
+        trap("INT") do
+          save_history
+          exit
+        end
+      end
+
+      # Restores history.
+      def restore_history
+        if File.exist?(@history)
+          File.readlines(@history).reverse.each do |line|
+            Readline::HISTORY.push line.chomp
+          end
+        end
+      end
+
+      # Saves history.
+      def save_history
+        File.open(@history, "w+") do |file|
+          Readline::HISTORY.to_a.reverse[0..100].each do |line|
+            file.puts(line) if /\S/.match(line)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/pione/command/pione-task-worker.rb

@@ -0,0 +1,123 @@
+module Pione
+  module Command
+    # This is a class for +pione-task-worker+ command. +pione-task-worker+
+    # starts a task worker agent with tuple space server.
+    class PioneTaskWorker < ChildProcess
+      set_program_name("pione-task-worker") do
+        parent_front = @no_parent_mode ? "nil" : @parent_front.uri
+        "<front=%s, parent-front=%s>" % [Global.front.uri, parent_front]
+      end
+
+      set_program_message <<TXT
+Runs task worker process. This command is launched by other processes like
+pione-client or pione-broker.
+TXT
+
+      use_option_module CommandOption::ChildProcessOption
+
+      # --connection-id
+      define_option('--connection-id=ID', 'set connection id') do |id|
+        @connection_id = id
+      end
+
+      # --feature
+      define_option('--features="FEATURES"', 'set features') do |features|
+        begin
+          features = DocumentTransformer.new.apply(
+            DocumentParser.new.feature_expr.parse(features)
+          )
+          @features = features
+        rescue Parslet::ParseFailed => e
+          puts "invalid parameters: " + str
+          Util::ErrorReport.print(e)
+          abort
+        end
+      end
+
+      attr_reader :connection_id
+      attr_reader :agent
+      attr_reader :tuple_space_server
+
+      private
+
+      def create_front
+        Pione::Front::TaskWorkerFront.new(self)
+      end
+
+      def validate_options
+        super
+
+        # check requisite options
+        if @connection_id.nil?
+          abort("error: no connection id")
+        end
+
+        # get the parent front server
+        begin
+          @parent_front.uuid
+        rescue => e
+          abort("pione-task-worker cannot get the parent front server: %s" % e)
+        end
+      end
+
+      def prepare
+        super
+
+        @tuple_space_server = @parent_front.get_tuple_space_server(@connection_id)
+        @features = Model::Feature::EmptyFeature.new unless @features
+        @agent = Pione::Agent[:task_worker].new(@tuple_space_server, @features)
+        @command_listener = Pione::Agent[:command_listener].new(@tuple_space_server, self)
+
+        # connect caller front
+        @parent_front.add_task_worker_front(Global.front, @connection_id)
+
+        abort("pione-task-worker error: no tuple space server") unless @tuple_space_server
+
+        # get base uri
+        if @tuple_space_server.base_uri.scheme == "dropbox"
+          Resource::Dropbox.init(@tuple_space_server)
+          unless Resource::Dropbox.ready?
+            abort("You aren't ready to access Dropbox.")
+          end
+        end
+      end
+
+      def start
+        super
+
+        # start task worker activity
+        @agent.start
+        @command_listener.start
+
+        # wait...
+        begin
+          @agent.running_thread.join
+          # terminate
+          terminate
+        rescue DRb::DRbConnError, DRb::ReplyReaderThreadError
+          terminate
+        end
+      end
+
+      def terminate
+        return if @terminated
+        @agent.terminate
+
+        while true
+          break if @agent.terminated? and @agent.running_thread.stop?
+          sleep 1
+        end
+
+        # disconnect parent front
+        @parent_front.remove_task_worker_front(self, @connection_id)
+
+        # flag
+        @terminated = true
+
+        super
+      rescue DRb::DRbConnError, DRb::ReplyReaderThreadError
+        abort
+      end
+    end
+  end
+end

data/lib/pione/command/pione-tuple-space-provider.rb

@@ -0,0 +1,87 @@
+module Pione
+  module Command
+    # PioneTupleSpaceProvider is for +pione-tuple-space-provider+ command.
+    class PioneTupleSpaceProvider < ChildProcess
+      set_program_name("pione-tuple-space-provider") do
+        parent_front = @no_parent_mode ? "nil" : @parent_front.uri
+        "<front=%s, parent-front=%s>" % [Global.front.uri, parent_front]
+      end
+
+      set_program_message <<TXT
+Runs tuple space provider process for sending tuple space presence
+notifier. This command is launched by other processes like pione-client or
+pione-relay normally, but you can force to start by calling with --no-parent
+option.
+TXT
+
+      use_option_module CommandOption::TupleSpaceProviderOption
+
+      attr_reader :tuple_space_provider
+
+      def initialize
+        super
+        @parent_front = nil
+        @notifier_addresses = []
+      end
+
+      # @api private
+      def validate_options
+        super
+
+        # broadcast addresses
+        @notifier_addresses.each do |uri|
+          unless uri.scheme == "broadcast"
+            abort("error: invalid broadcast address '%s'" % uri.to_s)
+          end
+        end
+      end
+
+      # @api private
+      def create_front
+        Pione::Front::TupleSpaceProviderFront.new(self)
+      end
+
+      # @api private
+      def prepare
+        super
+
+        # setup notifier addresses
+        unless @notifier_addresses.empty?
+          Global.tuple_space_provider_broadcast_addresses = @notifier_addresses
+        end
+
+        # make tuple space provider
+        @tuple_space_provider = TupleSpaceProvider.new
+      end
+
+      # @api private
+      def start
+        super
+
+        # start provider activity
+        @tuple_space_provider.start
+
+        # set my URI to caller front as its provider
+        unless @no_parent_mode
+          @parent_front.set_tuple_space_provider(Global.front.uri)
+        end
+
+        # wait
+        begin
+          DRb.thread.join
+        rescue DRb::DRbConnError, DRb::ReplyReaderThreadError
+          terminate
+        end
+      end
+
+      # @api private
+      def terminate
+        puts "terminate %s" % program_name
+        @tuple_space_provider.terminate
+        super
+      rescue DRb::DRbConnError, DRb::ReplyReaderThreadError
+        abort
+      end
+    end
+  end
+end

data/lib/pione/command/pione-tuple-space-receiver.rb

@@ -0,0 +1,55 @@
+module Pione
+  module Command
+    class PioneTupleSpaceReceiver < ChildProcess
+      set_program_name("pione-tuple-space-receiver") do
+        parent_front = @no_parent_mode ? "nil" : @parent_front.uri
+        "<front=%s, parent=%s>" % [Global.front.uri, parent_front]
+      end
+
+      set_program_message <<TXT
+Runs tuple space receiver process for receiving tuple space presence
+notifier. This command is launched by other processes like pione-broker
+normally, but you can force to start by calling with --no-parent option.
+TXT
+
+      use_option_module CommandOption::TupleSpaceReceiverOption
+
+      attr_reader :tuple_space_receiver
+
+      def create_front
+        Front::TupleSpaceReceiverFront.new(self)
+      end
+
+      def prepare
+        super
+        @tuple_space_receiver = TupleSpaceReceiver.new
+      end
+
+      def start
+        super
+
+        # start provider activity
+        @tuple_space_receiver.start
+
+        # set my uri to parent front as its provider
+        unless @no_parent_mode
+          @parent_front.set_tuple_space_receiver(Global.front.uri)
+        end
+
+        # wait
+        DRb.thread.join
+      rescue DRb::ReplyReaderThreadError => e
+        # ignore reply reader error
+      end
+
+      def terminate
+        puts "terminate %s" % program_name
+        begin
+          @tuple_space_receiver.terminate
+        rescue DRb::DRbConnError
+        end
+        super
+      end
+    end
+  end
+end