Implement Inspire/HEP citation counts

README.md

@@ -11,7 +11,7 @@ repository.
 
 Code for this extension is based on the [Zotero DOI
  Manager](https://github.com/bwiernik/zotero-shortdoi), which is based
- in part [Zotero Google Scholar
+ in part on [Zotero Google Scholar
  Citations](https://github.com/beloglazov/zotero-scholar-citations) by
  Anton Beloglazov.
 
@@ -26,7 +26,7 @@ Code for this extension is based on the [Zotero DOI
 - Run Zotero (version 5.x)
 - Go to `Tools -> Add-ons`
 - `Install Add-on From File`
-- Choose the file `zotero-citationcounts-0.1.2.xpi`
+- Choose the file `zotero-citationcounts-0.1.3.xpi`
 - Restart Zotero
 
 ## License

chrome/content/scripts/zoterocitationcounts.js

@@ -17,24 +17,24 @@ const operationNames = {
     "semanticscholar": "Semantic Scholar"
 };
 
-function getCitationCount(item, tag) {
-    let extra = item.getField('extra');
-    if (!extra) {
-        return -1;
-    }
-    let extras = extra.split("\n");
-    const patt = new RegExp("^Citations \\(" + tag + "\\): (\\d+).*", "i");
-    extras = extras.filter(ex => patt.test(ex));
-    if (length(extras) == 0) {
-        return -1;
-    }
-    let count = patt.exec(extras[1])
-    if (!count) {
-        return -1;
-    }
-    count = parseInt(count);
-    return count;
-}
+// function getCitationCount(item, tag) {
+//     let extra = item.getField('extra');
+//     if (!extra) {
+//         return -1;
+//     }
+//     let extras = extra.split("\n");
+//     const patt = new RegExp("^Citations \\(" + tag + "\\): (\\d+).*", "i");
+//     extras = extras.filter(ex => patt.test(ex));
+//     if (length(extras) == 0) {
+//         return -1;
+//     }
+//     let count = patt.exec(extras[1])[1]
+//     if (!count) {
+//         return -1;
+//     }
+//     count = parseInt(count);
+//     return count;
+// }
 
 function setCitationCount(item, tag, count) {
     let extra = item.getField('extra');
@@ -55,6 +55,93 @@ function setCitationCount(item, tag, count) {
     item.setField('extra', extra);
 }
 
+async function getCrossrefCount(item) {
+    const doi = item.getField('DOI');
+    if (!doi) {
+        // There is no DOI; skip item
+        return -1;
+    }
+    const edoi = encodeURIComponent(doi);
+
+    let response = null;
+
+    if (response === null) {
+        const style = "vnd.citationstyles.csl+json";
+        const xform = "transform/application/" + style;
+        const url = "https://api.crossref.org/works/" + edoi + "/" + xform;
+        response = await fetch(url)
+            .then(response => response.json())
+            .catch(err => null);
+    }
+
+    if (response === null) {
+        const url = "https://doi.org/" + edoi;
+        const style = "vnd.citationstyles.csl+json";
+        response = await fetch(url, {
+            headers: {
+                "Accept": "application/" + style
+            }
+        })
+            .then(response => response.json())
+            .catch(err => null);
+        }
+
+    if (response === null) {
+        // Something went wrong
+        return -1;
+    }
+
+    let str = null;
+    try {
+        str = response['is-referenced-by-count'];
+    } catch (err) {
+        // There are no citation counts
+        return -1;
+    }
+
+    const count = parseInt(str);
+    return count;
+}
+
+async function getInspireCount(item, idtype) {
+    let doi = null;
+    if (idtype == 'doi') {
+        doi = item.getField('DOI');
+    } else if (idtype == 'arxiv') {
+        const arxiv = item.getField('url'); // check URL for arXiv id
+        const patt = /(?:arxiv.org[/]abs[/]|arXiv:)([a-z.-]+[/]\d+|\d+[.]\d+)/i;
+        doi = patt.exec(arxiv)[1];
+    } else {
+        // Error
+        return -1;
+    }
+    if (!doi) {
+        // There is no DOI / arXiv id; skip item
+        return -1;
+    }
+    const edoi = encodeURIComponent(doi);
+
+    const url = "https://inspirehep.net/api/" + idtype + "/" + edoi;
+    const response = await fetch(url)
+          .then(response => response.json())
+          .catch(err => null);
+
+    if (response === null) {
+        // Something went wrong
+        return -1;
+    }
+
+    let str = null;
+    try {
+        str = response['metadata']['citation_count'];
+    } catch (err) {
+        // There are no citation counts
+        return -1;
+    }
+
+    const count = parseInt(str);
+    return count;
+}
 
 // Preference managers
 
@@ -93,7 +180,7 @@ Zotero.CitationCounts.notifierCallback = {
 
 // Controls for Tools menu
 
-// *********** Set the checkbox checks, frompref
+// *********** Set the checkbox checks, from pref
 Zotero.CitationCounts.setCheck = function() {
     let tools_crossref = document.getElementById(
         "menu_Tools-citationcounts-menu-popup-crossref");
@@ -270,41 +357,33 @@ Zotero.CitationCounts.updateNextItem = function(operation) {
         operation);
 };
 
-Zotero.CitationCounts.updateItem = function(item, operation) {
+Zotero.CitationCounts.updateItem = async function(item, operation) {
     if (operation == "crossref") {
-        const doi = item.getField('DOI');
-        if (!doi) {
-            // There is no DOI; skip item
-            Zotero.CitationCounts.updateNextItem(operation);
-            return;
+
+        const count = await getCrossrefCount(item);
+        if (count >= 0) {
+            setCitationCount(item, 'Crossref', count);
+            item.saveTx();
+            Zotero.CitationCounts.counter++;
         }
+        Zotero.CitationCounts.updateNextItem(operation);
 
-        const edoi = encodeURIComponent(doi);
+    } else if (operation == "inspire") {
 
-        const style = "vnd.citationstyles.csl+json";
-        const xform = "transform/application/" + style;
-        const url = "https://api.crossref.org/works/" + edoi + "/" + xform;
+        const count_doi = await getInspireCount(item, 'doi');
+        const count_arxiv = await getInspireCount(item, 'arxiv');
+        if (count_doi >= 0 || count_arxiv >= 0) {
+            if (count_doi >= 0) {
+                setCitationCount(item, 'Inspire/DOI', count_doi);
+            }
+            if (count_arxiv >= 0) {
+                setCitationCount(item, 'Inspire/arXiv', count_arxiv);
+            }
+            item.saveTx();
+            Zotero.CitationCounts.counter++;
+        }
+        Zotero.CitationCounts.updateNextItem(operation);
 
-        fetch(url)
-            .then(response => response.json())
-            .then(data => {
-                if (!data.hasOwnProperty('is-referenced-by-count')) {
-                    // There are no citation counts
-                    Zotero.CitationCounts.updateNextItem(operation);
-                    return;
-                }
-
-                const str = data['is-referenced-by-count'];
-                const count = parseInt(str);
-                setCitationCount(item, 'Crossref', count);
-                item.saveTx();
-                Zotero.CitationCounts.counter++;
-                Zotero.CitationCounts.updateNextItem(operation);
-            })
-            .catch(err => {
-                // Something went wrong; we don't care what
-                Zotero.CitationCounts.updateNextItem(operation);
-            });
     } else {
         Zotero.CitationCounts.updateNextItem(operation);
     }

install.rdf

@@ -7,7 +7,7 @@
         RDF:about="urn:mozilla:install-manifest"
         em:id="[email protected]"
         em:name="Zotero Citation Counts Manager"
-        em:version="0.1.2"
+        em:version="0.1.3"
         em:type="2"
         em:creator="Erik Schnetter"
         em:description="Automatically fetch and update citation counts"

update.rdf

@@ -5,13 +5,13 @@
       <rdf:Seq>
         <rdf:li>
           <rdf:Description>
-            <em:version>0.1.2</em:version>
+            <em:version>0.1.3</em:version>
             <em:targetApplication>
               <rdf:Description>
                 <em:id>[email protected]</em:id>
                 <em:minVersion>5.0</em:minVersion>
                 <em:maxVersion>5.*</em:maxVersion>
-                <em:updateLink>https://github.com/eschnett/zotero-citationcounts/releases/download/v0.1.2/zotero-citationcounts-0.1.2.xpi</em:updateLink>
+                <em:updateLink>https://github.com/eschnett/zotero-citationcounts/releases/download/v0.1.3/zotero-citationcounts-0.1.3.xpi</em:updateLink>
               </rdf:Description>
             </em:targetApplication>
           </rdf:Description>