view modules/http.tp @ 157:55e0dca7d3d7

Partial implementation of HTTP get requests
author Mike Pavone <pavone@retrodev.com>
date Sat, 10 Aug 2013 15:06:56 -0700
parents 075b1e71feff
children d81de309a51f
line wrap: on
line source

{
	response <- :_headers _status _sock _data {
		_open? <- true
		_body <- ""
		_length <- int32: (_headers get: "Content-Length" withDefault: "-1")
		_chunked? <- (_headers get: "Transfer-Encoding" withDefault: "") = "chunked"
		_code <- int32: _status
		#{
			headers <- { _headers }
			status <- { _status }
			statusCode <- { _code }
			body <- {
				if: _open? {
					if: _chunked? {

					} else: {
						if: _length >= 0 {
							_body <- _data . (_sock recvAll: (_length - (_data byte_length)))
						} else: {
							chunk <- ""
							while: {
								chunk <- _sock recv: 4096
								(chunk length) > 0
							} do: {
								_data <- _data . chunk
							}
							_body <- _data
						}
					}
					_data <- ""
					close
				}
				_body
			}
			close <- {
				if: _open? {
					_sock close
					_open? <- false
				}
			}
		}
	}
	#{
		client:usingPort <- :address :port{
			#{
				get <- :path {
					sock <- socket connectTo: address onPort: port
					sock send: "GET " . path . " HTTP/1.1\r\nHost: " . address . "\r\n\r\n"
					resp <- ""
					waiting <- true
					headerText <- ""
					rest <- ""
					status <- ""
					while: { waiting } do: {
						data <- sock recv 4096
						resp <- resp . data
						pos <- resp find: "\r\n\r\n" else: { -1 }
						if: pos >= 0 {
							waiting <- false
							statusEnd <- resp find: "\r\n" else: { 0 }
							statusStart <- (resp find: " " else: { 0 }) + 1
							status <- resp from: statusStart withLength: (statusEnd - statusStart)
							headerText <- resp from: statusEnd + 2 withLength: pos - (statusEnd + 2)
							rest <- resp from: pos + 4
						}
					}
					headers <- (headerText splitOn: "\r\n") fold: (dict linear) with: :acc curLine{
						//TODO: support multiple headers with the same name
						part <- curLine partitionOn: ":"
						acc set: (trim: (part before)) (trim: (part after))
					}


					response: headers status sock rest
				}
			}
		}

		client <- :address {
			client: address usingPort: 80
		}
	}
}