public with sharing class customFileUploadctrl {
    @AuraEnabled(cacheable=true)
    public static List<file> getRelatedFiles(String recordId, String sectionType, String screen){
        List<File> files = new List<File>();
        system.debug('screen value is' + screen);
        if(screen == null){
            screen = 'default';
        }

        //retrieve contentDocumentLIst
        List<ContentDocumentLink> contentDocumentList = getContentDocumentList(recordId);
        system.debug('contentDocumentList' + contentDocumentList);

        //setdocumentId
        Set<Id> contentDocumentIdSet = getContentDocumentIdSet(contentDocumentList);

        //retrieve the contentversion for the specific sectionType
        Set<Id> contentDocumentForSectionIdSet = getContentDocumentForSectionIdSet(contentDocumentIdSet,sectionType, screen);
        system.debug('contentDocumentForSectionIdSet' + contentDocumentForSectionIdSet);

        for(ContentDocumentLink conLink : contentDocumentList){
            if(contentDocumentForSectionIdSet.contains(conLink.ContentDocumentId)){
                File file = new File();
                file.Title = conLink.ContentDocument.Title;
                file.Id = conLink.ContentDocument.Id;
                file.CreatedDate = conLink.ContentDocument.CreatedDate;
                file.Type = conLink.ContentDocument.FileType;
                file.Description = conLink.ContentDocument.Description;
                files.add(file);

            }
        }            
        
        return files;
    }

    @AuraEnabled
    public static boolean updateType(List<string> documentArray, string sectionType, string screen){
        system.debug('in the updatetype@@' + sectionType);
        system.debug('screen' + screen);
        if(screen == null){
            screen = 'default';
        }
        List<contentversion> contentVersionList = new List<contentversion>();
        Set<Id> contentDocumentSetId = new Set<Id>();

        /*for(string documentDetails : documentArray){
            contentversion contentVersionItem = new contentversion();
            contentVersionItem.contentDocumentId = Id.valueOf(documentDetails);
            contentVersionItem.Type_fileupload__c = sectionType;
            contentVersionList.add(contentVersionItem);            
        }*/

        for(string documentDetails : documentArray){
            contentDocumentSetId.add(Id.valueOf(documentDetails));
        }
       

        for(contentversion contentversionItem : [SELECT Id, Type__c
                                                    FROM contentversion
                                                    WHERE contentDocumentId=:contentDocumentSetId]){
            contentversion contentVersionRec = new contentversion();
            contentVersionRec.Id = contentversionItem.Id;
            contentVersionRec.Type__c = sectionType;
            contentVersionRec.screen__c = screen;
            contentVersionList.add(contentVersionRec);                                             
        }

        system.debug('contentVersionLis' + contentVersionList);
        update contentVersionList;
        return true;

    }

    private static List<ContentDocumentLink> getContentDocumentList(String recordId){
        List<ContentDocumentLink> contentDocumentList = new List<ContentDocumentLink>();

        return [SELECT ContentDocument.Id, 
                        ContentDocumentId,
                                ContentDocument.Title, 
                                ContentDocument.CreatedDate, 
                                ContentDocument.FileType,
                                ContentDocument.Description
                                    FROM ContentDocumentLink                                     
                                   WHERE LinkedEntityId =: recordId] ;
    }

    private static set<Id> getContentDocumentIdSet(List<ContentDocumentLink> listContentDocumentLinkId){
        set<Id> contentDocumentSetId = new set<Id>();
        for(ContentDocumentLink contentDocItem: listContentDocumentLinkId){
            contentDocumentSetId.add(contentDocItem.ContentDocumentID);
        }
        return contentDocumentSetId;
    }


    private static Set<Id> getContentDocumentForSectionIdSet(Set<Id> contentDocumentSet, String sectionType, String screen){
        Set<Id> contentDocumentForSectionIdSet = new Set<Id>();

        for(ContentVersion conLink : [SELECT ContentDocumentId 
                                                    FROM ContentVersion                                            
                                                    WHERE ContentDocumentId =: contentDocumentSet 
                                                    AND Type__c =: sectionType
                                                    AND screen__C =: screen] ){

            contentDocumentForSectionIdSet.add(conLink.ContentDocumentId); }

        return contentDocumentForSectionIdSet;    
    }



    public class File{
        @AuraEnabled public String Title;
        @AuraEnabled public String Type;
        @AuraEnabled public Id Id;
        @AuraEnabled public Datetime CreatedDate;
        @AuraEnabled public String Description;
    }
}

@isTest
public class customFileUploadctrlTest {

    public static TestMethod void testScenario1(){
       List<String> newStr = new List<String>();
       
       string before = 'Testing base 64 encode';            
         Blob beforeblob = Blob.valueOf(before); 
        
        List<contentversion> conver = new List<contentversion>();
        contentversion rec = new contentversion();
        rec.Title = 'Test';
        rec.screen__c = 'Screen';
        rec.PathOnClient = 'PathTest';
        rec.VersionData = beforeblob;
        rec.Description = 'test description';
        rec.Type__c = 'Invoices';
        conver.add(rec);
        
        insert conver;
      
        List<ContentDocumentLink> cdlink = new List<ContentDocumentLink>();
        set<Id> contentDocumentSetId = new set<Id>();
        
        Test.startTest();
        	customFileUploadctrl.getRelatedFiles('a073G000002LM7hQAG','Test','Test1');
        	customFileUploadctrl.updateType(newStr, 'test','test1');
        Test.stopTest();
    }
}