77 lines
2.0 KiB
TypeScript
77 lines
2.0 KiB
TypeScript
import { AuthState } from "@/utils/features/auth/authSlice"
|
|
import { useMutation } from "@tanstack/react-query"
|
|
|
|
type ErrorResponse = {
|
|
error: string
|
|
message: string
|
|
statusCode: number
|
|
}
|
|
|
|
type onPostSuccessFunc = () => void
|
|
type ErrorFuncProps = { response?: ErrorResponse, error?: Error }
|
|
type onErrorFunc = (props: ErrorFuncProps) => void
|
|
|
|
type TrainProps = {
|
|
// addTrain: (payload: TrainPayload) => { payload: GamePayload, type: "train/addTrain" }
|
|
auth: AuthState
|
|
onPostSuccess?: onPostSuccessFunc
|
|
onError?: onErrorFunc
|
|
}
|
|
|
|
export const useAddTrainMutation = ({ auth, onPostSuccess, onError }: TrainProps) => {
|
|
return useMutation({
|
|
mutationFn: async (trainId: string) => {
|
|
return fetch(`${process.env.EXPO_PUBLIC_TRAINTRAPE_MOI_SERVER}/trains/import/`, {
|
|
method: "POST",
|
|
headers: {
|
|
"Authorization": `Bearer ${auth.token}`,
|
|
"Content-Type": "application/json",
|
|
},
|
|
body: JSON.stringify({
|
|
id: trainId,
|
|
}),
|
|
}).then(resp => resp.json())
|
|
},
|
|
onSuccess: async (data) => {
|
|
if (data.statusCode) {
|
|
if (onError)
|
|
onError({ response: data })
|
|
return
|
|
}
|
|
if (onPostSuccess)
|
|
onPostSuccess()
|
|
},
|
|
onError: async (error: Error) => {
|
|
if (onError)
|
|
onError({ error: error })
|
|
}
|
|
})
|
|
}
|
|
|
|
export const useDeleteTrainMutation = ({ auth, onPostSuccess, onError }: TrainProps) => {
|
|
return useMutation({
|
|
mutationFn: async (trainId: string) => {
|
|
return fetch(`${process.env.EXPO_PUBLIC_TRAINTRAPE_MOI_SERVER}/trains/${trainId}/`, {
|
|
method: "DELETE",
|
|
headers: {
|
|
"Authorization": `Bearer ${auth.token}`,
|
|
"Content-Type": "application/json",
|
|
},
|
|
})
|
|
},
|
|
onSuccess: async (resp) => {
|
|
if (resp.status >= 400) {
|
|
if (onError)
|
|
onError({ response: await resp.json() })
|
|
return
|
|
}
|
|
if (onPostSuccess)
|
|
onPostSuccess()
|
|
},
|
|
onError: async (error: Error) => {
|
|
if (onError)
|
|
onError({ error: error })
|
|
}
|
|
})
|
|
}
|