1 files changed, 50 insertions, 0 deletions
diff --git a/src/main/java/com/p4square/restlet/oauth/OAuthUser.java b/src/main/java/com/p4square/restlet/oauth/OAuthUser.java
new file mode 100644
index 0000000..11dbac1
--- /dev/null
+++ b/src/main/java/com/p4square/restlet/oauth/OAuthUser.java
@@ -0,0 +1,50 @@
+/*
+ * Copyright 2013 Jesse Morgan
+ */
+
+package com.p4square.restlet.oauth;
+
+import org.restlet.data.ChallengeResponse;
+import org.restlet.security.User;
+
+/**
+ * Simple User object which also contains an OAuth AccessToken.
+ *
+ * @author Jesse Morgan <jesse@jesterpm.net>
+ */
+public class OAuthUser extends User {
+    private final Token mToken;
+    private final String mContentLocation;
+
+    public OAuthUser(Token token) {
+        this(null, token);
+    }
+
+    public OAuthUser(String location, Token token) {
+        super();
+        mToken = token;
+        mContentLocation = location;
+    }
+
+    /**
+     * @return the Location associated with the user.
+     */
+    public String getLocation() {
+        return mContentLocation;
+    }
+
+    /**
+     * @return The AccessToken.
+     */
+    public Token getToken() {
+        return mToken;
+    }
+
+    /**
+     * Convenience method for getToken().getChallengeResponse().
+     * @return A ChallengeResponse based upon the access token.
+     */
+    public ChallengeResponse getChallengeResponse() {
+        return mToken.getChallengeResponse();
+    }
+}