69 lines
3.3 KiB
TypeScript
69 lines
3.3 KiB
TypeScript
import { HttpRequestHandlerAdapter } from './adapter'
|
|
import { AttributeKeys } from './constants'
|
|
|
|
import type { JavaServerOptions } from '../'
|
|
|
|
const DefaultHttpResponse = Java.type('io.netty.handler.codec.http.DefaultHttpResponse')
|
|
const DefaultFullHttpResponse = Java.type('io.netty.handler.codec.http.DefaultFullHttpResponse')
|
|
const HttpHeaders = Java.type('io.netty.handler.codec.http.HttpHeaders')
|
|
const HttpVersion = Java.type('io.netty.handler.codec.http.HttpVersion')
|
|
const HttpResponseStatus = Java.type('io.netty.handler.codec.http.HttpResponseStatus')
|
|
const LastHttpContent = Java.type('io.netty.handler.codec.http.LastHttpContent')
|
|
|
|
const File = Java.type('java.io.File')
|
|
const Runnable = Java.type('java.lang.Runnable')
|
|
const RandomAccessFile = Java.type('java.io.RandomAccessFile')
|
|
const DefaultFileRegion = Java.type('io.netty.channel.DefaultFileRegion')
|
|
const ChannelFutureListener = Java.type('io.netty.channel.ChannelFutureListener')
|
|
|
|
export class HttpRequestHandler extends HttpRequestHandlerAdapter {
|
|
private ws: string
|
|
private root: string
|
|
constructor(options: JavaServerOptions) {
|
|
super()
|
|
this.root = options.root
|
|
this.ws = options.path
|
|
if (options.httpRequestHandler) {
|
|
this.httpRequestHandler = options.httpRequestHandler
|
|
}
|
|
}
|
|
channelRead0(ctx: any, request: any) {
|
|
if (request.getUri().startsWith(this.ws)) {
|
|
ctx.channel().attr(AttributeKeys.Request).set(request)
|
|
ctx.fireChannelRead(request.retain())
|
|
} else {
|
|
this.httpRequestHandler(ctx, request)
|
|
}
|
|
}
|
|
httpRequestHandler(ctx: any, request: any) {
|
|
ctx.executor().execute(new Runnable({
|
|
run: () => {
|
|
if (HttpHeaders.is100ContinueExpected(request)) {
|
|
ctx.writeAndFlush(new DefaultFullHttpResponse(HttpVersion.HTTP_1_1, HttpResponseStatus.CONTINUE))
|
|
}
|
|
let filename = request.getUri().split('?')[0].substr(1)
|
|
let file = new File(this.root, filename || 'index.html')
|
|
if (!file.exists() || !file.isFile()) {
|
|
ctx.write(new DefaultHttpResponse(request.getProtocolVersion(), HttpResponseStatus.NOT_FOUND))
|
|
ctx.writeAndFlush(LastHttpContent.EMPTY_LAST_CONTENT).addListener(ChannelFutureListener.CLOSE)
|
|
return
|
|
}
|
|
let response = new DefaultHttpResponse(request.getProtocolVersion(), HttpResponseStatus.OK)
|
|
response.headers().set(HttpHeaders.Names.CONTENT_TYPE, "text/html charset=UTF-8")
|
|
let raf = new RandomAccessFile(file, 'r')
|
|
let keepAlive = HttpHeaders.isKeepAlive(request)
|
|
if (keepAlive) {
|
|
response.headers().set(HttpHeaders.Names.CONTENT_LENGTH, file.length())
|
|
response.headers().set(HttpHeaders.Names.CONNECTION, HttpHeaders.Values.KEEP_ALIVE)
|
|
}
|
|
ctx.write(response)
|
|
ctx.write(new DefaultFileRegion(raf.getChannel(), 0, raf.length()))
|
|
let future = ctx.writeAndFlush(LastHttpContent.EMPTY_LAST_CONTENT)
|
|
if (!keepAlive) {
|
|
future.addListener(ChannelFutureListener.CLOSE)
|
|
}
|
|
}
|
|
}))
|
|
}
|
|
}
|