AccountController.cls

AccountController.cls

public with sharing class AccountController {
    @AuraEnabled(cacheable=true)
    public static List<Account> getSubscribedAccounts() {
        List<Account> subscribedAccounts = new List<Account>();
        
        Set<Id> accountIds = new Set<Id>();
        
        List<EntitySubscription> subscriptions = [
            SELECT ParentId
            FROM EntitySubscription
            WHERE SubscriberId = :UserInfo.getUserId()
        ];
        
        Set<Id> accountRecordTypeIds = new Set<Id>();
        for (EntitySubscription subscription : subscriptions) {
            if (subscription.ParentId.getSObjectType() == Account.sObjectType) {
                accountIds.add(subscription.ParentId);
                accountRecordTypeIds.add(subscription.ParentId);
            }
        }
        
        if (!accountRecordTypeIds.isEmpty()) {
            subscribedAccounts = [
                SELECT Id, Name
                FROM Account
                WHERE Id IN :accountRecordTypeIds
            ];
        }
        
        return subscribedAccounts;
    }
    
    @AuraEnabled
    public static Boolean removeAccountSubscription(Id accountId) {
        try {
            List<EntitySubscription> subscriptions = [
                SELECT Id
                FROM EntitySubscription
                WHERE SubscriberId = :UserInfo.getUserId()
                AND ParentId = :accountId
                AND Parent.Type = 'Account'
            ];
            
            if (!subscriptions.isEmpty()) {
                delete subscriptions;
                return true;
            }
        } catch (Exception ex) {
            System.debug('Error removing account subscription: ' + ex.getMessage());
        }
        
        return false;
    }
}