blob: 699a4f700a25f63fed3deb45147f5b996cf387f0 [file] [log] [blame]
Scott Baker2c0ebda2019-05-06 16:55:47 -07001/*
2 * Portions copyright 2019-present Open Networking Foundation
3 * Original copyright 2019-present Ciena Corporation
4 *
5 * Licensed under the Apache License, Version 2.0 (the "License");
6 * you may not use this file except in compliance with the License.
7 * You may obtain a copy of the License at
8 *
9 * http://www.apache.org/licenses/LICENSE-2.0
10 *
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
16 */
17package commands
18
19import (
Scott Baker2c0ebda2019-05-06 16:55:47 -070020 "errors"
Scott Baker72efd752019-05-15 13:12:20 -070021 "fmt"
Scott Baker2c0ebda2019-05-06 16:55:47 -070022 flags "github.com/jessevdk/go-flags"
Scott Baker2c0ebda2019-05-06 16:55:47 -070023 "strings"
24)
25
26const (
27 DEFAULT_TRANSFER_FORMAT = "table{{ .Status }}\t{{ .Checksum }}\t{{ .Chunks }}\t{{ .Bytes }}"
28)
29
30type TransferOutput struct {
31 Status string `json:"status"`
32 Checksum string `json:"checksum"`
33 Chunks int `json:"chunks"`
34 Bytes int `json:"bytes"`
35}
36
37type TransferUpload struct {
38 OutputOptions
39 ChunkSize int `short:"h" long:"chunksize" default:"65536" description:"Host and port"`
40 Args struct {
41 LocalFileName string
42 URI string
43 } `positional-args:"yes" required:"yes"`
44}
45
46type TransferDownload struct {
47 OutputOptions
48 Args struct {
49 URI string
50 LocalFileName string
51 } `positional-args:"yes" required:"yes"`
52}
53
54type TransferOpts struct {
55 Upload TransferUpload `command:"upload"`
56 Download TransferDownload `command:"download"`
57}
58
59var transferOpts = TransferOpts{}
60
61func RegisterTransferCommands(parser *flags.Parser) {
62 parser.AddCommand("transfer", "file transfer commands", "Commands to transfer files to and from XOS", &transferOpts)
63}
64
Scott Baker2c0ebda2019-05-06 16:55:47 -070065/* Command processors */
66
67func (options *TransferUpload) Execute(args []string) error {
68
Scott Baker6cf525a2019-05-09 12:25:08 -070069 conn, descriptor, err := InitReflectionClient()
Scott Baker2c0ebda2019-05-06 16:55:47 -070070 if err != nil {
71 return err
72 }
73 defer conn.Close()
74
75 local_name := options.Args.LocalFileName
76 uri := options.Args.URI
77
Scott Baker6cf525a2019-05-09 12:25:08 -070078 if IsFileUri(local_name) {
79 return errors.New("local_name argument should not be a uri")
Scott Baker2c0ebda2019-05-06 16:55:47 -070080 }
81
Scott Baker6cf525a2019-05-09 12:25:08 -070082 if !IsFileUri(uri) {
83 return errors.New("uri argument should be a file:// uri")
Scott Baker2c0ebda2019-05-06 16:55:47 -070084 }
85
Scott Baker72efd752019-05-15 13:12:20 -070086 h, upload_result, err := UploadFile(conn, descriptor, local_name, uri, options.ChunkSize)
87
88 if upload_result.GetFieldByName("checksum").(string) != h.GetChecksum() {
89 return fmt.Errorf("Checksum mismatch, expected=%s, received=%s",
90 h.GetChecksum(),
91 upload_result.GetFieldByName("checksum").(string))
92 }
Scott Baker2c0ebda2019-05-06 16:55:47 -070093
Scott Baker2c0ebda2019-05-06 16:55:47 -070094 data := make([]TransferOutput, 1)
Scott Baker72efd752019-05-15 13:12:20 -070095 data[0].Checksum = upload_result.GetFieldByName("checksum").(string)
96 data[0].Chunks = int(upload_result.GetFieldByName("chunks_received").(int32))
97 data[0].Bytes = int(upload_result.GetFieldByName("bytes_received").(int32))
98 data[0].Status = GetEnumValue(upload_result, "status")
Scott Baker2c0ebda2019-05-06 16:55:47 -070099
Scott Baker5281d002019-05-16 10:45:26 -0700100 FormatAndGenerateOutput(&options.OutputOptions, DEFAULT_TRANSFER_FORMAT, "{{.Status}}", data)
Scott Baker2c0ebda2019-05-06 16:55:47 -0700101
102 return nil
103}
104
105func IsFileUri(s string) bool {
106 return strings.HasPrefix(s, "file://")
107}
108
109func (options *TransferDownload) Execute(args []string) error {
Scott Baker6cf525a2019-05-09 12:25:08 -0700110 conn, descriptor, err := InitReflectionClient()
Scott Baker2c0ebda2019-05-06 16:55:47 -0700111 if err != nil {
112 return err
113 }
114 defer conn.Close()
115
116 local_name := options.Args.LocalFileName
117 uri := options.Args.URI
118
119 if IsFileUri(local_name) {
120 return errors.New("local_name argument should not be a uri")
121 }
122
123 if !IsFileUri(uri) {
124 return errors.New("uri argument should be a file:// uri")
125 }
126
Scott Baker6cf525a2019-05-09 12:25:08 -0700127 h, err := DownloadFile(conn, descriptor, uri, local_name)
Scott Baker2c0ebda2019-05-06 16:55:47 -0700128 if err != nil {
129 return err
130 }
131
Scott Baker2c0ebda2019-05-06 16:55:47 -0700132 data := make([]TransferOutput, 1)
133 data[0].Chunks = h.chunks
134 data[0].Bytes = h.bytes
135 data[0].Status = h.status
Scott Baker5281d002019-05-16 10:45:26 -0700136 data[0].Checksum = h.GetChecksum()
Scott Baker2c0ebda2019-05-06 16:55:47 -0700137
Scott Baker5281d002019-05-16 10:45:26 -0700138 FormatAndGenerateOutput(&options.OutputOptions, DEFAULT_TRANSFER_FORMAT, "{{.Status}}", data)
Scott Baker2c0ebda2019-05-06 16:55:47 -0700139
140 return nil
141}