@@ -3,7 +3,14 @@ import { toast } from 'react-toastify'
33import { reject } from 'lodash'
44
55import { BaseModal , Button , FormToggleSwitch } from '~/libs/ui'
6- import { updateMemberTraitsAsync , UserProfile , UserTrait , UserTraitCategoryNames , UserTraitIds } from '~/libs/core'
6+ import {
7+ createMemberTraitsAsync ,
8+ updateMemberTraitsAsync ,
9+ UserProfile ,
10+ UserTrait ,
11+ UserTraitCategoryNames ,
12+ UserTraitIds ,
13+ } from '~/libs/core'
714
815import styles from './OpenForGigsModifyModal.module.scss'
916
@@ -15,6 +22,11 @@ interface OpenForGigsModifyModalProps {
1522 profile : UserProfile
1623}
1724
25+ const methodsMap : { [ key : string ] : any } = {
26+ create : createMemberTraitsAsync ,
27+ update : updateMemberTraitsAsync ,
28+ }
29+
1830const OpenForGigsModifyModal : FC < OpenForGigsModifyModalProps > = ( props : OpenForGigsModifyModalProps ) => {
1931 const [ isSaving , setIsSaving ] : [ boolean , Dispatch < SetStateAction < boolean > > ]
2032 = useState < boolean > ( false )
@@ -32,7 +44,7 @@ const OpenForGigsModifyModal: FC<OpenForGigsModifyModalProps> = (props: OpenForG
3244 const updatedPersonalizationTraits : UserTrait [ ]
3345 = reject ( props . memberPersonalizationTraitsFullData , ( trait : UserTrait ) => ! ! trait . availableForGigs )
3446
35- updateMemberTraitsAsync ( props . profile . handle , [ {
47+ methodsMap [ ! ! props . memberPersonalizationTraitsFullData ? 'update' : 'create' ] ( props . profile . handle , [ {
3648 categoryName : UserTraitCategoryNames . personalization ,
3749 traitId : UserTraitIds . personalization ,
3850 traits : {
0 commit comments