com.textrecruit.ustackserver.api.BasicAPI Maven / Gradle / Ivy
package com.textrecruit.ustackserver.api;
import com.mongodb.BasicDBObject;
import com.mongodb.DBObject;
import com.textrecruit.ustackserverapi.APIDocumentation;
import com.textrecruit.ustackserverapi.APIResponse;
import com.textrecruit.ustackserverapi.CallParameters;
import com.textrecruit.ustackserverapi.MethodDefinition;
import com.textrecruit.ustackserverapi.params.ParamNames;
import com.textrecruit.ustack.data.UniqueReference;
import com.textrecruit.ustack.data.UserAccount;
import com.textrecruit.ustack.main.UForgotPasswordSvc;
import org.apache.log4j.Logger;
import org.jboss.netty.buffer.ChannelBuffers;
import org.jboss.netty.channel.Channel;
import org.jboss.netty.channel.ChannelFuture;
import org.jboss.netty.channel.ChannelFutureListener;
import org.jboss.netty.handler.codec.http.DefaultHttpResponse;
import org.jboss.netty.handler.codec.http.HttpRequest;
import org.jboss.netty.handler.codec.http.HttpResponse;
import org.jboss.netty.handler.codec.http.HttpResponseStatus;
import org.jboss.netty.handler.stream.ChunkedStream;
import org.jboss.netty.util.CharsetUtil;
import org.w3c.dom.Document;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;
import javax.xml.transform.stream.StreamSource;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.StringWriter;
import java.util.HashMap;
import static org.jboss.netty.handler.codec.http.HttpHeaders.setContentLength;
import static org.jboss.netty.handler.codec.http.HttpVersion.HTTP_1_1;
public class BasicAPI {
static Logger logger = Logger.getLogger(BasicAPI.class);
public void documentationCss(MethodDefinition def, Channel channel, HttpRequest req, CallParameters callParams) throws Exception
{
HttpResponse response = new DefaultHttpResponse(HTTP_1_1, HttpResponseStatus.OK);
// Write the initial line and the header.
channel.write(response);
// Write the content.
ChannelFuture writeFuture = channel.write(new ChunkedStream(BasicAPI.class.getResourceAsStream("/com/untzuntz/ustackserver/docs/apidocs.css")));
writeFuture.addListener(ChannelFutureListener.CLOSE);
}
public void documentation(MethodDefinition def, Channel channel, HttpRequest req, CallParameters callParams) throws Exception
{
generateDocumentation(def, channel, req, callParams, null);
}
public static void generateDocumentation(MethodDefinition def, Channel channel, HttpRequest req, CallParameters callParams, HashMap documentationDefaults) throws Exception
{
TransformerFactory tFactory = TransformerFactory.newInstance();
String docHtmlResourcePath = (String)def.getData("documentationHtmlResourcePath");
if (docHtmlResourcePath == null)
docHtmlResourcePath = "/com/untzuntz/ustackserver/docs/documentation.html";
InputStream input = null;
try {
input = BasicAPI.class.getResourceAsStream(docHtmlResourcePath);
InputStreamReader xslReader = new InputStreamReader( input );
String clientId = callParams.get(ParamNames.client_id);
String apiKey = callParams.get(ParamNames.api_key);
String baseUrl = (String)def.getData("baseUrl");
Document doc = APIDocumentation.getDocumentationXML(baseUrl, "curl", clientId, apiKey, 12, documentationDefaults);
DOMSource xml = new DOMSource(doc);
StreamSource xsl = new StreamSource(xslReader) ;
Transformer transformer = tFactory.newTransformer(xsl);
transformer.clearParameters();
StringWriter strWriter = new StringWriter();
StreamResult result = new StreamResult( strWriter );
transformer.transform(xml, result);
HttpResponse res = new DefaultHttpResponse(HTTP_1_1, HttpResponseStatus.OK);
res.setHeader("Content-type", "text/html");
res.setContent(ChannelBuffers.copiedBuffer(strWriter.toString(), CharsetUtil.UTF_8));
setContentLength(res, res.getContent().readableBytes());
channel.write(res).addListener(ChannelFutureListener.CLOSE);
} finally {
if (input != null)
try { input.close(); } catch (Exception e) {}
}
}
public void hashtest(MethodDefinition def, Channel channel, HttpRequest req, CallParameters callParams) throws Exception
{
String serverSig = callParams.getRequestSignature(def.getHashKey());
DBObject resp = new BasicDBObject("serverSignature", serverSig);
String clientSig = "(NOT PROVIDED)";
if (callParams.get(ParamNames.RequestSignature) != null)
clientSig = callParams.get(ParamNames.RequestSignature);
resp.put("clientSignature", clientSig);
resp.put("signatureStatus", "Error");
if (clientSig.equals(serverSig))
resp.put("signatureStatus", "Match");
APIResponse.httpOk(channel, resp, req, callParams);
}
public void confirmEmailAddress(MethodDefinition def, Channel channel, HttpRequest req, CallParameters callParams) throws Exception
{
UniqueReference ref = UniqueReference.getByUID( callParams.get(ParamNames.UID) );
if (ref != null)
{
logger.info("Loaded unique reference from uid : " + callParams.get(ParamNames.UID));
ref.put("used", "true");
UniqueReference.save(ref, ref.getString("userName"));
UserAccount user = UserAccount.getUser(ref.getString("userName"));
if (user != null)
{
user.put("emailConfirmed", "true");
user.save(user.getUserName());
}
if (ref.get("successUrl") != null)
{
HttpResponse res = new DefaultHttpResponse(HTTP_1_1, HttpResponseStatus.TEMPORARY_REDIRECT);
res.setHeader("Location", (String)ref.get("successUrl"));
setContentLength(res, res.getContent().readableBytes());
channel.write(res).addListener(ChannelFutureListener.CLOSE);
}
else
APIResponse.httpOk(channel, APIResponse.success(""), req, callParams);
return;
}
APIResponse.httpError(channel, APIResponse.error("Error during email confirmation process"), req, callParams);
}
public void forgotPassword(MethodDefinition def, Channel channel, HttpRequest req, CallParameters callParams) throws Exception
{
UserAccount usrAcct = UserAccount.getUser(callParams.get(ParamNames.username));
if (usrAcct != null && !usrAcct.isDisabled())
{
try {
UForgotPasswordSvc.sendForgotPassword(usrAcct.getUserName(), getClientIPObject(channel, req), (String)def.getData("url"), usrAcct.getUserName(), usrAcct.getPrimaryEmail(), usrAcct.getFullName(), (String)def.getData("template"));
} catch (Exception e) {
logger.warn("User [" + usrAcct.getUserName() + "] hit an exception during forgot password", e);
APIResponse.httpError(channel, APIResponse.error("Error during forgot password request"), req, callParams);
return;
}
APIResponse.httpOk(channel, APIResponse.success(""), req, callParams);
}
else
APIResponse.httpError(channel, APIResponse.error("Error during forgot password request"), req, callParams);
}
public void resetPassword(MethodDefinition def, Channel channel, HttpRequest req, CallParameters callParams) throws Exception
{
UserAccount usrAcct = UserAccount.getUser(callParams.get(ParamNames.username));
if (usrAcct != null && !usrAcct.isDisabled())
{
try {
UForgotPasswordSvc.sendResetPassword(usrAcct.getUserName(), getClientIPObject(channel, req), usrAcct.getUserName(), usrAcct.getPrimaryEmail(), usrAcct.getFullName(), (String)def.getData("template"), null);
} catch (Exception e) {
logger.warn("User [" + usrAcct.getUserName() + "] hit an exception during forgot password", e);
APIResponse.httpError(channel, APIResponse.error("Error during reset password request"), req, callParams);
return;
}
APIResponse.httpOk(channel, APIResponse.success(""), req, callParams);
}
else
APIResponse.httpError(channel, APIResponse.error("Error during reset password request"), req, callParams);
}
public static DBObject getClientIPObject(Channel channel, HttpRequest req)
{
String clientIP = req.getHeader("X-Real-IP");
if (clientIP == null)
clientIP = req.getHeader("X-Forwarded-For");
if (clientIP == null)
clientIP = channel.getRemoteAddress().toString();
String clientCountry = req.getHeader("X-Country-Code");
if (clientCountry == null)
clientCountry = "UNK";
logger.info("IP: " + clientIP + " -> " + clientCountry);
DBObject ret = new BasicDBObject();
ret.put("requestIP", clientIP);
ret.put("requestCountry", clientCountry);
return ret;
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy