add wsgi support

Samuel · Samuel · commit 941a914d09d9 · 2018-12-09T18:09:32.000+08:00
diff --git a/nginx/http/http.pyx b/nginx/http/http.pyx
@@ -1,7 +1,27 @@
 from cpython cimport Py_INCREF, Py_DECREF
-from .nginx_core cimport ngx_log_error, NGX_LOG_CRIT, NGX_AGAIN, from_nginx_str
-from .ngx_http cimport ngx_http_request_t, ngx_http_core_run_phases
-from .ngx_http cimport ngx_http_get_module_ctx, ngx_http_set_ctx
+from .nginx_core cimport (
+    ngx_log_error,
+    NGX_LOG_CRIT,
+    NGX_AGAIN,
+    from_nginx_str,
+    ngx_calloc,
+    ngx_free,
+    ngx_memcpy,
+)
+from .ngx_http cimport (
+    ngx_http_request_t,
+    ngx_http_core_run_phases,
+    ngx_http_get_module_ctx,
+    ngx_http_set_ctx,
+    ngx_http_send_header,
+    ngx_list_push,
+    ngx_table_elt_t,
+    ngx_str_set,
+    ngx_http_output_filter,
+    ngx_chain_t,
+    ngx_buf_t,
+    ngx_calloc_buf,
+)
 
 import traceback
 
@@ -17,6 +37,8 @@ cdef class Request:
         public str extension
         public str unparsed_uri
         public str method_name
+        public str content_type
+        public str content_length
         public str http_protocol
 
     def __init__(self, *args):
@@ -37,17 +59,82 @@ cdef class Request:
             return self.future.result()
         return NGX_AGAIN
 
+    def send_header(self):
+        return ngx_http_send_header(self.request)
+
+    def add_response_header(self, key, value):
+        cdef:
+            ngx_table_elt_t *h
+            char *cstr
+            char *csource
+            bytes key_data, value_data
+        h = ngx_list_push(&self.request.headers_out.headers)
+        if h == NULL:
+            raise MemoryError()
+        h.hash = 1
+
+        key_data = str(key).encode('iso8859-1')
+        cstr = <char *>ngx_calloc(sizeof(char) * len(key_data), self.request.connection.log)
+        h.key.len = len(key_data)
+        csource = key_data
+        ngx_memcpy(cstr, csource, len(key_data))
+        h.key.data = cstr
+
+        value_data = str(value).encode('iso8859-1')
+        cstr = <char *>ngx_calloc(sizeof(char) * len(value_data), self.request.connection.log)
+        h.value.len = len(value_data)
+        csource = value_data
+        ngx_memcpy(cstr, csource, len(value_data))
+        h.value.data = cstr
+
+    def send_response(self, pos):
+        cdef:
+            ngx_chain_t out
+            ngx_buf_t *b
+            bytes data = pos
+            char* cstr = data
+        b = ngx_calloc_buf(self.request.pool)
+        if b == NULL:
+            raise MemoryError
+        b.last_buf = 1
+        b.last_in_chain = 1
+        b.memory = 1
+        b.pos = cstr
+        b.last = b.pos + len(data)
+
+        out.buf = b
+        out.next = NULL
+
+        return ngx_http_output_filter(self.request, &out)
+
+    property response_status:
+        def __get__(self):
+            return self.request.headers_out.status
+
+        def __set__(self, value):
+            self.request.headers_out.status = value
+
+    property response_content_length:
+        def __get__(self):
+            if self.request.headers_out.content_length:
+                return self.request.headers_out.content_length.value
+
+        def __set__(self, value):
+            self.request.headers_out.content_length.value = value
+
     def __repr__(self):
         return f'Request({self.method_name} {self.uri})'
 
     def __str__(self):
         return f''' request_line: {self.request_line}
-          uri: {self.uri}
-         args: {self.args}
-    extension: {self.extension}
- unparsed_uri: {self.unparsed_uri}
-  method_name: {self.method_name}
-http_protocol: {self.http_protocol}'''
+           uri: {self.uri}
+          args: {self.args}
+     extension: {self.extension}
+  unparsed_uri: {self.unparsed_uri}
+   method_name: {self.method_name}
+  content_type: {self.content_type}
+content_length: {self.content_length}
+ http_protocol: {self.http_protocol}'''
 
     @staticmethod
     cdef Request from_ptr(ngx_http_request_t *request):
@@ -67,6 +154,12 @@ http_protocol: {self.http_protocol}'''
             new_req.unparsed_uri = from_nginx_str(request.unparsed_uri)
             new_req.method_name = from_nginx_str(request.method_name)
             new_req.http_protocol = from_nginx_str(request.http_protocol)
+            if request.headers_in.content_type:
+                new_req.content_type = from_nginx_str(
+                    request.headers_in.content_type.value)
+            if request.headers_in.content_length:
+                new_req.content_length = from_nginx_str(
+                    request.headers_in.content_length.value)
 
             ngx_http_set_ctx(request, <void *>new_req, ngx_python_module)
             return new_req
