always fetch English attachment first

NJUPTAAA · Oct 24, 2021 · 144b02e · 144b02e
1 parent cf74c8e
commit 144b02e
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 3 deletions.
diff --git a/Crawler.php b/Crawler.php
@@ -104,8 +104,23 @@ private function _extractCodeForces($pcode, $url)
 
         if (stripos($content, "<title>Attachments") !== false) {
             // refetching actual attachment
-            $attachmentDOM = HtmlDomParser::str_get_html($content, true, true, DEFAULT_TARGET_CHARSET, false);
-            $attachmentURL = $attachmentDOM->find('#pageContent div.datatable tbody tr a', 0)->href;
+            $attachmentURL = null;
+            $attachmentsDOM = HtmlDomParser::str_get_html($content, true, true, DEFAULT_TARGET_CHARSET, false);
+            foreach ($attachmentsDOM->find('#pageContent div.datatable tbody tr') as $attachmentDOM) {
+                if(filled($attachmentDOM->find('a', 0)) && trim($attachmentDOM->find('td', 1)->plaintext == 'English')) {
+                    $attachmentURL = $attachmentDOM->find('a', 0)->href;
+                    break;
+                }
+            }
+            if(blank($attachmentURL)) {
+                $firstAttachmentDOM = $attachmentsDOM->find('#pageContent div.datatable tbody tr a', 0);
+                if(filled($firstAttachmentDOM)) {
+                    $attachmentURL = $firstAttachmentDOM->href;
+                } else {
+                    $this->line("\n  <bg=red;fg=white> Exception </> : <fg=yellow>Statement is not available.</>\n");
+                    return false;
+                }
+            }
             $url = $this->globalizeCodeForcesURL($attachmentURL);
             $response = $this->getCodeForcesResponse($url);
             $contentType = $response->headers['content-type'];

diff --git a/babel.json b/babel.json
@@ -6,7 +6,7 @@
     "description": "CodeForces Gym Interface for NOJ",
     "license": "MIT",
     "repository": "https://github.com/NJUPTAAA/NOJ_Extension_Babel_Gym",
-    "version": "0.1.3",
+    "version": "0.1.4",
     "website": "https://codeforces.com/gyms",
     "custom": {
         "css": "resources/custom.css"