From 00f2528f8954a0e5bf26e6df6bad5bf3151f5091 Mon Sep 17 00:00:00 2001
From: gek64 <10346089+gek64@users.noreply.github.com>
Date: Wed, 28 Feb 2024 11:23:59 +0800
Subject: [PATCH] update

---
 go.mod  |  4 ++--
 go.sum  |  4 ++++
 main.go | 37 +++++++++++++++++++++++--------------
 3 files changed, 29 insertions(+), 16 deletions(-)

diff --git a/go.mod b/go.mod
index d6ec39a..5fec2ee 100644
--- a/go.mod
+++ b/go.mod
@@ -3,8 +3,8 @@ module redl
 go 1.21.4
 
 require (
-	github.com/gek64/gek v1.1.2-0.20231222091541-e37b3f00c2a9
-	github.com/urfave/cli/v2 v2.26.0
+	github.com/gek64/gek v1.1.2-0.20240122081904-9f44691c0810
+	github.com/urfave/cli/v2 v2.27.1
 )
 
 require (
diff --git a/go.sum b/go.sum
index 50cefb4..786132f 100644
--- a/go.sum
+++ b/go.sum
@@ -2,9 +2,13 @@ github.com/cpuguy83/go-md2man/v2 v2.0.3 h1:qMCsGGgs+MAzDFyp9LpAe1Lqy/fY/qCovCm0q
 github.com/cpuguy83/go-md2man/v2 v2.0.3/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46tRHOmNcaadrF8o=
 github.com/gek64/gek v1.1.2-0.20231222091541-e37b3f00c2a9 h1:3+foBBZXBQm3ZugExmeO9EdQkn27GyP+f0Er61WxhqU=
 github.com/gek64/gek v1.1.2-0.20231222091541-e37b3f00c2a9/go.mod h1:oalulJSH0OM6LRbjGwpw305/d7h9VE5P1wDbDR0KUyQ=
+github.com/gek64/gek v1.1.2-0.20240122081904-9f44691c0810 h1:7NmGfZLYP3no24WA7xookJK5BEA6IgoqJEOC48Kt4rY=
+github.com/gek64/gek v1.1.2-0.20240122081904-9f44691c0810/go.mod h1:35dDRHKelCrseBkA6Gjfg9FPOQhAVD/JluMzGqeQ5z8=
 github.com/russross/blackfriday/v2 v2.1.0 h1:JIOH55/0cWyOuilr9/qlrm0BSXldqnqwMsf35Ld67mk=
 github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
 github.com/urfave/cli/v2 v2.26.0 h1:3f3AMg3HpThFNT4I++TKOejZO8yU55t3JnnSr4S4QEI=
 github.com/urfave/cli/v2 v2.26.0/go.mod h1:8qnjx1vcq5s2/wpsqoZFndg2CE5tNFyrTvS6SinrnYQ=
+github.com/urfave/cli/v2 v2.27.1 h1:8xSQ6szndafKVRmfyeUMxkNUJQMjL1F2zmsZ+qHpfho=
+github.com/urfave/cli/v2 v2.27.1/go.mod h1:8qnjx1vcq5s2/wpsqoZFndg2CE5tNFyrTvS6SinrnYQ=
 github.com/xrash/smetrics v0.0.0-20231213231151-1d8dd44e695e h1:+SOyEddqYF09QP7vr7CgJ1eti3pY9Fn3LHO1M1r/0sI=
 github.com/xrash/smetrics v0.0.0-20231213231151-1d8dd44e695e/go.mod h1:N3UwUGtsrSj3ccvlPHLoLsHnpR27oXr4ZE984MbSER8=
diff --git a/main.go b/main.go
index 3d3f989..cbe19c8 100644
--- a/main.go
+++ b/main.go
@@ -11,24 +11,31 @@ import (
 )
 
 func main() {
-	var github_release string
-	var sourceforge_release string
+	var github string
+	var gitlab string
+	var sourceforge string
 	var included_parts cli.StringSlice
 	var excluded_parts cli.StringSlice
 	var output string
 
 	flags := []cli.Flag{
 		&cli.StringFlag{
-			Name:        "github_repository",
-			Aliases:     []string{"gr"},
-			Usage:       "set github release url",
-			Destination: &github_release,
+			Name:        "github",
+			Aliases:     []string{"gh"},
+			Usage:       "set github repo url",
+			Destination: &github,
 		},
 		&cli.StringFlag{
-			Name:        "sourceforge_release",
-			Aliases:     []string{"sr"},
-			Usage:       "set sourceforge release url",
-			Destination: &sourceforge_release,
+			Name:        "gitlab",
+			Aliases:     []string{"gl"},
+			Usage:       "set gitlab repo url",
+			Destination: &gitlab,
+		},
+		&cli.StringFlag{
+			Name:        "sourceforge",
+			Aliases:     []string{"sf"},
+			Usage:       "set sourceforge repo url",
+			Destination: &sourceforge,
 		},
 		&cli.StringSliceFlag{
 			Name:        "included_parts",
@@ -63,13 +70,15 @@ func main() {
 			var downloadLink string
 
 			// 获取下载链接
-			if github_release != "" {
-				downloadLink, err = internal.GetGithubDownloadLink(github_release, included_parts.Value(), excluded_parts.Value())
+			if github != "" {
+				downloadLink, err = internal.GetGithubDownloadLink(github, included_parts.Value(), excluded_parts.Value())
 				if err != nil {
 					return err
 				}
-			} else if sourceforge_release != "" {
-				downloadLink, err = internal.GetSourceForgeDownloadLink(sourceforge_release, included_parts.Value(), excluded_parts.Value())
+			} else if gitlab != "" {
+				fmt.Println("download release from gitlab is under development")
+			} else if sourceforge != "" {
+				downloadLink, err = internal.GetSourceForgeDownloadLink(sourceforge, included_parts.Value(), excluded_parts.Value())
 				if err != nil {
 					return err
 				}