Carlos Florêncio
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
"""Minimal Go import parsing shared by import-direction rules."""
|
|
|
|
import re
|
|
|
|
|
|
IMPORT_LINE = re.compile(r'^\s*(?:(?:[A-Za-z_][\w]*|\.|_)\s+)?["`]([^"`]+)["`]')
|
|
SINGLE_IMPORT = re.compile(
|
|
r'^\s*import\s+(?:(?:[A-Za-z_][\w]*|\.|_)\s+)?["`]([^"`]+)["`]'
|
|
)
|
|
|
|
|
|
def import_lines(source: str) -> list[tuple[int, str]]:
|
|
imports: list[tuple[int, str]] = []
|
|
in_block = False
|
|
for line_number, line in enumerate(source.splitlines(), start=1):
|
|
stripped = line.strip()
|
|
if not in_block:
|
|
if re.match(r"^import\s*\(", stripped):
|
|
in_block = True
|
|
continue
|
|
match = SINGLE_IMPORT.match(line)
|
|
if match:
|
|
imports.append((line_number, match.group(1)))
|
|
continue
|
|
if stripped.startswith(")"):
|
|
in_block = False
|
|
continue
|
|
match = IMPORT_LINE.match(line)
|
|
if match:
|
|
imports.append((line_number, match.group(1)))
|
|
return imports
|
|
|
|
|
|
def is_import_at_or_below(import_path: str, root: str) -> bool:
|
|
return import_path == root or import_path.startswith(root + "/")
|