source: code/trunk/service.go@ 571

Last change on this file since 571 was 570, checked in by contact, 4 years ago

Merge "change-password" into "user update"

Add a -password flag to the "user update" command.

File size: 23.0 KB
RevLine 
[117]1package soju
2
3import (
[307]4 "crypto"
5 "crypto/ecdsa"
6 "crypto/ed25519"
7 "crypto/elliptic"
8 "crypto/rand"
9 "crypto/rsa"
10 "crypto/sha1"
11 "crypto/sha256"
12 "crypto/x509"
13 "crypto/x509/pkix"
14 "encoding/hex"
15 "errors"
[120]16 "flag"
[117]17 "fmt"
[120]18 "io/ioutil"
[307]19 "math/big"
[339]20 "sort"
[542]21 "strconv"
[117]22 "strings"
[307]23 "time"
[566]24 "unicode"
[117]25
[252]26 "golang.org/x/crypto/bcrypt"
[117]27 "gopkg.in/irc.v3"
28)
29
30const serviceNick = "BouncerServ"
[478]31const serviceNickCM = "bouncerserv"
[343]32const serviceRealname = "soju bouncer service"
[117]33
[220]34var servicePrefix = &irc.Prefix{
35 Name: serviceNick,
36 User: serviceNick,
37 Host: serviceNick,
38}
39
[150]40type serviceCommandSet map[string]*serviceCommand
41
[117]42type serviceCommand struct {
[150]43 usage string
44 desc string
45 handle func(dc *downstreamConn, params []string) error
46 children serviceCommandSet
[328]47 admin bool
[117]48}
49
[218]50func sendServiceNOTICE(dc *downstreamConn, text string) {
51 dc.SendMessage(&irc.Message{
[220]52 Prefix: servicePrefix,
[218]53 Command: "NOTICE",
54 Params: []string{dc.nick, text},
55 })
56}
57
[117]58func sendServicePRIVMSG(dc *downstreamConn, text string) {
59 dc.SendMessage(&irc.Message{
[220]60 Prefix: servicePrefix,
[117]61 Command: "PRIVMSG",
62 Params: []string{dc.nick, text},
63 })
64}
65
[566]66func splitWords(s string) ([]string, error) {
67 var words []string
68 var lastWord strings.Builder
69 escape := false
70 prev := ' '
71 wordDelim := ' '
72
73 for _, r := range s {
74 if escape {
75 // last char was a backslash, write the byte as-is.
76 lastWord.WriteRune(r)
77 escape = false
78 } else if r == '\\' {
79 escape = true
80 } else if wordDelim == ' ' && unicode.IsSpace(r) {
81 // end of last word
82 if !unicode.IsSpace(prev) {
83 words = append(words, lastWord.String())
84 lastWord.Reset()
85 }
86 } else if r == wordDelim {
87 // wordDelim is either " or ', switch back to
88 // space-delimited words.
89 wordDelim = ' '
90 } else if r == '"' || r == '\'' {
91 if wordDelim == ' ' {
92 // start of (double-)quoted word
93 wordDelim = r
94 } else {
95 // either wordDelim is " and r is ' or vice-versa
96 lastWord.WriteRune(r)
97 }
98 } else {
99 lastWord.WriteRune(r)
100 }
101
102 prev = r
103 }
104
105 if !unicode.IsSpace(prev) {
106 words = append(words, lastWord.String())
107 }
108
109 if wordDelim != ' ' {
110 return nil, fmt.Errorf("unterminated quoted string")
111 }
112 if escape {
113 return nil, fmt.Errorf("unterminated backslash sequence")
114 }
115
116 return words, nil
117}
118
[117]119func handleServicePRIVMSG(dc *downstreamConn, text string) {
[566]120 words, err := splitWords(text)
[117]121 if err != nil {
[566]122 sendServicePRIVMSG(dc, fmt.Sprintf(`error: failed to parse command: %v`, err))
[117]123 return
124 }
125
[150]126 cmd, params, err := serviceCommands.Get(words)
127 if err != nil {
128 sendServicePRIVMSG(dc, fmt.Sprintf(`error: %v (type "help" for a list of commands)`, err))
[117]129 return
130 }
[328]131 if cmd.admin && !dc.user.Admin {
132 sendServicePRIVMSG(dc, fmt.Sprintf(`error: you must be an admin to use this command`))
133 return
134 }
[117]135
[334]136 if cmd.handle == nil {
137 if len(cmd.children) > 0 {
138 var l []string
[335]139 appendServiceCommandSetHelp(cmd.children, words, dc.user.Admin, &l)
[334]140 sendServicePRIVMSG(dc, "available commands: "+strings.Join(l, ", "))
141 } else {
142 // Pretend the command does not exist if it has neither children nor handler.
143 // This is obviously a bug but it is better to not die anyway.
144 dc.logger.Printf("command without handler and subcommands invoked:", words[0])
145 sendServicePRIVMSG(dc, fmt.Sprintf("command %q not found", words[0]))
146 }
147 return
148 }
149
[117]150 if err := cmd.handle(dc, params); err != nil {
151 sendServicePRIVMSG(dc, fmt.Sprintf("error: %v", err))
152 }
153}
154
[150]155func (cmds serviceCommandSet) Get(params []string) (*serviceCommand, []string, error) {
156 if len(params) == 0 {
157 return nil, nil, fmt.Errorf("no command specified")
158 }
[117]159
[150]160 name := params[0]
161 params = params[1:]
162
163 cmd, ok := cmds[name]
164 if !ok {
165 for k := range cmds {
166 if !strings.HasPrefix(k, name) {
167 continue
168 }
169 if cmd != nil {
170 return nil, params, fmt.Errorf("command %q is ambiguous", name)
171 }
172 cmd = cmds[k]
173 }
174 }
175 if cmd == nil {
176 return nil, params, fmt.Errorf("command %q not found", name)
177 }
178
179 if len(params) == 0 || len(cmd.children) == 0 {
180 return cmd, params, nil
181 }
182 return cmd.children.Get(params)
183}
184
[339]185func (cmds serviceCommandSet) Names() []string {
186 l := make([]string, 0, len(cmds))
187 for name := range cmds {
188 l = append(l, name)
189 }
190 sort.Strings(l)
191 return l
192}
193
[150]194var serviceCommands serviceCommandSet
195
[117]196func init() {
[150]197 serviceCommands = serviceCommandSet{
[117]198 "help": {
199 usage: "[command]",
200 desc: "print help message",
201 handle: handleServiceHelp,
202 },
[150]203 "network": {
204 children: serviceCommandSet{
205 "create": {
[542]206 usage: "-addr <addr> [-name name] [-username username] [-pass pass] [-realname realname] [-nick nick] [-enabled enabled] [-connect-command command]...",
[150]207 desc: "add a new network",
[325]208 handle: handleServiceNetworkCreate,
[150]209 },
[151]210 "status": {
211 desc: "show a list of saved networks and their current status",
212 handle: handleServiceNetworkStatus,
213 },
[313]214 "update": {
[542]215 usage: "<name> [-addr addr] [-name name] [-username username] [-pass pass] [-realname realname] [-nick nick] [-enabled enabled] [-connect-command command]...",
[315]216 desc: "update a network",
[313]217 handle: handleServiceNetworkUpdate,
218 },
[202]219 "delete": {
220 usage: "<name>",
221 desc: "delete a network",
222 handle: handleServiceNetworkDelete,
223 },
[150]224 },
[120]225 },
[307]226 "certfp": {
227 children: serviceCommandSet{
228 "generate": {
229 usage: "[-key-type rsa|ecdsa|ed25519] [-bits N] <network name>",
230 desc: "generate a new self-signed certificate, defaults to using RSA-3072 key",
231 handle: handleServiceCertfpGenerate,
232 },
233 "fingerprint": {
234 usage: "<network name>",
235 desc: "show fingerprints of certificate associated with the network",
236 handle: handleServiceCertfpFingerprints,
237 },
238 },
239 },
[363]240 "sasl": {
241 children: serviceCommandSet{
242 "set-plain": {
243 usage: "<network name> <username> <password>",
244 desc: "set SASL PLAIN credentials",
245 handle: handleServiceSASLSetPlain,
246 },
[364]247 "reset": {
248 usage: "<network name>",
249 desc: "disable SASL authentication and remove stored credentials",
250 handle: handleServiceSASLReset,
251 },
[363]252 },
253 },
[329]254 "user": {
255 children: serviceCommandSet{
256 "create": {
[568]257 usage: "-username <username> -password <password> [-realname <realname>] [-admin]",
[329]258 desc: "create a new soju user",
259 handle: handleUserCreate,
260 admin: true,
261 },
[568]262 "update": {
[570]263 usage: "[-password <password>] [-realname <realname>]",
[568]264 desc: "update the current user",
265 handle: handleUserUpdate,
266 },
[379]267 "delete": {
268 usage: "<username>",
269 desc: "delete a user",
270 handle: handleUserDelete,
271 admin: true,
272 },
[329]273 },
274 },
[436]275 "channel": {
276 children: serviceCommandSet{
[539]277 "status": {
278 usage: "[-network name]",
279 desc: "show a list of saved channels and their current status",
280 handle: handleServiceChannelStatus,
281 },
[436]282 "update": {
283 usage: "<name> [-relay-detached <default|none|highlight|message>] [-reattach-on <default|none|highlight|message>] [-detach-after <duration>] [-detach-on <default|none|highlight|message>]",
284 desc: "update a channel",
285 handle: handleServiceChannelUpdate,
286 },
287 },
288 },
[117]289 }
290}
291
[328]292func appendServiceCommandSetHelp(cmds serviceCommandSet, prefix []string, admin bool, l *[]string) {
[339]293 for _, name := range cmds.Names() {
294 cmd := cmds[name]
[328]295 if cmd.admin && !admin {
296 continue
297 }
[150]298 words := append(prefix, name)
299 if len(cmd.children) == 0 {
300 s := strings.Join(words, " ")
301 *l = append(*l, s)
302 } else {
[328]303 appendServiceCommandSetHelp(cmd.children, words, admin, l)
[150]304 }
305 }
306}
307
[117]308func handleServiceHelp(dc *downstreamConn, params []string) error {
309 if len(params) > 0 {
[150]310 cmd, rest, err := serviceCommands.Get(params)
311 if err != nil {
312 return err
[117]313 }
[150]314 words := params[:len(params)-len(rest)]
[117]315
[150]316 if len(cmd.children) > 0 {
317 var l []string
[328]318 appendServiceCommandSetHelp(cmd.children, words, dc.user.Admin, &l)
[150]319 sendServicePRIVMSG(dc, "available commands: "+strings.Join(l, ", "))
320 } else {
321 text := strings.Join(words, " ")
322 if cmd.usage != "" {
323 text += " " + cmd.usage
324 }
325 text += ": " + cmd.desc
326
327 sendServicePRIVMSG(dc, text)
[117]328 }
329 } else {
330 var l []string
[328]331 appendServiceCommandSetHelp(serviceCommands, nil, dc.user.Admin, &l)
[117]332 sendServicePRIVMSG(dc, "available commands: "+strings.Join(l, ", "))
333 }
334 return nil
335}
[120]336
[202]337func newFlagSet() *flag.FlagSet {
[120]338 fs := flag.NewFlagSet("", flag.ContinueOnError)
339 fs.SetOutput(ioutil.Discard)
[202]340 return fs
341}
342
[313]343type stringSliceFlag []string
[263]344
[313]345func (v *stringSliceFlag) String() string {
[263]346 return fmt.Sprint([]string(*v))
347}
348
[313]349func (v *stringSliceFlag) Set(s string) error {
[263]350 *v = append(*v, s)
351 return nil
352}
353
[313]354// stringPtrFlag is a flag value populating a string pointer. This allows to
355// disambiguate between a flag that hasn't been set and a flag that has been
356// set to an empty string.
357type stringPtrFlag struct {
358 ptr **string
359}
360
361func (f stringPtrFlag) String() string {
[333]362 if f.ptr == nil || *f.ptr == nil {
[313]363 return ""
364 }
365 return **f.ptr
366}
367
368func (f stringPtrFlag) Set(s string) error {
369 *f.ptr = &s
370 return nil
371}
372
[542]373type boolPtrFlag struct {
374 ptr **bool
375}
376
377func (f boolPtrFlag) String() string {
378 if f.ptr == nil || *f.ptr == nil {
379 return "<nil>"
380 }
381 return strconv.FormatBool(**f.ptr)
382}
383
384func (f boolPtrFlag) Set(s string) error {
385 v, err := strconv.ParseBool(s)
386 if err != nil {
387 return err
388 }
389 *f.ptr = &v
390 return nil
391}
392
[313]393type networkFlagSet struct {
394 *flag.FlagSet
395 Addr, Name, Nick, Username, Pass, Realname *string
[542]396 Enabled *bool
[315]397 ConnectCommands []string
[313]398}
399
400func newNetworkFlagSet() *networkFlagSet {
401 fs := &networkFlagSet{FlagSet: newFlagSet()}
402 fs.Var(stringPtrFlag{&fs.Addr}, "addr", "")
403 fs.Var(stringPtrFlag{&fs.Name}, "name", "")
404 fs.Var(stringPtrFlag{&fs.Nick}, "nick", "")
405 fs.Var(stringPtrFlag{&fs.Username}, "username", "")
406 fs.Var(stringPtrFlag{&fs.Pass}, "pass", "")
407 fs.Var(stringPtrFlag{&fs.Realname}, "realname", "")
[542]408 fs.Var(boolPtrFlag{&fs.Enabled}, "enabled", "")
[313]409 fs.Var((*stringSliceFlag)(&fs.ConnectCommands), "connect-command", "")
410 return fs
411}
412
413func (fs *networkFlagSet) update(network *Network) error {
414 if fs.Addr != nil {
415 if addrParts := strings.SplitN(*fs.Addr, "://", 2); len(addrParts) == 2 {
416 scheme := addrParts[0]
417 switch scheme {
[358]418 case "ircs", "irc+insecure", "unix":
[313]419 default:
[358]420 return fmt.Errorf("unknown scheme %q (supported schemes: ircs, irc+insecure, unix)", scheme)
[313]421 }
422 }
423 network.Addr = *fs.Addr
424 }
425 if fs.Name != nil {
426 network.Name = *fs.Name
427 }
428 if fs.Nick != nil {
429 network.Nick = *fs.Nick
430 }
431 if fs.Username != nil {
432 network.Username = *fs.Username
433 }
434 if fs.Pass != nil {
435 network.Pass = *fs.Pass
436 }
437 if fs.Realname != nil {
438 network.Realname = *fs.Realname
439 }
[542]440 if fs.Enabled != nil {
441 network.Enabled = *fs.Enabled
442 }
[313]443 if fs.ConnectCommands != nil {
444 if len(fs.ConnectCommands) == 1 && fs.ConnectCommands[0] == "" {
445 network.ConnectCommands = nil
446 } else {
447 for _, command := range fs.ConnectCommands {
448 _, err := irc.ParseMessage(command)
449 if err != nil {
450 return fmt.Errorf("flag -connect-command must be a valid raw irc command string: %q: %v", command, err)
451 }
452 }
453 network.ConnectCommands = fs.ConnectCommands
454 }
455 }
456 return nil
457}
458
[325]459func handleServiceNetworkCreate(dc *downstreamConn, params []string) error {
[313]460 fs := newNetworkFlagSet()
[120]461 if err := fs.Parse(params); err != nil {
462 return err
463 }
[313]464 if fs.Addr == nil {
[150]465 return fmt.Errorf("flag -addr is required")
[120]466 }
467
[313]468 record := &Network{
[542]469 Addr: *fs.Addr,
470 Nick: dc.nick,
471 Enabled: true,
[269]472 }
[313]473 if err := fs.update(record); err != nil {
474 return err
[263]475 }
476
[313]477 network, err := dc.user.createNetwork(record)
[120]478 if err != nil {
479 return fmt.Errorf("could not create network: %v", err)
480 }
481
[202]482 sendServicePRIVMSG(dc, fmt.Sprintf("created network %q", network.GetName()))
[120]483 return nil
484}
[151]485
486func handleServiceNetworkStatus(dc *downstreamConn, params []string) error {
[546]487 n := 0
[151]488 dc.user.forEachNetwork(func(net *network) {
489 var statuses []string
490 var details string
[279]491 if uc := net.conn; uc != nil {
[271]492 if dc.nick != uc.nick {
493 statuses = append(statuses, "connected as "+uc.nick)
494 } else {
495 statuses = append(statuses, "connected")
496 }
[478]497 details = fmt.Sprintf("%v channels", uc.channels.Len())
[542]498 } else if !net.Enabled {
499 statuses = append(statuses, "disabled")
[151]500 } else {
501 statuses = append(statuses, "disconnected")
[219]502 if net.lastError != nil {
503 details = net.lastError.Error()
504 }
[151]505 }
506
507 if net == dc.network {
508 statuses = append(statuses, "current")
509 }
510
[224]511 name := net.GetName()
512 if name != net.Addr {
513 name = fmt.Sprintf("%v (%v)", name, net.Addr)
514 }
515
516 s := fmt.Sprintf("%v [%v]", name, strings.Join(statuses, ", "))
[151]517 if details != "" {
518 s += ": " + details
519 }
520 sendServicePRIVMSG(dc, s)
[546]521
522 n++
[151]523 })
[546]524
525 if n == 0 {
526 sendServicePRIVMSG(dc, `No network configured, add one with "network create".`)
527 }
528
[151]529 return nil
530}
[202]531
[313]532func handleServiceNetworkUpdate(dc *downstreamConn, params []string) error {
533 if len(params) < 1 {
[436]534 return fmt.Errorf("expected at least one argument")
[313]535 }
536
537 fs := newNetworkFlagSet()
538 if err := fs.Parse(params[1:]); err != nil {
539 return err
540 }
541
542 net := dc.user.getNetwork(params[0])
543 if net == nil {
544 return fmt.Errorf("unknown network %q", params[0])
545 }
546
547 record := net.Network // copy network record because we'll mutate it
548 if err := fs.update(&record); err != nil {
549 return err
550 }
551
552 network, err := dc.user.updateNetwork(&record)
553 if err != nil {
554 return fmt.Errorf("could not update network: %v", err)
555 }
556
557 sendServicePRIVMSG(dc, fmt.Sprintf("updated network %q", network.GetName()))
558 return nil
559}
560
[202]561func handleServiceNetworkDelete(dc *downstreamConn, params []string) error {
562 if len(params) != 1 {
563 return fmt.Errorf("expected exactly one argument")
564 }
565
566 net := dc.user.getNetwork(params[0])
567 if net == nil {
568 return fmt.Errorf("unknown network %q", params[0])
569 }
570
571 if err := dc.user.deleteNetwork(net.ID); err != nil {
572 return err
573 }
574
575 sendServicePRIVMSG(dc, fmt.Sprintf("deleted network %q", net.GetName()))
576 return nil
577}
[252]578
[325]579func handleServiceCertfpGenerate(dc *downstreamConn, params []string) error {
580 fs := newFlagSet()
581 keyType := fs.String("key-type", "rsa", "key type to generate (rsa, ecdsa, ed25519)")
582 bits := fs.Int("bits", 3072, "size of key to generate, meaningful only for RSA")
583
584 if err := fs.Parse(params); err != nil {
585 return err
586 }
587
588 if len(fs.Args()) != 1 {
589 return errors.New("exactly one argument is required")
590 }
591
592 net := dc.user.getNetwork(fs.Arg(0))
593 if net == nil {
594 return fmt.Errorf("unknown network %q", fs.Arg(0))
595 }
596
597 var (
598 privKey crypto.PrivateKey
599 pubKey crypto.PublicKey
600 )
601 switch *keyType {
602 case "rsa":
603 key, err := rsa.GenerateKey(rand.Reader, *bits)
604 if err != nil {
605 return err
606 }
607 privKey = key
608 pubKey = key.Public()
609 case "ecdsa":
610 key, err := ecdsa.GenerateKey(elliptic.P521(), rand.Reader)
611 if err != nil {
612 return err
613 }
614 privKey = key
615 pubKey = key.Public()
616 case "ed25519":
617 var err error
618 pubKey, privKey, err = ed25519.GenerateKey(rand.Reader)
619 if err != nil {
620 return err
621 }
622 }
623
624 // Using PKCS#8 allows easier extension for new key types.
625 privKeyBytes, err := x509.MarshalPKCS8PrivateKey(privKey)
626 if err != nil {
627 return err
628 }
629
630 notBefore := time.Now()
631 // Lets make a fair assumption nobody will use the same cert for more than 20 years...
632 notAfter := notBefore.Add(24 * time.Hour * 365 * 20)
633 serialNumberLimit := new(big.Int).Lsh(big.NewInt(1), 128)
634 serialNumber, err := rand.Int(rand.Reader, serialNumberLimit)
635 if err != nil {
636 return err
637 }
638 cert := &x509.Certificate{
639 SerialNumber: serialNumber,
640 Subject: pkix.Name{CommonName: "soju auto-generated certificate"},
641 NotBefore: notBefore,
642 NotAfter: notAfter,
643 KeyUsage: x509.KeyUsageKeyEncipherment | x509.KeyUsageDigitalSignature,
644 ExtKeyUsage: []x509.ExtKeyUsage{x509.ExtKeyUsageClientAuth},
645 }
646 derBytes, err := x509.CreateCertificate(rand.Reader, cert, cert, pubKey, privKey)
647 if err != nil {
648 return err
649 }
650
651 net.SASL.External.CertBlob = derBytes
652 net.SASL.External.PrivKeyBlob = privKeyBytes
653 net.SASL.Mechanism = "EXTERNAL"
654
[421]655 if err := dc.srv.db.StoreNetwork(dc.user.ID, &net.Network); err != nil {
[325]656 return err
657 }
658
659 sendServicePRIVMSG(dc, "certificate generated")
660
661 sha1Sum := sha1.Sum(derBytes)
662 sendServicePRIVMSG(dc, "SHA-1 fingerprint: "+hex.EncodeToString(sha1Sum[:]))
663 sha256Sum := sha256.Sum256(derBytes)
664 sendServicePRIVMSG(dc, "SHA-256 fingerprint: "+hex.EncodeToString(sha256Sum[:]))
665
666 return nil
667}
668
669func handleServiceCertfpFingerprints(dc *downstreamConn, params []string) error {
670 if len(params) != 1 {
671 return fmt.Errorf("expected exactly one argument")
672 }
673
674 net := dc.user.getNetwork(params[0])
675 if net == nil {
676 return fmt.Errorf("unknown network %q", params[0])
677 }
678
679 sha1Sum := sha1.Sum(net.SASL.External.CertBlob)
680 sendServicePRIVMSG(dc, "SHA-1 fingerprint: "+hex.EncodeToString(sha1Sum[:]))
681 sha256Sum := sha256.Sum256(net.SASL.External.CertBlob)
682 sendServicePRIVMSG(dc, "SHA-256 fingerprint: "+hex.EncodeToString(sha256Sum[:]))
683 return nil
684}
685
[364]686func handleServiceSASLSetPlain(dc *downstreamConn, params []string) error {
687 if len(params) != 3 {
688 return fmt.Errorf("expected exactly 3 arguments")
[325]689 }
690
691 net := dc.user.getNetwork(params[0])
692 if net == nil {
693 return fmt.Errorf("unknown network %q", params[0])
694 }
695
[364]696 net.SASL.Plain.Username = params[1]
697 net.SASL.Plain.Password = params[2]
698 net.SASL.Mechanism = "PLAIN"
[325]699
[421]700 if err := dc.srv.db.StoreNetwork(dc.user.ID, &net.Network); err != nil {
[325]701 return err
702 }
703
[364]704 sendServicePRIVMSG(dc, "credentials saved")
[325]705 return nil
706}
707
[364]708func handleServiceSASLReset(dc *downstreamConn, params []string) error {
709 if len(params) != 1 {
710 return fmt.Errorf("expected exactly one argument")
[363]711 }
712
713 net := dc.user.getNetwork(params[0])
714 if net == nil {
715 return fmt.Errorf("unknown network %q", params[0])
716 }
717
[364]718 net.SASL.Plain.Username = ""
719 net.SASL.Plain.Password = ""
720 net.SASL.External.CertBlob = nil
721 net.SASL.External.PrivKeyBlob = nil
722 net.SASL.Mechanism = ""
[363]723
[421]724 if err := dc.srv.db.StoreNetwork(dc.user.ID, &net.Network); err != nil {
[363]725 return err
726 }
727
[364]728 sendServicePRIVMSG(dc, "credentials reset")
[363]729 return nil
730}
731
[329]732func handleUserCreate(dc *downstreamConn, params []string) error {
733 fs := newFlagSet()
734 username := fs.String("username", "", "")
735 password := fs.String("password", "", "")
[568]736 realname := fs.String("realname", "", "")
[329]737 admin := fs.Bool("admin", false, "")
738
739 if err := fs.Parse(params); err != nil {
740 return err
741 }
742 if *username == "" {
743 return fmt.Errorf("flag -username is required")
744 }
745 if *password == "" {
746 return fmt.Errorf("flag -password is required")
747 }
748
749 hashed, err := bcrypt.GenerateFromPassword([]byte(*password), bcrypt.DefaultCost)
750 if err != nil {
751 return fmt.Errorf("failed to hash password: %v", err)
752 }
753
754 user := &User{
755 Username: *username,
756 Password: string(hashed),
[568]757 Realname: *realname,
[329]758 Admin: *admin,
759 }
760 if _, err := dc.srv.createUser(user); err != nil {
761 return fmt.Errorf("could not create user: %v", err)
762 }
763
764 sendServicePRIVMSG(dc, fmt.Sprintf("created user %q", *username))
765 return nil
766}
[379]767
[568]768func handleUserUpdate(dc *downstreamConn, params []string) error {
[570]769 var password, realname *string
[568]770 fs := newFlagSet()
[570]771 fs.Var(stringPtrFlag{&password}, "password", "")
[569]772 fs.Var(stringPtrFlag{&realname}, "realname", "")
[568]773
774 if err := fs.Parse(params); err != nil {
775 return err
776 }
777
[570]778 if password != nil {
779 hashed, err := bcrypt.GenerateFromPassword([]byte(*password), bcrypt.DefaultCost)
780 if err != nil {
781 return fmt.Errorf("failed to hash password: %v", err)
782 }
783 if err := dc.user.updatePassword(string(hashed)); err != nil {
784 return err
785 }
786 }
[569]787 if realname != nil {
788 if err := dc.user.updateRealname(*realname); err != nil {
789 return err
790 }
[568]791 }
792
793 sendServicePRIVMSG(dc, fmt.Sprintf("updated user %q", dc.user.Username))
794 return nil
795}
796
[379]797func handleUserDelete(dc *downstreamConn, params []string) error {
798 if len(params) != 1 {
799 return fmt.Errorf("expected exactly one argument")
800 }
801 username := params[0]
802
803 u := dc.srv.getUser(username)
804 if u == nil {
805 return fmt.Errorf("unknown username %q", username)
806 }
807
808 u.stop()
809
[508]810 if err := dc.srv.db.DeleteUser(u.ID); err != nil {
[379]811 return fmt.Errorf("failed to delete user: %v", err)
812 }
813
814 sendServicePRIVMSG(dc, fmt.Sprintf("deleted user %q", username))
815 return nil
816}
[436]817
[539]818func handleServiceChannelStatus(dc *downstreamConn, params []string) error {
819 var defaultNetworkName string
820 if dc.network != nil {
821 defaultNetworkName = dc.network.GetName()
822 }
823
824 fs := newFlagSet()
825 networkName := fs.String("network", defaultNetworkName, "")
826
827 if err := fs.Parse(params); err != nil {
828 return err
829 }
830
[546]831 n := 0
832
[539]833 sendNetwork := func(net *network) {
834 for _, entry := range net.channels.innerMap {
835 ch := entry.value.(*Channel)
836
837 var uch *upstreamChannel
838 if net.conn != nil {
839 uch = net.conn.channels.Value(ch.Name)
840 }
841
842 name := ch.Name
843 if *networkName == "" {
844 name += "/" + net.GetName()
845 }
846
847 var status string
848 if uch != nil {
849 status = "joined"
850 } else if net.conn != nil {
851 status = "parted"
852 } else {
853 status = "disconnected"
854 }
855
856 if ch.Detached {
857 status += ", detached"
858 }
859
860 s := fmt.Sprintf("%v [%v]", name, status)
861 sendServicePRIVMSG(dc, s)
[546]862
863 n++
[539]864 }
865 }
866
867 if *networkName == "" {
868 dc.user.forEachNetwork(sendNetwork)
869 } else {
870 net := dc.user.getNetwork(*networkName)
871 if net == nil {
872 return fmt.Errorf("unknown network %q", *networkName)
873 }
874 sendNetwork(net)
875 }
876
[546]877 if n == 0 {
878 sendServicePRIVMSG(dc, "No channel configured.")
879 }
880
[539]881 return nil
882}
883
[436]884type channelFlagSet struct {
885 *flag.FlagSet
886 RelayDetached, ReattachOn, DetachAfter, DetachOn *string
887}
888
889func newChannelFlagSet() *channelFlagSet {
890 fs := &channelFlagSet{FlagSet: newFlagSet()}
891 fs.Var(stringPtrFlag{&fs.RelayDetached}, "relay-detached", "")
892 fs.Var(stringPtrFlag{&fs.ReattachOn}, "reattach-on", "")
893 fs.Var(stringPtrFlag{&fs.DetachAfter}, "detach-after", "")
894 fs.Var(stringPtrFlag{&fs.DetachOn}, "detach-on", "")
895 return fs
896}
897
898func (fs *channelFlagSet) update(channel *Channel) error {
899 if fs.RelayDetached != nil {
900 filter, err := parseFilter(*fs.RelayDetached)
901 if err != nil {
902 return err
903 }
904 channel.RelayDetached = filter
905 }
906 if fs.ReattachOn != nil {
907 filter, err := parseFilter(*fs.ReattachOn)
908 if err != nil {
909 return err
910 }
911 channel.ReattachOn = filter
912 }
913 if fs.DetachAfter != nil {
914 dur, err := time.ParseDuration(*fs.DetachAfter)
915 if err != nil || dur < 0 {
916 return fmt.Errorf("unknown duration for -detach-after %q (duration format: 0, 300s, 22h30m, ...)", *fs.DetachAfter)
917 }
918 channel.DetachAfter = dur
919 }
920 if fs.DetachOn != nil {
921 filter, err := parseFilter(*fs.DetachOn)
922 if err != nil {
923 return err
924 }
925 channel.DetachOn = filter
926 }
927 return nil
928}
929
930func handleServiceChannelUpdate(dc *downstreamConn, params []string) error {
931 if len(params) < 1 {
932 return fmt.Errorf("expected at least one argument")
933 }
934 name := params[0]
935
936 fs := newChannelFlagSet()
937 if err := fs.Parse(params[1:]); err != nil {
938 return err
939 }
940
941 uc, upstreamName, err := dc.unmarshalEntity(name)
942 if err != nil {
943 return fmt.Errorf("unknown channel %q", name)
944 }
945
[478]946 ch := uc.network.channels.Value(upstreamName)
947 if ch == nil {
[436]948 return fmt.Errorf("unknown channel %q", name)
949 }
950
951 if err := fs.update(ch); err != nil {
952 return err
953 }
954
955 uc.updateChannelAutoDetach(upstreamName)
956
957 if err := dc.srv.db.StoreChannel(uc.network.ID, ch); err != nil {
958 return fmt.Errorf("failed to update channel: %v", err)
959 }
960
961 sendServicePRIVMSG(dc, fmt.Sprintf("updated channel %q", name))
962 return nil
963}
Note: See TracBrowser for help on using the repository browser.