atomWithDefault should accept an update function that returns RESET like atomWithReset

[scamden]

Bug Description

atomWithReset accepts set updates as a function that potentially returns RESET based on previous value and the types reflect this. atomWithDefault doesn't. It only accepts RESET as literal not as the result of the update function. This makes the types of the two atoms incompatible. Would be great to have atomWithDefault do the same behavior.

Reproduction Link

https://www.typescriptlang.org/play/?#code/MYewdgzgLgBASgUQMoICowLwyQTwLYBGIANgBQDkATgKYTVTkCUA3AFBQ4AO1MAhpQHMImGAG1SpRpgB8MMAFdC1SjEYAfOYoLKYGjtxAAzeMjQBddlx406UAIKDhWcZJkxSCpZT1WjJlKiMUhqe2io+BsaIARashvJgwFAAluAw4NSkAHQ5-EIwAFx8jlIA3qysAL4V8YkpaVAA7iDZuY5FNvQOQmUV1aygkLC8BfrUfk0gIhnMQA

[dai-shi]

It's not about the typing, but about the capability of atomWithDefault, right?
Sounds fine. Would you like to send a PR?

[scamden]

Ya good point the types are just pointing out the true difference in function. I'd love to! Will send soon

[scamden]

Here you go! Let me know what you think! #3137