services.go 2.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273
  1. package services
  2. import (
  3. "flag"
  4. "fmt"
  5. "io"
  6. "log"
  7. "github.com/fgm/izidic"
  8. )
  9. const (
  10. // Flags
  11. PAddr = "addr"
  12. PProfile = "profile"
  13. PQName = "queue-name"
  14. PRegion = "region"
  15. PCSRFSecret = "csrf-secret"
  16. PStoreSecret = "store-secret"
  17. PURL = "url"
  18. PWait = "wait"
  19. // Non-flags
  20. PArgs = "args"
  21. PHandler = "handler"
  22. PName = "name"
  23. PWriter = "writer"
  24. // Services
  25. SvcClient = "sqs"
  26. SvcConsumer = "consumeMessage"
  27. SvcFlags = "flags"
  28. SvcHttp = "http"
  29. SvcLister = "lister"
  30. SvcLogger = "logger"
  31. SvcProducer = "producer"
  32. SvcReceiver = "receiver"
  33. SvcRedriver = "redriver"
  34. SvcRenderer = "renderer"
  35. )
  36. func FlagsService(dic *izidic.Container) (any, error) {
  37. fs := flag.NewFlagSet(dic.MustParam(PName).(string), flag.ContinueOnError)
  38. addr := fs.String(PAddr, ":8080", "The IP address on which to listen")
  39. profile := fs.String(PProfile, "test-profile", "The AWS profile")
  40. region := fs.String(PRegion, "eu-west-3", "The AWS region to connect to")
  41. qName := fs.String(PQName, "dummy-queue", "The queue name")
  42. csrfSecret := fs.String(PCSRFSecret, "csrfSecret", "The CSRF secret")
  43. storeSecret := fs.String(PStoreSecret, "storeSecret", "The session store secret")
  44. sqsURL := fs.String(PURL, "http://localhost:4566", "The SQS endpoint URL")
  45. wait := fs.Int(PWait, 3, "The maximum number of seconds to wait when receiving messages")
  46. if err := fs.Parse(dic.MustParam(PArgs).([]string)); err != nil {
  47. return nil, fmt.Errorf("cannot obtain CLI args")
  48. }
  49. dic.Store(PAddr, *addr)
  50. dic.Store(PProfile, *profile)
  51. dic.Store(PQName, *qName)
  52. dic.Store(PRegion, *region)
  53. dic.Store(PCSRFSecret, []byte(*csrfSecret))
  54. dic.Store(PStoreSecret, []byte(*storeSecret))
  55. dic.Store(PURL, *sqsURL)
  56. dic.Store(PWait, *wait)
  57. return fs, nil
  58. }
  59. // LoggerService is an izidic.Service also containing a one-time initialization action.
  60. func LoggerService(dic *izidic.Container) (any, error) {
  61. w := dic.MustParam(PWriter).(io.Writer)
  62. log.SetOutput(w) // Support dependency code not taking an injected logger.
  63. logger := log.New(w, "", log.LstdFlags)
  64. return logger, nil
  65. }