Create file constant

src/app/[name]/(Profile)/profile/address/new/hooks/useAddressForm.ts

@@ -1,5 +1,5 @@
 import { useState, useEffect } from 'react';
-import { useRouter } from 'next/navigation';
+import { useRouter, useSearchParams } from 'next/navigation';
 import { useCreateAddress, useUpdateAddress } from '../../hooks/useAddressData';
 import { toast } from '@/components/Toast';
 import { CreateAddressType, UpdateAddressType, NominatimReverseGeocodingResponse, Address } from '../../types/Types';
@@ -20,6 +20,8 @@ export const useAddressForm = ({
     addressData = null,
 }: UseAddressFormProps) => {
     const router = useRouter();
+    const searchParams = useSearchParams();
+    const redirect = searchParams.get('redirect');
     const { mutate: createAddress, isPending: isCreating } = useCreateAddress();
     const { mutate: updateAddress, isPending: isUpdating } = useUpdateAddress();
     const isPending = isCreating || isUpdating;
@@ -92,7 +94,11 @@ export const useAddressForm = ({
             updateAddress(updateData, {
                 onSuccess: () => {
                     toast('آدرس با موفقیت به‌روزرسانی شد', 'success');
-                    router.back();
+                    if (redirect) {
+                        router.replace(redirect);
+                    } else {
+                        router.back();
+                    }
                 },
                 onError: (error) => {
                     const errorMessage = error instanceof Error ? error.message : 'خطا در به‌روزرسانی آدرس';
@@ -105,7 +111,11 @@ export const useAddressForm = ({
             createAddress(createData, {
                 onSuccess: () => {
                     toast('آدرس با موفقیت ثبت شد', 'success');
-                    router.back();
+                    if (redirect) {
+                        router.replace(redirect);
+                    } else {
+                        router.back();
+                    }
                 },
                 onError: (error) => {
                     const errorMessage = error instanceof Error ? error.message : 'خطا در ثبت آدرس';