Let AvatarType be the receiver of avatarShape()

libraries/designsystem/src/main/kotlin/io/element/android/libraries/designsystem/components/avatar/AvatarShape.kt

@@ -13,11 +13,9 @@ import androidx.compose.runtime.Composable
 import androidx.compose.ui.graphics.Shape
 
 @Composable
-fun avatarShape(
+fun AvatarType.avatarShape(): Shape {
-    avatarType: AvatarType,
+    return when (this) {
-): Shape {
+        is AvatarType.Space -> RoundedCornerShape(cornerSize)
-    return when (avatarType) {
-        is AvatarType.Space -> RoundedCornerShape(avatarType.cornerSize)
         is AvatarType.Room,
         AvatarType.User -> CircleShape
     }

libraries/designsystem/src/main/kotlin/io/element/android/libraries/designsystem/components/avatar/ImageAvatar.kt

@@ -36,7 +36,7 @@ internal fun ImageAvatar(
         contentScale = ContentScale.Companion.Crop,
         modifier = modifier
                 .size(size)
-                .clip(avatarShape(avatarType))
+                .clip(avatarType.avatarShape())
     ) {
         val collectedState by painter.state.collectAsState()
         when (val state = collectedState) {

libraries/designsystem/src/main/kotlin/io/element/android/libraries/designsystem/components/avatar/TextAvatar.kt

@@ -42,7 +42,7 @@ internal fun TextAvatar(
     Box(
         modifier
             .size(size)
-            .clip(avatarShape(avatarType))
+            .clip(avatarType.avatarShape())
             .background(color = colors.background)
     ) {
         val fontSize = size.toSp() / 2

libraries/matrixui/src/main/kotlin/io/element/android/libraries/matrix/ui/components/UnsavedAvatar.kt

@@ -48,7 +48,7 @@ fun UnsavedAvatar(
 ) {
     val commonModifier = modifier
         .size(70.dp)
-        .clip(avatarShape(avatarType))
+        .clip(avatarType.avatarShape())
 
     if (avatarUri != null) {
         val context = LocalContext.current