added token in local storage, and genrated cookie when register to au…

…to-login

backend/src/resolvers/UserResolver.ts

@@ -67,7 +67,8 @@ class UserResolver {
     @Arg("email") email: string,
     @Arg("firstname") firstname: string,
     @Arg("lastname") lastname: string,
-    @Arg("password") password: string
+    @Arg("password") password: string,
+    @Ctx() context: any
   ) {
     console.log("process", process.env);
 
@@ -89,7 +90,7 @@ class UserResolver {
       { id: userFromDB.id, email: userFromDB.email, role: userFromDB.role },
       process.env.JWT_SECRET_KEY
     );
-
+    context.res.setHeader("Set-Cookie", `token=${token}; Secure; HttpOnly`);
     return token;
   }
 
@@ -102,7 +103,7 @@ class UserResolver {
         role: user.role,
         firstname: user.firstname,
         lastname: user.lastname,
-        isLoggedIn: true
+        isLoggedIn: true,
       };
     } else {
       return { isLoggedIn: false };

frontend/src/pages/Register.tsx

@@ -3,10 +3,13 @@ import { useCreateNewUserMutation } from "../generated/graphql-types";
 import { Form, Input, Button, Card, message } from "antd";
 import { UserOutlined, LockOutlined, MailOutlined } from "@ant-design/icons";
 import "../styles/commonLoginRegister.css";
+import { useContext } from "react";
+import { UserContext } from "../components/Layout";
 
 const Register = () => {
   const [createUser] = useCreateNewUserMutation();
   const navigate = useNavigate();
+  const userInfo = useContext(UserContext);
 
   const onFinish = async (values: {
     email: string;
@@ -21,6 +24,8 @@ const Register = () => {
       });
 
       if (data && data.createUser) {
+        localStorage.setItem("token", data.createUser);
+        userInfo.refetch();
         navigate("/");
         message.success("Registration successful!");
       } else {