66 "github.com/qiniu/api.v7/storage"
77 "github.com/qiniu/qshell/iqshell"
88 "github.com/spf13/cobra"
9- "github.com/spf13/viper"
109 "os"
1110 "strings"
1211 "time"
2322 mimeType string
2423 fileType int
2524 workerCount int
26- upHost string
25+ rupHost string
26+ fupHost string
2727 callbackUrls string
2828 callbackHost string
2929)
@@ -47,21 +47,18 @@ func init() {
4747 formPutCmd .Flags ().StringVarP (& mimeType , "mimetype" , "t" , "" , "file mime type" )
4848 formPutCmd .Flags ().IntVarP (& fileType , "storage" , "s" , 0 , "storage type" )
4949 formPutCmd .Flags ().IntVarP (& workerCount , "worker" , "c" , 16 , "worker count" )
50- formPutCmd .Flags ().StringVarP (& upHost , "up-host" , "u" , "" , "uphost" )
50+ formPutCmd .Flags ().StringVarP (& fupHost , "up-host" , "u" , "" , "uphost" )
5151 formPutCmd .Flags ().StringVarP (& callbackUrls , "callback-urls" , "l" , "" , "upload callback urls, separated by comma" )
5252 formPutCmd .Flags ().StringVarP (& callbackHost , "callback-host" , "T" , "" , "upload callback host" )
5353
5454 RePutCmd .Flags ().BoolVarP (& pOverwrite , "overwrite" , "w" , false , "overwrite mode" )
5555 RePutCmd .Flags ().StringVarP (& mimeType , "mimetype" , "t" , "" , "file mime type" )
5656 RePutCmd .Flags ().IntVarP (& fileType , "storage" , "s" , 0 , "storage type" )
5757 RePutCmd .Flags ().IntVarP (& workerCount , "worker" , "c" , 16 , "worker count" )
58- RePutCmd .Flags ().StringVarP (& upHost , "up-host" , "u" , "" , "uphost" )
58+ RePutCmd .Flags ().StringVarP (& rupHost , "up-host" , "u" , "" , "uphost" )
5959 RePutCmd .Flags ().StringVarP (& callbackUrls , "callback-urls" , "l" , "" , "upload callback urls, separated by comma" )
6060 RePutCmd .Flags ().StringVarP (& callbackHost , "callback-host" , "T" , "" , "upload callback host" )
6161
62- viper .BindPFlag ("hosts.up_host" , formPutCmd .Flags ().Lookup ("up-host" ))
63- viper .BindPFlag ("hosts.up_host" , RePutCmd .Flags ().Lookup ("up-host" ))
64-
6562 RootCmd .AddCommand (formPutCmd , RePutCmd )
6663}
6764
@@ -105,16 +102,16 @@ func FormPut(cmd *cobra.Command, params []string) {
105102 }
106103
107104 var putExtra storage.PutExtra
105+ var upHost string
108106
109- if upHost ! = "" {
110- if ! strings . HasPrefix ( upHost , "http" ) {
111- upHost = "http://" + upHost
112- }
113- putExtra = storage. PutExtra {
114- UpHost : upHost ,
115- }
107+ if fupHost = = "" {
108+ upHost = iqshell . UpHost ()
109+ } else {
110+ upHost = fupHost
111+ }
112+ putExtra = storage. PutExtra {
113+ UpHost : upHost ,
116114 }
117-
118115 if mimeType != "" {
119116 putExtra .MimeType = mimeType
120117 }
@@ -223,13 +220,15 @@ func ResumablePut(cmd *cobra.Command, params []string) {
223220 }
224221
225222 var putExtra storage.RputExtra
226- if upHost != "" {
227- if ! strings .HasPrefix (upHost , "http" ) {
228- upHost = "http://" + upHost
229- }
230- putExtra = storage.RputExtra {
231- UpHost : upHost ,
232- }
223+ var upHost string
224+
225+ if rupHost == "" {
226+ upHost = iqshell .UpHost ()
227+ } else {
228+ upHost = rupHost
229+ }
230+ putExtra = storage.RputExtra {
231+ UpHost : upHost ,
233232 }
234233 if mimeType != "" {
235234 putExtra .MimeType = mimeType
@@ -250,6 +249,7 @@ func ResumablePut(cmd *cobra.Command, params []string) {
250249
251250 resume_uploader := storage .NewResumeUploader (nil )
252251 err = resume_uploader .PutFile (context .Background (), & putRet , uptoken , key , localFile , & putExtra )
252+
253253 fmt .Println ()
254254 if err != nil {
255255 if v , ok := err .(* storage.ErrorInfo ); ok {
0 commit comments