diff --git a/nginx/http/module.py b/nginx/http/module.py
@@ -1,4 +1,7 @@
 import asyncio
+import io
+import sys
+import functools
 
 from . import log
 from .._nginx import run_phases
@@ -13,14 +16,48 @@ def __init__(self):
     def init_process(self):
         self.loop = asyncio.get_event_loop()
 
+    def start_response(self, request, status, response_headers, exc_info=None):
+        request.response_status = int(status.split(maxsplit=1)[0])
+        for key, value in response_headers:
+            request.add_response_header(key, value)
+
     def post_read(self, request):
-        if request._started():
-            log.debug('post_read end')
-            return request._result()
+        environ = {
+            'REQUEST_METHOD': request.method_name,
+            'SCRIPT_NAME': None,
+            'PATH_INFO': request.uri,
+            'QUERY_STRING': request.args,
+            'CONTENT_TYPE': request.content_type,
+            'CONTENT_LENGTH': request.content_length,
+            'SERVER_NAME': 'localhost',
+            'SERVER_PORT': '8080',
+            'SERVER_PROTOCOL': request.http_protocol,
+            'wsgi.input': io.BytesIO(),
+            'wsgi.errors': sys.stderr,
+            'wsgi.version': (1, 0),
+            'wsgi.multithread': False,
+            'wsgi.multiprocess': True,
+            'wsgi.run_once': True,
+        }
+        if environ.get('HTTPS', 'off') in ('on', '1'):
+            environ['wsgi.url_scheme'] = 'https'
         else:
-            log.debug('post_read request:\n%s', request)
-            return request._start(self.loop.create_task(
-                self._post_read_async(request)))
+            environ['wsgi.url_scheme'] = 'http'
+        from myapp import app
+        resp = app(environ, functools.partial(
+            self.start_response, request))
+        request.send_header()
+        rv = 404
+        for pos in resp:
+            rv = request.send_response(pos)
+        return rv
+        # if request._started():
+        #     log.debug('post_read end')
+        #     return request._result()
+        # else:
+        #     log.debug('post_read request:\n%s', request)
+        #     return request._start(self.loop.create_task(
+        #         self._post_read_async(request)))
 
     async def _post_read_async(self, request):
         try:
diff --git a/nginx/nginx_core.pxd b/nginx/nginx_core.pxd
@@ -40,6 +40,7 @@ cdef extern from "ngx_core.h":
 
     void *ngx_calloc(size_t size, ngx_log_t *log)
     void ngx_free(void *p)
+    void *ngx_memcpy(void *dst, const void *src, size_t n)
     void ngx_log_error(ngx_uint_t level,
                        ngx_log_t *log,
                        ngx_err_t err,
diff --git a/nginx/ngx_http.pxd b/nginx/ngx_http.pxd
@@ -1,7 +1,39 @@
-from .nginx_core cimport ngx_str_t, ngx_module_t, ngx_log_t
+from .nginx_core cimport ngx_str_t, ngx_module_t, ngx_log_t, ngx_uint_t, ngx_int_t
 
 
 cdef extern from "ngx_http.h":
+
+    ctypedef struct ngx_table_elt_t:
+        ngx_uint_t hash
+        ngx_str_t key
+        ngx_str_t value
+
+    ctypedef struct ngx_list_t:
+        pass
+
+    ctypedef struct ngx_http_headers_in_t:
+        ngx_table_elt_t *content_type
+        ngx_table_elt_t *content_length
+
+    ctypedef struct ngx_http_headers_out_t:
+        ngx_uint_t status
+        ngx_table_elt_t *content_length
+        ngx_list_t headers
+
+    ctypedef struct ngx_chain_t:
+        ngx_buf_t *buf
+        ngx_chain_t *next
+
+    ctypedef struct ngx_buf_t:
+        unsigned last_buf
+        unsigned last_in_chain
+        unsigned memory
+        char *pos
+        char *last
+
+    ctypedef struct ngx_pool_t:
+        pass
+
     ctypedef struct ngx_connection_t:
         ngx_log_t *log
 
@@ -14,9 +46,24 @@ cdef extern from "ngx_http.h":
         ngx_str_t unparsed_uri
         ngx_str_t method_name
         ngx_str_t http_protocol
+        ngx_pool_t *pool
+        ngx_http_headers_in_t headers_in
+        ngx_http_headers_out_t headers_out
 
     void ngx_http_core_run_phases(ngx_http_request_t *request)
     void *ngx_http_get_module_ctx(ngx_http_request_t *request,
                                   ngx_module_t module)
     void ngx_http_set_ctx(ngx_http_request_t *request, void *ctx,
                           ngx_module_t module)
+
+    void ngx_http_send_header(ngx_http_request_t *r)
+
+    ngx_table_elt_t *ngx_list_push(ngx_list_t *list)
+
+    void ngx_str_set(ngx_str_t *str, char *text)
+
+    ngx_int_t ngx_http_output_filter(ngx_http_request_t *r, ngx_chain_t *input)
+
+    ngx_buf_t *ngx_calloc_buf(ngx_pool_t *pool)
+
+
