workavia-mail-front/lib/features/mailbox/domain/extensions/presentation_mailbox_extension.dart

import 'package:core/core.dart';
import 'package:model/model.dart';

extension PresentationMailboxExtension on PresentationMailbox {

  String getMailboxIcon(ImagePaths imagePaths) {
     if (hasRole()) {
        switch(role!.value) {
          case 'inbox':
            return imagePaths.icMailboxInbox;
          case 'drafts':
            return imagePaths.icMailboxDrafts;
          case 'archive':
            return imagePaths.icMailboxArchived;
          case 'sent':
            return imagePaths.icMailboxSent;
          case 'trash':
            return imagePaths.icMailboxTrash;
          case 'spam':
            return imagePaths.icMailboxSpam;
          case 'templates':
            return imagePaths.icMailboxTemplate;
          case 'all_mail':
            return imagePaths.icMailboxAllMail;
          default:
            return imagePaths.icFolderMailbox;
        }
     } else if (isChildOfTeamMailboxes) {
        switch(name!.name.toLowerCase()) {
          case 'inbox':
            return imagePaths.icMailboxInbox;
          case 'drafts':
            return imagePaths.icMailboxDrafts;
          case 'archive':
            return imagePaths.icMailboxArchived;
          case 'sent':
            return imagePaths.icMailboxSent;
          case 'trash':
            return imagePaths.icMailboxTrash;
          case 'spam':
            return imagePaths.icMailboxSpam;
          case 'templates':
            return imagePaths.icMailboxTemplate;
          default:
            return imagePaths.icFolderMailbox;
        }
     } else {
        return imagePaths.icFolderMailbox;
     }
  }
}