summaryrefslogtreecommitdiff
path: root/lib/soap/rpc/soaplet.rb
blob: ec172fd85e876119328fab23ec3ad5fb2ac3c2fb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
# SOAP4R - SOAP handler servlet for WEBrick
# Copyright (C) 2001, 2002, 2003  NAKAMURA, Hiroshi <nahi@ruby-lang.org>.

# This program is copyrighted free software by NAKAMURA, Hiroshi.  You can
# redistribute it and/or modify it under the same terms of Ruby's license;
# either the dual license version in 2003, or any later version.


require 'webrick/httpservlet/abstract'
require 'webrick/httpstatus'
require 'soap/rpc/router'
require 'soap/streamHandler'

module SOAP
module RPC


class SOAPlet < WEBrick::HTTPServlet::AbstractServlet
public
  attr_reader :app_scope_router

  def initialize
    @router_map = {}
    @app_scope_router = ::SOAP::RPC::Router.new(self.class.name)
  end

  # Add servant klass whose object has request scope.  A servant object is
  # instantiated for each request.
  #
  # Bare in mind that servant klasses are distinguished by HTTP SOAPAction
  # header in request.  Client which calls request-scoped servant must have a
  # SOAPAction header which is a namespace of the servant klass.
  # I mean, use Driver#add_method_with_soapaction instead of Driver#add_method
  # at client side.
  #
  def add_rpc_request_servant(klass, namespace, mapping_registry = nil)
    router = RequestRouter.new(klass, namespace, mapping_registry)
    add_router(namespace, router)
  end

  # Add servant object which has application scope.
  def add_rpc_servant(obj, namespace)
    router = @app_scope_router
    SOAPlet.add_servant_to_router(router, obj, namespace)
    add_router(namespace, router)
  end
  alias add_servant add_rpc_servant


  ###
  ## Servlet interfaces for WEBrick.
  #
  def get_instance(config, *options)
    @config = config
    self
  end

  def require_path_info?
    false
  end

  def do_GET(req, res)
    res.header['Allow'] = 'POST'
    raise WEBrick::HTTPStatus::MethodNotAllowed, "GET request not allowed."
  end

  def do_POST(req, res)
    namespace = parse_soapaction(req.meta_vars['HTTP_SOAPACTION'])
    router = lookup_router(namespace)

    is_fault = false

    charset = ::SOAP::StreamHandler.parse_media_type(req['content-type'])
    begin
      response_stream, is_fault = router.route(req.body, charset)
    rescue Exception => e
      response_stream = router.create_fault_response(e)
      is_fault = true
    end

    res.body = response_stream
    res['content-type'] = "text/xml; charset=\"#{charset}\""
    if response_stream.is_a?(IO)
      res.chunked = true
    end

    if is_fault
      res.status = WEBrick::HTTPStatus::RC_INTERNAL_SERVER_ERROR
    end
  end

private

  class RequestRouter < ::SOAP::RPC::Router
    def initialize(klass, namespace, mapping_registry = nil)
      super(namespace)
      if mapping_registry
	self.mapping_registry = mapping_registry
      end
      @klass = klass
      @namespace = namespace
    end

    def route(soap_string)
      obj = @klass.new
      namespace = self.actor
      router = ::SOAP::RPC::Router.new(@namespace)
      SOAPlet.add_servant_to_router(router, obj, namespace)
      router.route(soap_string)
    end
  end

  def add_router(namespace, router)
    @router_map[namespace] = router
  end

  def parse_soapaction(soapaction)
    if /^"(.*)"$/ =~ soapaction
      soapaction = $1
    end
    if soapaction.empty?
      return nil
    end
    soapaction
  end

  def lookup_router(namespace)
    if namespace
      @router_map[namespace] || @app_scope_router
    else
      @app_scope_router
    end
  end

  class << self
  public
    def add_servant_to_router(router, obj, namespace)
      ::SOAP::RPC.defined_methods(obj).each do |name|
	add_servant_method_to_router(router, obj, namespace, name)
      end
    end

    def add_servant_method_to_router(router, obj, namespace, name)
      qname = XSD::QName.new(namespace, name)
      soapaction = nil
      method = obj.method(name)
      param_def = ::SOAP::RPC::SOAPMethod.create_param_def(
        (1..method.arity.abs).collect { |i| "p#{ i }" })
      router.add_method(obj, qname, soapaction, name, param_def)
    end
  end
end


end
end