refactor(parser): use pyarn for parsing yarn.lock

adfinis · Feb 26, 2024 · 4042fd8 · 4042fd8
1 parent b77ae99
commit 4042fd8
Show file tree

Hide file tree

Showing 3 changed files with 285 additions and 278 deletions.
diff --git a/api/outdated/outdated/parser.py b/api/outdated/outdated/parser.py
@@ -6,6 +6,7 @@
 import requests
 from dateutil.parser import parse as parse_date
 from django.conf import settings
+from pyarn.lockfile import Lockfile as YarnLockfile
 from semver import Version as SemVer
 from tomllib import loads
 from yaml import safe_load
@@ -143,12 +144,13 @@ def parse(self) -> list[models.Version]:
 
             dependencies = []
             if name == "yarn.lock":
-                regex = r'"?([\S]+)@.*:\n  version "(.+)"'
+                yarn_lockfile = YarnLockfile.from_file(lockfile)
                 dependencies = [
-                    dependency
-                    for dependency in findall(regex, data)
-                    if dependency[0] in settings.TRACKED_DEPENDENCIES
+                    (dependency.name, dependency.version)
+                    for dependency in yarn_lockfile.packages()
+                    if dependency.name in settings.TRACKED_DEPENDENCIES
                 ]
+
             elif name == "poetry.lock":
                 dependencies = [
                     (dependency["name"], dependency["version"])