From c56b5ec87bf43245cba94dc92c00b4cba448a74f Mon Sep 17 00:00:00 2001
From: zouap <zouap@pcl.ac.cn>
Date: Mon, 8 Nov 2021 16:11:08 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E6=B5=8B=E8=AF=95=E4=BB=A3?=
 =?UTF-8?q?=E7=A0=81=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: zouap <zouap@pcl.ac.cn>
---
 routers/repo/ai_model_manage.go          |  8 +++--
 templates/repo/modelmanage/download.tmpl | 59 ++++++++++++++++++++++++++++++++
 templates/repo/modelmanage/index.tmpl    |  2 +-
 3 files changed, 65 insertions(+), 4 deletions(-)
 create mode 100644 templates/repo/modelmanage/download.tmpl

diff --git a/routers/repo/ai_model_manage.go b/routers/repo/ai_model_manage.go
index b1ebe9478..2139df236 100644
--- a/routers/repo/ai_model_manage.go
+++ b/routers/repo/ai_model_manage.go
@@ -16,8 +16,9 @@ import (
 )
 
 const (
-	Model_prefix        = "aimodels/"
-	tplModelManageIndex = "repo/modelmanage/index"
+	Model_prefix           = "aimodels/"
+	tplModelManageIndex    = "repo/modelmanage/index"
+	tplModelManageDownload = "repo/modelmanage/download"
 )
 
 func SaveModelByParameters(jobId string, name string, version string, label string, description string, userId int64) error {
@@ -210,6 +211,7 @@ func ShowSingleModel(ctx *context.Context) {
 	ctx.Data["Dirs"] = models
 	ctx.Data["task"] = task
 	ctx.Data["ID"] = id
+	ctx.HTML(200, tplModelManageDownload)
 
 }
 
@@ -240,7 +242,7 @@ func ShowModelPageInfo(ctx *context.Context) {
 	ctx.Data["Page"] = pager
 	ctx.Data["PageIsCloudBrain"] = true
 	ctx.Data["Tasks"] = modelResult
-	ctx.HTML(200, tplModelManageIndex)
+
 }
 
 func ModifyModel(id string, description string) error {
diff --git a/templates/repo/modelmanage/download.tmpl b/templates/repo/modelmanage/download.tmpl
new file mode 100644
index 000000000..495e2e371
--- /dev/null
+++ b/templates/repo/modelmanage/download.tmpl
@@ -0,0 +1,59 @@
+{{template "base/head" .}}
+<div class="repository dataset dir-list view">
+	{{template "repo/header" .}}
+	<form class="ui container">
+    <div class="ui stackable grid {{if .Error}}hide{{end}}" id="dir-content">
+      <div class="row">
+        <div class="column sixteen wide">
+          <p>
+            {{ range $index, $item := .Path }}<a href='{{$.Link}}/?parentDir={{if gt $index 0}}{{DatasetPathJoin $.Path $index "/"}}{{else}}{{end}}'>{{ $item }}</a><span class="directory-seperator">/</span>{{ end }}
+          </p>
+        </div>
+      </div>
+    </div>
+
+    <div class="ui grid">
+      <div class="row">
+        <div class="ui sixteen wide column">
+          <div class="dir list">
+             <a class="title" href="{{$.RepoLink}}/modelmanage/downloadall?id={{.ID}}">
+                  下载所有
+            </a>
+            {{if .Dirs}}
+                <table id="repo-files-table" class="ui single line table">
+                  <tbody>
+                    {{range .Dirs}}
+                    <tr>
+                      <td class="name four wide">
+                        <span class="truncate">
+                            <span class="octicon octicon-file-directory"></span>
+                           <a class="title" href="{{if .IsDir}}{{$.RepoLink}}/modelmanage/downloadsingle?id={{$.ID}}&parentDir={{.ParenDir}}{{else}}{{$.RepoLink}}/modelmanage/downloadsingle?id={{$.ID}}&parentDir={{.ParenDir}}&fileName={{.FileName}}{{end}}">
+                              <span class="fitted">{{if .IsDir}} {{svg "octicon-file-directory" 16}}{{else}}{{svg "octicon-file" 16}}{{end}}</span> {{.FileName}}
+                           </a>
+                        </span>
+                      </td>
+                      <td class="message nine wide">
+                        <span class="truncate has-emoji">
+                          {{.Size | FileSize}}
+                        </span>
+                      </td>
+                      <td class="text right age three wide">
+                        <span class="time-since poping up">{{.ModTime}}</span>
+                      </td>
+                    </tr>
+                    {{end}}
+                  </tbody>
+                </table>
+
+            {{end}}
+
+          </div>
+        </div>
+      </div>
+    </div>
+  </form>
+</div>
+
+
+
+{{template "base/footer" .}}
diff --git a/templates/repo/modelmanage/index.tmpl b/templates/repo/modelmanage/index.tmpl
index a60c8fda3..3be22ca9e 100644
--- a/templates/repo/modelmanage/index.tmpl
+++ b/templates/repo/modelmanage/index.tmpl
@@ -308,7 +308,7 @@
                                    
                                     <div class="ui compact buttons" style="margin-right:10px;">
                                             <!-- 模型下载 -->
-                                            <a  class="ui basic blue button" href="{{$.Link}}/{{.ID}}/modelmanage" target="_blank">
+                                            <a  class="ui basic blue button" href="{{$.Link}}/{{.ID}}" target="_blank">
                                                 {{$.i18n.Tr "repo.model_download"}}
                                             </a>