[proxy] display some more details in mapping/resource version view in the web UI

1 files changed, 129 insertions, 0 deletions

diff --git a/src/hydrilla/proxy/web_ui/items.py b/src/hydrilla/proxy/web_ui/items.py
index f34b89b..467be2f 100644
--- a/src/hydrilla/proxy/web_ui/items.py
+++ b/src/hydrilla/proxy/web_ui/items.py
@@ -34,6 +34,8 @@ from __future__ import annotations
 
 import typing as t
 
+from urllib.parse import unquote
+
 import flask
 import werkzeug
 
@@ -231,3 +233,130 @@ def alter_library_version(item_version_id: str) -> werkzeug.Response:
 @bp.route('/packages/viewversion/<string:item_version_id>', methods=['POST'])
 def alter_package_version(item_version_id: str) -> werkzeug.Response:
     return alter_item_version(item_version_id, item_infos.ItemType.MAPPING)
+
+def show_file(
+        item_version_id: str,
+        item_type:       item_infos.ItemType,
+        file_type:       str,
+        name:            str,
+) -> werkzeug.Response:
+    if file_type not in ('license', 'web_resource'):
+        flask.abort(404)
+
+    try:
+        store = item_version_store(_app.get_haketilo_state(), item_type)
+        item_version_ref = store.get(item_version_id)
+
+        try:
+            if file_type == 'license':
+                file_data = item_version_ref.get_license_file(name)
+            else:
+                assert isinstance(item_version_ref, st.ResourceVersionRef)
+                file_data = item_version_ref.get_resource_file(name)
+
+            return werkzeug.Response(
+                file_data.contents,
+                mimetype = file_data.mime_type
+            )
+        except st.MissingItemError:
+            if file_type == 'license':
+                url = item_version_ref.get_upstream_license_file_url(name)
+            else:
+                assert isinstance(item_version_ref, st.ResourceVersionRef)
+                url = item_version_ref.get_upstream_resource_file_url(name)
+
+            return flask.redirect(url)
+
+    except st.MissingItemError:
+        flask.abort(404)
+
+@bp.route('/packages/viewversion/<string:item_version_id>/<string:file_type>/<path:name>')
+def show_mapping_file(item_version_id: str, file_type: str, name: str) \
+    -> werkzeug.Response:
+    item_type = item_infos.ItemType.MAPPING
+    return show_file(item_version_id, item_type, file_type, name)
+
+@bp.route('/libraries/viewversion/<string:item_version_id>/<string:file_type>/<path:name>')
+def show_resource_file(item_version_id: str, file_type: str, name: str) \
+    -> werkzeug.Response:
+    item_type = item_infos.ItemType.RESOURCE
+    return show_file(item_version_id, item_type, file_type, name)
+
+@bp.route('/libraries/viewdep/<string:item_version_id>/<string:dep_identifier>')
+def show_library_dep(item_version_id: str, dep_identifier: str) \
+    -> werkzeug.Response:
+    state = _app.get_haketilo_state()
+
+    try:
+        store = state.resource_version_store()
+        dep_id = store.get(item_version_id).get_dependency(dep_identifier).id
+        url = flask.url_for('.show_library_version', item_version_id=dep_id)
+    except st.MissingItemError:
+        try:
+            versionless_store = state.resource_store()
+            item_ref = versionless_store.get_by_identifier(dep_identifier)
+            url = flask.url_for('.show_library', item_id=item_ref.id)
+        except st.MissingItemError:
+            flask.abort(404)
+
+    return flask.redirect(url)
+
+@bp.route('/<string:item_type>/viewrequired/<string:item_version_id>/<string:required_identifier>')
+def show_required_mapping(
+        item_type:           str,
+        item_version_id:     str,
+        required_identifier: str
+) -> werkzeug.Response:
+    state = _app.get_haketilo_state()
+
+    if item_type not in ('package', 'library'):
+        flask.abort(404)
+
+    found = False
+
+    if item_type == 'package':
+        try:
+            ref = state.mapping_version_store().get(item_version_id)
+            mapping_ver_id = ref.get_required_mapping(required_identifier).id
+            url = flask.url_for(
+                '.show_package_version',
+                item_version_id = mapping_ver_id
+            )
+            found = True
+        except st.MissingItemError:
+            pass
+
+    if not found:
+        try:
+            versionless_store = state.mapping_store()
+            mapping_ref = versionless_store\
+                .get_by_identifier(required_identifier)
+            url = flask.url_for('.show_package', item_id=mapping_ref.id)
+        except st.MissingItemError:
+            flask.abort(404)
+
+    return flask.redirect(url)
+
+@bp.route('/package/viewpayload/<string:item_version_id>/<string:pattern>/<string:lib_identifier>')
+def show_payload(item_version_id: str, pattern: str, lib_identifier: str) \
+    -> werkzeug.Response:
+    state = _app.get_haketilo_state()
+
+    try:
+        ref = state.mapping_version_store().get(item_version_id)
+
+        try:
+            resource_ver_ref = \
+                ref.get_payload_resource(unquote(pattern), lib_identifier)
+            url = flask.url_for(
+                '.show_library_version',
+                item_version_id = resource_ver_ref.id
+            )
+        except st.MissingItemError:
+            resource_ref = \
+                state.resource_store().get_by_identifier(lib_identifier)
+            url = flask.url_for('.show_library', item_id=resource_ref.id)
+    except st.MissingItemError:
+        flask.abort(404)
+
+    return flask.redirect(url)