com.dounine.clouddisk360.parser.deserializer.authtoken.AuthTokenResponseHandle Maven / Gradle / Ivy

Go to download
package com.dounine.clouddisk360.parser.deserializer.authtoken;

import java.io.IOException;
import java.util.List;

import org.apache.http.HttpResponse;
import org.apache.http.client.ClientProtocolException;
import org.apache.http.client.ResponseHandler;
import org.apache.http.cookie.Cookie;

import com.dounine.clouddisk360.parser.AuthTokenParser;
import com.dounine.clouddisk360.parser.deserializer.BaseResponseHandle;

public class AuthTokenResponseHandle extends BaseResponseHandle
		implements ResponseHandler {

	public AuthTokenResponseHandle(AuthTokenParser parse) {
		super(parse);
	}

	@Override
	public AuthToken handleResponse(HttpResponse response) throws ClientProtocolException, IOException {
		List cookies = parse.getHttpClientContext().getCookieStore().getCookies();
		AuthToken authToken = this.desializer(cookies.stream()
				.filter(c -> c.getName().equals(AuthTokenConst.COOKIE_NAME)).findFirst().get().getValue());
		saveCookie();
		return authToken;
	}

	public String disassemblyResult(String result) {
		return String.format("{'token':'%s'}", result);
	}

	@Override
	public void saveCookie() {
		parse.getCookieStoreUT().writeCookieStoreToDisk(parse.getHttpClientContext().getCookieStore(),
				AuthTokenConst.TOKEN_NAME, true);
	}

}