Added author_id to login response and methods to store and retrieve a…

client/src/components/authentication/Login.tsx

@@ -12,7 +12,7 @@ import { toast } from "react-toastify";
 import { useNavigate } from "react-router-dom";
 import axios from "axios";
 
-import { storeToken } from "../../utils/localStorageUtils";
+import { storeAuthorId, storeToken } from "../../utils/localStorageUtils";
 
 const Login = () => {
   const navigate = useNavigate();
@@ -35,6 +35,7 @@ const Login = () => {
         headers: { "Content-Type": "multipart/form-data" },
       })
       .then((response: any) => {
+        storeAuthorId(response.data.author_id);
         storeToken(response.data.token);
 
         toast.success("You are now logged in");

client/src/utils/localStorageUtils.ts

@@ -5,3 +5,11 @@ export const storeToken = (token: string) => {
 export const getToken = () => {
   return localStorage.getItem("authToken");
 };
+
+export const storeAuthorId = (authorId: string) => {
+  localStorage.setItem("author_id", authorId);
+};
+
+export const getAuthorId = () => {
+  return localStorage.getItem("author_id");
+};

server/socialdistribution/views.py

@@ -115,7 +115,9 @@ def post(self, request):
             if success:
                 # create token 
                 token, created = Token.objects.get_or_create(user=user)
-                data = {"token": token.key}
+                matching_author = Author.objects.get(user=user)
+
+                data = {"token": token.key, "author_id": matching_author.id}
                 return Response(data, status=status.HTTP_201_CREATED)
             # on wrong password
             else: