<aura:component controller="DataTableEditTestC"
                implements="force:appHostable,flexipage:availableForAllPageTypes,flexipage:availableForRecordHome,force:hasRecordId,forceCommunity:availableForAllPageTypes,force:lightningQuickAction" access="global" >
    
    <aura:attribute name="recordId" type="Id" />
    <aura:attribute name="ServiceAppointments" type="Account" />
    <aura:attribute name="Columns" type="List" />
    <aura:attribute name="saveDraftValues" type="Object" />
    
    <aura:handler name="init" value="{!this}" action="{!c.myAction}" />
    <aura:handler event="force:refreshView" action="{!c.myAction}" />
    
    <force:recordData aura:id="workOrderRecord" recordId="{!v.recordId}" targetFields="{!v.WorkOrder}" layoutType="FULL" />
    
    <lightning:card title="{! 'Service Appointment List for ' + v.WorkOrder.WorkOrderNumber}">
        <!-- Service Appointment list goes here -->
        <lightning:datatable 
                             aura:id="SADataTable"
                             keyField="Id" 
                             data="{! v.ServiceAppointments }" 
                             columns="{! v.Columns }" 
                             onsave="{!c.handleSave}" />
    </lightning:card> 
</aura:component>

({
    myAction : function(component, event, helper) {
        component.set('v.Columns', [
            {label: 'Account Name', fieldName: 'Name', type: 'text', editable: true, initialWidth: 750},
            {label: 'Phone', fieldName: 'Phone', type: 'phone', editable: true}
        ]);
        
        var action = component.get("c.getServiceAppointments");
        action.setCallback(this, function(response) {
            var state = response.getState();
            
            if (state === "SUCCESS") {
                var res = response.getReturnValue();
                component.set("v.ServiceAppointments", res);
            }
            else if (state === "ERROR") {
                var errors = response.getError();
                if (errors) {
                    if (errors[0] && errors[0].message) {
                        console.log("Error message: " + 
                                    errors[0].message);
                    }
                } 
                else {
                    console.log(response.getReturnValue());
                }
            }
        });
        $A.enqueueAction(action);
    },
    
    handleSave : function(component, event, helper) {
        var editedRecords =  component.find("SADataTable").get("v.draftValues");
        var totalRecordEdited = editedRecords.length;
        var action = component.get("c.updateServiceAppointments");
        action.setParams({
            'editedSAList' : editedRecords
        });
        action.setCallback(this,function(response) {
            var state = response.getState();
            if (state === "SUCCESS") {
                //***if update is successful ***
                if(response.getReturnValue() === true){
                    $A.get('e.force:refreshView').fire();
                    var showToast = $A.get("e.force:showToast"); 
                    showToast.setParams({ 
                        'title' : 'Record Update',
                        'type': 'success',
                        'message' : totalRecordEdited+" Service Appointment Records Updated" 
                    }); 
                    showToast.fire(); 
                    }
                } 
            else{ 
                //***if update failed ***
                    showToast.setParams({
                        "title": "Error!!",
                        "type": "error",
                        "message": "Error in update"
                    });
                }
            }
        );
        $A.enqueueAction(action);
    },
})

public class DataTableEditTestC {
    
    @AuraEnabled
    public static List<Account> getServiceAppointments() {
        return [SELECT Id, Name, Phone  FROM Account LIMIT 10];
    }
    
    @AuraEnabled
    public static boolean updateServiceAppointments(List<Account> editedSAList) {
        try {
            update editedSAList;
            return true;
        } catch(Exception e){
            return false;
        }
    }
}