add codemeta_doi postprocess plugin

Author: notactuallyfinn

pyproject.toml

@@ -70,6 +70,7 @@ rodare = "hermes.commands.deposit.rodare:RodareDepositPlugin"
 config_invenio_record_id = "hermes.commands.postprocess.invenio:config_record_id"
 config_invenio_rdm_record_id = "hermes.commands.postprocess.invenio_rdm:config_record_id"
 cff_doi = "hermes.commands.postprocess.invenio:cff_doi"
+codemeta_doi = "hermes.commands.postprocess.invenio:codemeta_doi"
 
 [project.entry-points."hermes.process"]
 codemeta = "hermes.commands.process.standard_merge:CodemetaProcessPlugin"

src/hermes/commands/postprocess/invenio.py

@@ -6,6 +6,7 @@
 # SPDX-FileContributor: Michael Fritzsche
 # SPDX-FileContributor: Stephan Druskat
 
+import json
 import logging
 
 from ruamel.yaml import YAML
@@ -73,3 +74,28 @@ def __call__(self, command: HermesCommand):
             yaml.dump(cff, open('CITATION.cff', 'w'))
         except Exception as e:
             raise RuntimeError("Update of CITATION.cff failed.") from e
+
+
+class codemeta_doi(HermesPostprocessPlugin):
+    def __call__(self, command: HermesCommand):
+        ctx = HermesContext()
+        ctx.prepare_step("deposit")
+        with ctx["invenio"] as manager:
+            deposition = manager["result"]
+        ctx.finalize_step("deposit")
+
+        try:
+            with open("codemeta.json", "r") as file:
+                codemeta = json.load(file)
+            if "@id" not in codemeta:
+                codemeta["@id"] = deposition['doi']
+            if "referencePublication" not in codemeta:
+                codemeta["referencePublication"] = deposition['doi']
+            elif isinstance(codemeta["referencePublication"], list):
+                codemeta["referencePublication"].append(deposition['doi'])
+            else:
+                codemeta["referencePublication"] = [codemeta["referencePublication"], deposition['doi']]
+            with open("codemeta.json", "w") as file:
+                json.dump(codemeta, file)
+        except Exception as e:
+            raise RuntimeError("Update of CITATION.cff failed.") from e

test/hermes_test/commands/postprocess/test_invenio_postprocess.py

@@ -4,6 +4,7 @@
 
 # SPDX-FileContributor: Michael Fritzsche
 
+import json
 import sys
 
 from ruamel import yaml
@@ -23,10 +24,20 @@ def test_invenio_postprocess(tmp_path, monkeypatch):
     citation_file = tmp_path / "CITATION.cff"
     citation_file.write_text("cff-version: 1.2.0\ntitle: Test")
 
+    codemeta_file = tmp_path / "codemeta.json"
+    codemeta_file.write_text(
+        """{
+    "@context": "https://doi.org/10.5063/schema/codemeta-2.0",
+    "type": "SoftwareSourceCode",
+    "name": "Test"
+}
+"""
+    )
+
     config_file = tmp_path / "hermes.toml"
     config_file.write_text(
         """[postprocess]
-run = ["config_invenio_record_id", "cff_doi"]
+run = ["config_invenio_record_id", "cff_doi", "codemeta_doi"]
 [deposit.invenio]
 site_url = "https://zenodo.org"
 """
@@ -45,11 +56,12 @@ def test_invenio_postprocess(tmp_path, monkeypatch):
     finally:
         result_toml = toml.load(config_file)
         result_cff = yaml.YAML().load(citation_file)
+        result_codemeta = json.loads(codemeta_file.read_text())
         sys.argv = orig_argv
 
     assert result_toml == toml.loads(
         """[postprocess]
-run = ["config_invenio_record_id", "cff_doi"]
+run = ["config_invenio_record_id", "cff_doi", "codemeta_doi"]
 [deposit.invenio]
 site_url = "https://zenodo.org"
 record_id = "foo"
@@ -64,3 +76,13 @@ def test_invenio_postprocess(tmp_path, monkeypatch):
     description: DOI for the published version 1.0.0 [generated by hermes]
 """
     )
+    assert result_codemeta == json.loads(
+        """{
+    "@context": "https://doi.org/10.5063/schema/codemeta-2.0",
+    "type": "SoftwareSourceCode",
+    "@id": "my_doi",
+    "name": "Test",
+    "referencePublication": "my_doi"
+}
+"""
+    )