org.webpieces.httpparser.impl.MementoImpl Maven / Gradle / Ivy

Go to download
package org.webpieces.httpparser.impl;

import java.util.ArrayList;
import java.util.List;

import org.webpieces.data.api.DataWrapper;
import org.webpieces.httpparser.api.Memento;
import org.webpieces.httpparser.api.ParsingState;
import org.webpieces.httpparser.api.UnparsedState;
import org.webpieces.httpparser.api.dto.HttpChunk;
import org.webpieces.httpparser.api.dto.HttpPayload;

public class MementoImpl implements Memento {

	//State held to keep parsing messages
	private List leftOverMarkedPositions = new ArrayList<>();
	private DataWrapper leftOverData;

	//If the stream is expecting chunks of data
	private boolean inChunkParsingMode;
	private HttpChunk halfParsedChunk;
	private int numBytesLeftToReadOnChunk;
	
	//Return state for client to access
	private List parsedMessages = new ArrayList<>();
	private int indexBytePointer;
	private boolean isHttp2;
	
	private Integer contentLengthLeftToRead;
	private boolean hasHttpMarkerMsg;
	private int numBytesJustParsed;

	@Override
	public List getParsedMessages() {
		return parsedMessages;
	}

	public void addMessage(HttpPayload message) {
		this.parsedMessages.add(message);
	}
	
	public void setParsedMessages(List parsedMessages) {
		this.parsedMessages = parsedMessages;
	}

	@Override
    public DataWrapper getLeftOverData() {
		return leftOverData;
	}

	public void setLeftOverData(DataWrapper data) {
		this.leftOverData = data;
	}

	public void addDemarcation(int i) {
		leftOverMarkedPositions.add(i);
	}

	public List getLeftOverMarkedPositions() {
		return leftOverMarkedPositions;
	}

	public void setLeftOverMarkedPositions(List leftOverMarkedPositions) {
		this.leftOverMarkedPositions = leftOverMarkedPositions;
	}

	public int getNumBytesLeftToReadOnChunk() {
		return numBytesLeftToReadOnChunk;
	}

	public void setNumBytesLeftToReadOnChunk(int length) {
		numBytesLeftToReadOnChunk = length;
	}

	public void setReadingHttpMessagePointer(int indexBytePointer) {
		this.indexBytePointer = indexBytePointer;
	}

	public int getReadingHttpMessagePointer() {
		return indexBytePointer;
	}

	public void setInChunkParsingMode(boolean inChunkParsingMode) {
		this.inChunkParsingMode = inChunkParsingMode;
	}

	public boolean isInChunkParsingMode() {
		return inChunkParsingMode;
	}

	@Override
	public UnparsedState getUnParsedState() {
		if(inChunkParsingMode) {
			return new UnparsedState(ParsingState.CHUNK, leftOverData.getReadableSize());
		} else if(contentLengthLeftToRead != null) {
			return new UnparsedState(ParsingState.BODY, leftOverData.getReadableSize());
		}

		return new UnparsedState(ParsingState.HEADERS, leftOverData.getReadableSize());
	}

	public boolean isHttp2() {
		return isHttp2;
	}

	public void setHttp2(boolean isHttp2) {
		this.isHttp2 = isHttp2;
	}

	public void setHalfParsedChunk(HttpChunk chunk) {
		this.halfParsedChunk = chunk;
	}

	public HttpChunk getHalfParsedChunk() {
		return halfParsedChunk;
	}

	public Integer getContentLengthLeftToRead() {
		return contentLengthLeftToRead;
	}

	public void setContentLengthLeftToRead(Integer contentLengthLeftToRead) {
		this.contentLengthLeftToRead = contentLengthLeftToRead;
	}

	public boolean isHasHttpMarkerMsg() {
		return hasHttpMarkerMsg;
	}

	public void setHasHttpMarkerMsg(boolean hasHttpMarkerMsg) {
		this.hasHttpMarkerMsg = hasHttpMarkerMsg;
	}

	@Override
	public int getNumBytesJustParsed() {
		return numBytesJustParsed;
	}

	public void setNumBytesJustParsed(int numBytesJustParsed) {
		this.numBytesJustParsed = numBytesJustParsed;
	}

}