feat: add insecure options for pull (#37)

Signed-off-by: Gaius <[email protected]>

Author: gaius-qi

cmd/pull.go

@@ -46,6 +46,7 @@ var pullCmd = &cobra.Command{
 func init() {
 	flags := pullCmd.Flags()
 	flags.BoolVar(&pullConfig.PlainHTTP, "plain-http", false, "use plain HTTP instead of HTTPS")
+	flags.BoolVar(&pullConfig.Insecure, "insecure", false, "use insecure connection for the pull operation and skip TLS verification")
 	flags.StringVar(&pullConfig.Proxy, "proxy", "", "use proxy for the pull operation")
 	flags.StringVar(&pullConfig.ExtractDir, "extract-dir", "", "specify the extract dir for extracting the model artifact")
 
@@ -65,7 +66,7 @@ func runPull(ctx context.Context, target string) error {
 		return fmt.Errorf("target is required")
 	}
 
-	opts := []backend.Option{}
+	opts := []backend.Option{backend.WithInsecure(pullConfig.Insecure)}
 	if pullConfig.PlainHTTP {
 		opts = append(opts, backend.WithPlainHTTP())
 	}

pkg/backend/options.go

@@ -21,6 +21,7 @@ type Option func(*Options)
 type Options struct {
 	plainHTTP bool
 	proxy     string
+	insecure  bool
 	output    string
 }
 
@@ -38,6 +39,13 @@ func WithProxy(proxy string) Option {
 	}
 }
 
+// WithInsecure sets the insecure option.
+func WithInsecure(insecure bool) Option {
+	return func(opts *Options) {
+		opts.insecure = insecure
+	}
+}
+
 // WithOutput sets the output option.
 func WithOutput(output string) Option {
 	return func(opts *Options) {

pkg/backend/pull.go

@@ -18,6 +18,7 @@ package backend
 
 import (
 	"context"
+	"crypto/tls"
 	"encoding/json"
 	"fmt"
 	"io"
@@ -70,6 +71,9 @@ func (b *backend) Pull(ctx context.Context, target string, opts ...Option) error
 
 		httpClient.Transport = &http.Transport{
 			Proxy: http.ProxyURL(proxyURL),
+			TLSClientConfig: &tls.Config{
+				InsecureSkipVerify: options.insecure,
+			},
 		}
 	}
 

pkg/config/pull.go

@@ -19,13 +19,15 @@ package config
 type Pull struct {
 	PlainHTTP  bool
 	Proxy      string
+	Insecure   bool
 	ExtractDir string
 }
 
 func NewPull() *Pull {
 	return &Pull{
 		PlainHTTP:  false,
 		Proxy:      "",
+		Insecure:   false,
 		ExtractDir: "",
 	}
 }