config.go 2.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104
  1. package main
  2. import (
  3. "encoding/json"
  4. "os"
  5. "github.com/rivo/tview"
  6. )
  7. var conf *config
  8. type keybindings struct {
  9. GuildsTreeViewFocus string
  10. MessagesTextViewFocus string
  11. MessagesTextViewSelectPrevious string
  12. MessagesTextViewSelectNext string
  13. MessagesTextViewSelectFirst string
  14. MessagesTextViewSelectLast string
  15. MessagesTextViewReplySelected string
  16. MessagesTextViewMentionReplySelected string
  17. MessageInputFieldFocus string
  18. }
  19. type config struct {
  20. Token string
  21. Mouse bool
  22. Notifications bool
  23. UserAgent string
  24. GetMessagesLimit int
  25. Theme tview.Theme
  26. Keybindings keybindings
  27. }
  28. func loadConfig() *config {
  29. u, err := os.UserHomeDir()
  30. if err != nil {
  31. panic(err)
  32. }
  33. configPath := u + "/.config/discordo/config.json"
  34. if _, err = os.Stat(configPath); os.IsNotExist(err) {
  35. err = os.MkdirAll(u+"/.config/discordo", 0700)
  36. if err != nil {
  37. panic(err)
  38. }
  39. f, err := os.Create(configPath)
  40. if err != nil {
  41. panic(err)
  42. }
  43. defer f.Close()
  44. c := config{
  45. Mouse: true,
  46. Notifications: true,
  47. UserAgent: "" +
  48. "Mozilla/5.0 (X11; Linux x86_64) " +
  49. "AppleWebKit/537.36 (KHTML, like Gecko) " +
  50. "Chrome/92.0.4515.131 Safari/537.36",
  51. GetMessagesLimit: 50,
  52. Theme: tview.Styles,
  53. Keybindings: keybindings{
  54. GuildsTreeViewFocus: "Alt+Rune[1]",
  55. MessagesTextViewFocus: "Alt+Rune[2]",
  56. MessagesTextViewSelectPrevious: "Up",
  57. MessagesTextViewSelectNext: "Down",
  58. MessagesTextViewSelectFirst: "Home",
  59. MessagesTextViewSelectLast: "End",
  60. MessagesTextViewReplySelected: "Rune[r]",
  61. MessagesTextViewMentionReplySelected: "Rune[R]",
  62. MessageInputFieldFocus: "Alt+Rune[3]",
  63. },
  64. }
  65. d, err := json.MarshalIndent(c, "", "\t")
  66. if err != nil {
  67. panic(err)
  68. }
  69. _, err = f.Write(d)
  70. if err != nil {
  71. panic(err)
  72. }
  73. f.Sync()
  74. return &c
  75. }
  76. d, err := os.ReadFile(configPath)
  77. if err != nil {
  78. panic(err)
  79. }
  80. var c config
  81. if err = json.Unmarshal(d, &c); err != nil {
  82. panic(err)
  83. }
  84. return &c
  85. }