Update getMessagesAtom to include copyright information in original p…

…osition
DEFRA · Jun 10, 2024 · a6d41ec · a6d41ec
1 parent 318d65c
commit a6d41ec
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/lib/functions/getMessagesAtom.js b/lib/functions/getMessagesAtom.js
@@ -44,11 +44,13 @@ module.exports.getMessagesAtom = async (event) => {
     })
   }
 
-  let atomFeed = feed.atom1()
+  const atomFeed = feed.atom1()
 
   const copyrightText = 'Copyright (c) 2024, Environment Agency. Licensed under Creative Commons BY 4.0'
-  atomFeed = atomFeed.replace('</feed>', `<copyright>${copyrightText}</copyright></feed>`)
 
-  response.body = atomFeed
+  const position = atomFeed.indexOf('<entry>')
+  const feedWithCopyright = `${atomFeed.slice(0, position)}${copyrightText}${atomFeed.slice(position)}`
+
+  response.body = feedWithCopyright
   return response
 }