lib/kamerling/registrar.rb in kamerling-0.0.2 vs lib/kamerling/registrar.rb in kamerling-0.0.3

@@ -1,9 +1,46 @@
-module Kamerling class Registrar
-  def register addr: req(:addr), client_uuid: req(:client_uuid),
-               project_uuid: req(:project_uuid), repos: Repos
-    client  = repos[Client][client_uuid]
-    project = repos[Project][project_uuid]
-    reg     = Registration.new addr: addr, client: client, project: project
-    repos[Registration] << reg
+require_relative 'client'
+require_relative 'project'
+require_relative 'registration'
+require_relative 'repos'
+require_relative 'uuid'
+
+module Kamerling
+  class Registrar
+    def self.register(addr:, message:, repos: Repos)
+      new(addr: addr, message: message, repos: repos).register
+    end
+
+    def initialize(addr:, message:, repos:)
+      @addr, @message, @repos = addr, message, repos
+    end
+
+    def register
+      client.addr = addr
+      repos << client
+      repos << registration
+    end
+
+    attr_reader :addr, :message, :repos
+    private     :addr, :message, :repos
+
+    private
+
+    def client
+      @client ||= find_or_create_client
+    end
+
+    def find_or_create_client
+      repos[Client][message.client_uuid]
+    rescue Repo::NotFound
+      Client.new(addr: addr, uuid: message.client_uuid)
+    end
+
+    def project
+      @project ||= repos[Project][message.project_uuid]
+    end
+
+    def registration
+      Registration.new(addr: addr, client: client, project: project)
+    end
   end
-end end
+end