-
Notifications
You must be signed in to change notification settings - Fork 2
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #42 from LucaBernstein/subcommands
Subcommands + Config + Tags + Reminders
- Loading branch information
Showing
19 changed files
with
692 additions
and
89 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,138 @@ | ||
| package bot | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "strconv" | ||
| "strings" | ||
|
|
||
| h "github.com/LucaBernstein/beancount-bot-tg/helpers" | ||
| tb "gopkg.in/tucnak/telebot.v2" | ||
| ) | ||
|
|
||
| func (bc *BotController) configHandler(m *tb.Message) { | ||
| sc := h.MakeSubcommandHandler("/"+CMD_CONFIG, true) | ||
| sc. | ||
| Add("currency", bc.configHandleCurrency). | ||
| Add("tag", bc.configHandleTag). | ||
| Add("notify", bc.configHandleNotification) | ||
| err := sc.Handle(m) | ||
| if err != nil { | ||
| bc.configHelp(m, nil) | ||
| } | ||
| } | ||
|
|
||
| func (bc *BotController) configHelp(m *tb.Message, err error) { | ||
| errorMsg := "" | ||
| if err != nil { | ||
| errorMsg += fmt.Sprintf("Error executing your command: %s\n\n", err.Error()) | ||
| } | ||
| bc.Bot.Send(m.Sender, errorMsg+fmt.Sprintf("Usage help for /%s:\n\n/%s currency <c> - Change default currency"+ | ||
| "\n\nTags will be added to each new transaction with a '#':\n"+ | ||
| "\n/%s tag - Get currently set tag"+ | ||
| "\n/%s tag off - Turn off tag"+ | ||
| "\n/%s tag <name> - Set tag to apply to new transactions, e.g. when on vacation"+ | ||
| "\n\nCreate a schedule to be notified of open transactions (i.e. not archived or deleted):\n"+ | ||
| "\n/%s notify - Get current notification status"+ | ||
| "\n/%s notify off - Disable reminder notifications"+ | ||
| "\n/%s notify <delay> <hour> - Notify of open transaction after <delay> days at <hour> of the day", | ||
| CMD_CONFIG, CMD_CONFIG, CMD_CONFIG, CMD_CONFIG, CMD_CONFIG, CMD_CONFIG, CMD_CONFIG, CMD_CONFIG)) | ||
| } | ||
|
|
||
| func (bc *BotController) configHandleCurrency(m *tb.Message, params ...string) { | ||
| currency := bc.Repo.UserGetCurrency(m) | ||
| if len(params) == 0 { // 0 params: GET currency | ||
| // Return currently set currency | ||
| bc.Bot.Send(m.Sender, fmt.Sprintf("Your current currency is set to '%s'. To change it add the new currency to use to the command like this: '/%s currency EUR'.", currency, CMD_CONFIG)) | ||
| return | ||
| } else if len(params) > 1 { // 2 or more params: too many | ||
| bc.configHelp(m, fmt.Errorf("invalid amount of parameters specified")) | ||
| return | ||
| } | ||
| // Set new currency | ||
| newCurrency := params[0] | ||
| err := bc.Repo.UserSetCurrency(m, newCurrency) | ||
| if err != nil { | ||
| bc.Bot.Send(m.Sender, "An error ocurred saving your currency preference: "+err.Error()) | ||
| return | ||
| } | ||
| bc.Bot.Send(m.Sender, fmt.Sprintf("Changed default currency for all future transactions from '%s' to '%s'.", currency, newCurrency)) | ||
| } | ||
|
|
||
| func (bc *BotController) configHandleTag(m *tb.Message, params ...string) { | ||
| if len(params) == 0 { | ||
| // GET tag | ||
| tag := bc.Repo.UserGetTag(m) | ||
| if tag != "" { | ||
| bc.Bot.Send(m.Sender, fmt.Sprintf("All new transactions automatically get the tag #%s added (vacation mode enabled)", tag)) | ||
| } else { | ||
| bc.Bot.Send(m.Sender, "No tags are currently added to new transactions (vacation mode disabled).") | ||
| } | ||
| return | ||
| } else if len(params) > 1 { // Only 0 or 1 allowed | ||
| bc.configHelp(m, fmt.Errorf("invalid amount of parameters specified")) | ||
| return | ||
| } | ||
| if params[0] == "off" { | ||
| // DELETE tag | ||
| bc.Repo.UserSetTag(m, "") | ||
| bc.Bot.Send(m.Sender, "Disabled automatically set tags on new transactions") | ||
| return | ||
| } | ||
| // SET tag | ||
| tag := strings.TrimPrefix(params[0], "#") | ||
| bc.Repo.UserSetTag(m, tag) | ||
| bc.Bot.Send(m.Sender, fmt.Sprintf("From now on all new transactions automatically get the tag #%s added (vacation mode enabled)", tag)) | ||
| } | ||
|
|
||
| func (bc *BotController) configHandleNotification(m *tb.Message, params ...string) { | ||
| if len(params) == 0 { | ||
| // GET schedule | ||
| daysDelay, hour, err := bc.Repo.UserGetNotificationSetting(m) | ||
| if err != nil { | ||
| bc.configHelp(m, fmt.Errorf("an application error occurred while retrieving user information from database")) | ||
| return | ||
| } | ||
| if daysDelay < 0 { | ||
| bc.Bot.Send(m.Sender, "Notifications are disabled for open transactions.") | ||
| return | ||
| } | ||
| plural_s := "s" | ||
| if daysDelay == 1 { | ||
| plural_s = "" | ||
| } | ||
| bc.Bot.Send(m.Sender, fmt.Sprintf("The bot will notify you daily at hour %d if transactions are open for more than %d day%s", hour, daysDelay, plural_s)) | ||
| return | ||
| } else if len(params) == 1 { | ||
| // DELETE schedule | ||
| if params[0] == "off" { | ||
| err := bc.Repo.UserSetNotificationSetting(m, -1, -1) | ||
| if err != nil { | ||
| bc.configHelp(m, fmt.Errorf("error setting notification schedule: %s", err.Error())) | ||
| return | ||
| } | ||
| bc.Bot.Send(m.Sender, "Successfully disabled notifications for open transactions.") | ||
| return | ||
| } | ||
| bc.configHelp(m, fmt.Errorf("invalid parameters")) | ||
| return | ||
| } else if len(params) == 2 { | ||
| // SET schedule | ||
| daysDelay, err := strconv.Atoi(params[0]) | ||
| if err != nil { | ||
| bc.configHelp(m, fmt.Errorf("error converting daysDelay to number: %s: %s", params[0], err.Error())) | ||
| return | ||
| } | ||
| hour, err := strconv.Atoi(params[1]) | ||
| if err != nil { | ||
| bc.configHelp(m, fmt.Errorf("error converting hour to number: %s: %s", params[1], err.Error())) | ||
| return | ||
| } | ||
| err = bc.Repo.UserSetNotificationSetting(m, daysDelay, hour) | ||
| if err != nil { | ||
| bc.configHelp(m, fmt.Errorf("error setting notification schedule: %s", err.Error())) | ||
| } | ||
| bc.configHandleNotification(m) // Recursively call with zero params --> GET | ||
| return | ||
| } | ||
| bc.configHelp(m, fmt.Errorf("invalid amount of parameters specified")) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,76 @@ | ||
| package bot | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "log" | ||
| "strings" | ||
| "testing" | ||
|
|
||
| "github.com/DATA-DOG/go-sqlmock" | ||
| tb "gopkg.in/tucnak/telebot.v2" | ||
| ) | ||
|
|
||
| func TestConfigCurrency(t *testing.T) { | ||
| // Test dependencies | ||
| chat := &tb.Chat{ID: 12345} | ||
| db, mock, err := sqlmock.New() | ||
| if err != nil { | ||
| log.Fatal(err) | ||
| } | ||
| mock. | ||
| ExpectQuery(`SELECT "currency" FROM "auth::user" WHERE "tgChatId" = ?`). | ||
| WithArgs(chat.ID). | ||
| WillReturnRows(sqlmock.NewRows([]string{"currency"})) | ||
| mock. | ||
| ExpectQuery(`SELECT "currency" FROM "auth::user" WHERE "tgChatId" = ?`). | ||
| WithArgs(chat.ID). | ||
| WillReturnRows(sqlmock.NewRows([]string{"currency"}).AddRow("SOMEEUR")) | ||
| mock. | ||
| ExpectQuery(`SELECT "currency" FROM "auth::user" WHERE "tgChatId" = ?`). | ||
| WithArgs(chat.ID). | ||
| WillReturnRows(sqlmock.NewRows([]string{"currency"}).AddRow("SOMEEUR")) | ||
| mock. | ||
| ExpectExec(`UPDATE "auth::user"`). | ||
| WithArgs(12345, "USD"). | ||
| WillReturnResult(sqlmock.NewResult(1, 1)) | ||
|
|
||
| bc := NewBotController(db) | ||
|
|
||
| bot := &MockBot{} | ||
| bc.ConfigureAndAttachBot(bot) | ||
|
|
||
| bc.commandConfig(&tb.Message{Text: "/config", Chat: chat}) | ||
| if !strings.Contains(fmt.Sprintf("%v", bot.LastSentWhat), "Usage help for /config") { | ||
| t.Errorf("/config: %s", bot.LastSentWhat) | ||
| } | ||
|
|
||
| // Default currency | ||
| bc.commandConfig(&tb.Message{Text: "/config currency", Chat: chat}) | ||
| if strings.Contains(fmt.Sprintf("%v", bot.LastSentWhat), "Usage help") { | ||
| t.Errorf("/config currency: Unexpected usage help: %s", bot.LastSentWhat) | ||
| } | ||
| if !strings.Contains(fmt.Sprintf("%v", bot.LastSentWhat), "Your current currency is set to 'EUR'") { | ||
| t.Errorf("/config currency default: Expected currency to be retrieved from db: %s", bot.LastSentWhat) | ||
| } | ||
|
|
||
| // Currency set in db | ||
| bc.commandConfig(&tb.Message{Text: "/config currency", Chat: chat}) | ||
| if strings.Contains(fmt.Sprintf("%v", bot.LastSentWhat), "Usage help") { | ||
| t.Errorf("/config currency: Unexpected usage help: %s", bot.LastSentWhat) | ||
| } | ||
| if !strings.Contains(fmt.Sprintf("%v", bot.LastSentWhat), "Your current currency is set to 'SOMEEUR") { | ||
| t.Errorf("/config currency set (2): Expected currency to be retrieved from db: %s", bot.LastSentWhat) | ||
| } | ||
|
|
||
| bc.commandConfig(&tb.Message{Text: "/config currency USD", Chat: chat}) | ||
| if strings.Contains(fmt.Sprintf("%v", bot.LastSentWhat), "Usage help") { | ||
| t.Errorf("/config currency USD: Unexpected usage help: %s", bot.LastSentWhat) | ||
| } | ||
| if !strings.Contains(fmt.Sprintf("%v", bot.LastSentWhat), "'SOMEEUR' to 'USD'") { | ||
| t.Errorf("/config currency (2): Expected currency to be retrieved from db %s", bot.LastSentWhat) | ||
| } | ||
|
|
||
| if err := mock.ExpectationsWereMet(); err != nil { | ||
| t.Errorf("there were unfulfilled expectations: %s", err) | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.