Doing some improvements on the op_endpoint generation in the identity…

… page app
roman · Mar 23, 2009 · 426d0dd · 426d0dd
1 parent 2cd19d1
commit 426d0dd
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/lib/rots/identity_page_app.rb b/lib/rots/identity_page_app.rb
@@ -11,14 +11,13 @@ def initialize(config, server_options)
   end
 
   def call(env)
-    request = Rack::Request.new(env)
-    flag = request.params['openid.success'] == 'true' ? '?openid.success=true' : ''
+    @request = Rack::Request.new(env)
     Rack::Response.new do |response|
       response.write <<-HERE
 <html>
   <head>
-  <link rel="openid2.provider" href="http://localhost:#{@server_options[:port]}/server/#{flag}" />
-  <link rel="openid.server" href="http://localhost:#{@server_options[:port]}/server/#{flag}" />
+  <link rel="openid2.provider" href="#{op_endpoint}" />
+  <link rel="openid.server" href="#{op_endpoint}" />
   </head>
   <body>
     <h1>This is #{@config['identity']} identity page</h1>
@@ -28,4 +27,10 @@ def call(env)
     end.finish
   end
 
+  def op_endpoint
+    "http://%s:%d/server/%s" % [@request.host, 
+                           @request.port, 
+                           (@request.params['openid.success'] ? '?openid.success=true' : '')]
+  end
+
 end