221 lines
8.2 KiB
Kotlin
221 lines
8.2 KiB
Kotlin
/* Copyright 2018 Conny Duck
|
|
*
|
|
* This file is a part of Tusky.
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it under the terms of the
|
|
* GNU General Public License as published by the Free Software Foundation; either version 3 of the
|
|
* License, or (at your option) any later version.
|
|
*
|
|
* Tusky is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even
|
|
* the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
|
|
* Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along with Tusky; if not,
|
|
* see <http://www.gnu.org/licenses>. */
|
|
|
|
package com.keylesspalace.tusky.viewmodel
|
|
|
|
import android.app.Application
|
|
import android.net.Uri
|
|
import androidx.core.net.toUri
|
|
import androidx.lifecycle.MutableLiveData
|
|
import androidx.lifecycle.ViewModel
|
|
import androidx.lifecycle.viewModelScope
|
|
import at.connyduck.calladapter.networkresult.fold
|
|
import com.keylesspalace.tusky.appstore.EventHub
|
|
import com.keylesspalace.tusky.appstore.ProfileEditedEvent
|
|
import com.keylesspalace.tusky.entity.Account
|
|
import com.keylesspalace.tusky.entity.Instance
|
|
import com.keylesspalace.tusky.entity.StringField
|
|
import com.keylesspalace.tusky.network.MastodonApi
|
|
import com.keylesspalace.tusky.util.Error
|
|
import com.keylesspalace.tusky.util.Loading
|
|
import com.keylesspalace.tusky.util.Resource
|
|
import com.keylesspalace.tusky.util.Success
|
|
import com.keylesspalace.tusky.util.randomAlphanumericString
|
|
import kotlinx.coroutines.launch
|
|
import okhttp3.MediaType.Companion.toMediaTypeOrNull
|
|
import okhttp3.MultipartBody
|
|
import okhttp3.RequestBody
|
|
import okhttp3.RequestBody.Companion.asRequestBody
|
|
import okhttp3.RequestBody.Companion.toRequestBody
|
|
import org.json.JSONException
|
|
import org.json.JSONObject
|
|
import retrofit2.HttpException
|
|
import java.io.File
|
|
import javax.inject.Inject
|
|
|
|
private const val HEADER_FILE_NAME = "header.png"
|
|
private const val AVATAR_FILE_NAME = "avatar.png"
|
|
|
|
class EditProfileViewModel @Inject constructor(
|
|
private val mastodonApi: MastodonApi,
|
|
private val eventHub: EventHub,
|
|
private val application: Application
|
|
) : ViewModel() {
|
|
|
|
val profileData = MutableLiveData<Resource<Account>>()
|
|
val avatarData = MutableLiveData<Uri>()
|
|
val headerData = MutableLiveData<Uri>()
|
|
val saveData = MutableLiveData<Resource<Nothing>>()
|
|
val instanceData = MutableLiveData<Resource<Instance>>()
|
|
|
|
private var oldProfileData: Account? = null
|
|
|
|
fun obtainProfile() = viewModelScope.launch {
|
|
if (profileData.value == null || profileData.value is Error) {
|
|
|
|
profileData.postValue(Loading())
|
|
|
|
mastodonApi.accountVerifyCredentials().fold(
|
|
{ profile ->
|
|
oldProfileData = profile
|
|
profileData.postValue(Success(profile))
|
|
},
|
|
{
|
|
profileData.postValue(Error())
|
|
}
|
|
)
|
|
}
|
|
}
|
|
|
|
fun getAvatarUri() = getCacheFileForName(AVATAR_FILE_NAME).toUri()
|
|
|
|
fun getHeaderUri() = getCacheFileForName(HEADER_FILE_NAME).toUri()
|
|
|
|
fun newAvatarPicked() {
|
|
avatarData.value = getAvatarUri()
|
|
}
|
|
|
|
fun newHeaderPicked() {
|
|
headerData.value = getHeaderUri()
|
|
}
|
|
|
|
fun save(newDisplayName: String, newNote: String, newLocked: Boolean, newFields: List<StringField>) {
|
|
|
|
if (saveData.value is Loading || profileData.value !is Success) {
|
|
return
|
|
}
|
|
|
|
saveData.value = Loading()
|
|
|
|
val displayName = if (oldProfileData?.displayName == newDisplayName) {
|
|
null
|
|
} else {
|
|
newDisplayName.toRequestBody(MultipartBody.FORM)
|
|
}
|
|
|
|
val note = if (oldProfileData?.source?.note == newNote) {
|
|
null
|
|
} else {
|
|
newNote.toRequestBody(MultipartBody.FORM)
|
|
}
|
|
|
|
val locked = if (oldProfileData?.locked == newLocked) {
|
|
null
|
|
} else {
|
|
newLocked.toString().toRequestBody(MultipartBody.FORM)
|
|
}
|
|
|
|
val avatar = if (avatarData.value != null) {
|
|
val avatarBody = getCacheFileForName(AVATAR_FILE_NAME).asRequestBody("image/png".toMediaTypeOrNull())
|
|
MultipartBody.Part.createFormData("avatar", randomAlphanumericString(12), avatarBody)
|
|
} else {
|
|
null
|
|
}
|
|
|
|
val header = if (headerData.value != null) {
|
|
val headerBody = getCacheFileForName(HEADER_FILE_NAME).asRequestBody("image/png".toMediaTypeOrNull())
|
|
MultipartBody.Part.createFormData("header", randomAlphanumericString(12), headerBody)
|
|
} else {
|
|
null
|
|
}
|
|
|
|
// when one field changed, all have to be sent or they unchanged ones would get overridden
|
|
val fieldsUnchanged = oldProfileData?.source?.fields == newFields
|
|
val field1 = calculateFieldToUpdate(newFields.getOrNull(0), fieldsUnchanged)
|
|
val field2 = calculateFieldToUpdate(newFields.getOrNull(1), fieldsUnchanged)
|
|
val field3 = calculateFieldToUpdate(newFields.getOrNull(2), fieldsUnchanged)
|
|
val field4 = calculateFieldToUpdate(newFields.getOrNull(3), fieldsUnchanged)
|
|
|
|
if (displayName == null && note == null && locked == null && avatar == null && header == null &&
|
|
field1 == null && field2 == null && field3 == null && field4 == null
|
|
) {
|
|
/** if nothing has changed, there is no need to make a network request */
|
|
saveData.postValue(Success())
|
|
return
|
|
}
|
|
|
|
viewModelScope.launch {
|
|
mastodonApi.accountUpdateCredentials(
|
|
displayName, note, locked, avatar, header,
|
|
field1?.first, field1?.second, field2?.first, field2?.second, field3?.first, field3?.second, field4?.first, field4?.second
|
|
).fold(
|
|
{ newProfileData ->
|
|
saveData.postValue(Success())
|
|
eventHub.dispatch(ProfileEditedEvent(newProfileData))
|
|
},
|
|
{ throwable ->
|
|
if (throwable is HttpException) {
|
|
val errorResponse = throwable.response()?.errorBody()?.string()
|
|
val errorMsg = if (!errorResponse.isNullOrBlank()) {
|
|
try {
|
|
JSONObject(errorResponse).optString("error", "")
|
|
} catch (e: JSONException) {
|
|
null
|
|
}
|
|
} else {
|
|
null
|
|
}
|
|
saveData.postValue(Error(errorMessage = errorMsg))
|
|
} else {
|
|
saveData.postValue(Error())
|
|
}
|
|
}
|
|
)
|
|
}
|
|
}
|
|
|
|
// cache activity state for rotation change
|
|
fun updateProfile(newDisplayName: String, newNote: String, newLocked: Boolean, newFields: List<StringField>) {
|
|
if (profileData.value is Success) {
|
|
val newProfileSource = profileData.value?.data?.source?.copy(note = newNote, fields = newFields)
|
|
val newProfile = profileData.value?.data?.copy(
|
|
displayName = newDisplayName,
|
|
locked = newLocked, source = newProfileSource
|
|
)
|
|
|
|
profileData.postValue(Success(newProfile))
|
|
}
|
|
}
|
|
|
|
private fun calculateFieldToUpdate(newField: StringField?, fieldsUnchanged: Boolean): Pair<RequestBody, RequestBody>? {
|
|
if (fieldsUnchanged || newField == null) {
|
|
return null
|
|
}
|
|
return Pair(
|
|
newField.name.toRequestBody(MultipartBody.FORM),
|
|
newField.value.toRequestBody(MultipartBody.FORM)
|
|
)
|
|
}
|
|
|
|
private fun getCacheFileForName(filename: String): File {
|
|
return File(application.cacheDir, filename)
|
|
}
|
|
|
|
fun obtainInstance() = viewModelScope.launch {
|
|
if (instanceData.value == null || instanceData.value is Error) {
|
|
instanceData.postValue(Loading())
|
|
|
|
mastodonApi.getInstance().fold(
|
|
{ instance ->
|
|
instanceData.postValue(Success(instance))
|
|
},
|
|
{
|
|
instanceData.postValue(Error())
|
|
}
|
|
)
|
|
}
|
|
}
|
|
}
|