Mercurial > projects > managesieve
comparison cmd/sievemgr/put.go @ 0:6369453d47a3
Initial revision
author | Guido Berhoerster <guido+managesieve@berhoerster.name> |
---|---|
date | Thu, 15 Oct 2020 09:11:05 +0200 |
parents | |
children |
comparison
equal
deleted
inserted
replaced
-1:000000000000 | 0:6369453d47a3 |
---|---|
1 // Copyright (C) 2020 Guido Berhoerster <guido+managesieve@berhoerster.name> | |
2 // | |
3 // Permission is hereby granted, free of charge, to any person obtaining | |
4 // a copy of this software and associated documentation files (the | |
5 // "Software"), to deal in the Software without restriction, including | |
6 // without limitation the rights to use, copy, modify, merge, publish, | |
7 // distribute, sublicense, and/or sell copies of the Software, and to | |
8 // permit persons to whom the Software is furnished to do so, subject to | |
9 // the following conditions: | |
10 // | |
11 // The above copyright notice and this permission notice shall be included | |
12 // in all copies or substantial portions of the Software. | |
13 // | |
14 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, | |
15 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF | |
16 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. | |
17 // IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY | |
18 // CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, | |
19 // TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE | |
20 // SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. | |
21 | |
22 package main | |
23 | |
24 import ( | |
25 "fmt" | |
26 "io" | |
27 "net" | |
28 "os" | |
29 | |
30 "go.guido-berhoerster.org/managesieve" | |
31 ) | |
32 | |
33 func init() { | |
34 cmdPut.Flag.StringVar(&username, "u", "", "Set the username") | |
35 cmdPut.Flag.StringVar(&passwordFilename, "P", "", | |
36 "Set the name of the password file") | |
37 } | |
38 | |
39 var cmdPut = &command{ | |
40 UsageLine: "put [options] host[:port] name [file]", | |
41 Run: runPut, | |
42 } | |
43 | |
44 func runPut(cmd *command, args []string) error { | |
45 var scriptName string | |
46 var r io.Reader = os.Stdin | |
47 var host, port string | |
48 var err error | |
49 switch len(args) { | |
50 case 3: // name and filename | |
51 scriptFile, err := os.Open(args[2]) | |
52 if err != nil { | |
53 return fmt.Errorf("failed to open script file: %s\n", | |
54 err) | |
55 } | |
56 defer scriptFile.Close() | |
57 r = scriptFile | |
58 fallthrough | |
59 case 2: // only name | |
60 host, port, err = parseHostPort(args[0]) | |
61 if err != nil { | |
62 return err | |
63 } | |
64 | |
65 scriptName = args[1] | |
66 default: | |
67 return usageError("invalid number of arguments") | |
68 } | |
69 | |
70 script, err := readLimitedString(r, managesieve.ReadLimit) | |
71 if err != nil { | |
72 return fmt.Errorf("failed to read script: %s\n", err) | |
73 } | |
74 | |
75 username, password, err := usernamePassword(host, port, username, | |
76 passwordFilename) | |
77 if err != nil { | |
78 return err | |
79 } | |
80 | |
81 c, err := dialPlainAuth(net.JoinHostPort(host, port), username, | |
82 password) | |
83 if err != nil { | |
84 return err | |
85 } | |
86 defer c.Logout() | |
87 | |
88 if err := c.PutScript(scriptName, script); err != nil { | |
89 return err | |
90 } | |
91 | |
92 return nil | |
93 } |