Skip to content

Commit

Permalink
extension: Add img encoding in readmes
Browse files Browse the repository at this point in the history
  • Loading branch information
JoaoMario109 committed Aug 1, 2024
1 parent 2d343c7 commit 133866b
Showing 1 changed file with 13 additions and 2 deletions.
15 changes: 13 additions & 2 deletions blueos_repository/extension/extension.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@
Image,
Platform,
)
from extension.utils.markdown_img_encoder import MarkdownImageEncoder
from logger import Logger
from utils import valid_semver

Expand Down Expand Up @@ -74,6 +75,11 @@ async def fetch_readme(url: str) -> str:
raise Exception(f"Could not get readme, expected type to be text/plain but got {resp.content_type}")
return await resp.text()

@staticmethod
async def process_readme_md(readme: str, resources_url: str) -> str:
encoder = MarkdownImageEncoder(readme, resources_url)
return str(await encoder.get_processed_markdown())

def __extract_images_from_tag(self, tag: Tag) -> List[Image]:
active_images = [
image
Expand Down Expand Up @@ -156,9 +162,14 @@ async def __create_version_from_tag_blob(self, version_tag: Tag, blob: Blob) ->

readme = labels.get("readme", None)
if readme is not None:
readme = readme.replace(r"{tag}", version_tag.name)
url = readme.replace(r"{tag}", version_tag.name)
try:
readme = await Extension.fetch_readme(readme)
readme = await Extension.fetch_readme(url)
try:
url = url.rsplit("/", 1)[0]
readme = await Extension.process_readme_md(readme, url)
except Exception as error: # pylint: disable=broad-except
Logger.warning(self.identifier, str(error))
except Exception as error: # pylint: disable=broad-except
Logger.warning(self.identifier, str(error))
readme = str(error)
Expand Down

0 comments on commit 133866b

Please sign in to comment.