124 lines
3.4 KiB
Go
124 lines
3.4 KiB
Go
package controller
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
"ticket-pimp/internal/domain"
|
|
"ticket-pimp/internal/storage/db"
|
|
"time"
|
|
|
|
"github.com/bwmarrin/discordgo"
|
|
"github.com/jackc/pgx/v5/pgtype"
|
|
)
|
|
|
|
func (wc *WorkflowController) InitTask(t *domain.Task) (*domain.Task, error) {
|
|
|
|
// Записываем в базу созданную задачу ------------------------------------------
|
|
dbtask, err := wc.q.InsertTask(context.TODO(), db.InsertTaskParams{
|
|
Creator: pgtype.Text{String: t.Creator, Valid: true},
|
|
CreatorLink: pgtype.Text{
|
|
String: t.CreatorLink,
|
|
Valid: true,
|
|
},
|
|
Description: pgtype.Text{
|
|
String: t.Description,
|
|
Valid: true,
|
|
},
|
|
})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("unable to create task at the db: %v", err)
|
|
}
|
|
// ------------------------------------------------------------------------------------
|
|
|
|
task := newConvertable(&dbtask).ExtractDomain()
|
|
|
|
// Инициализируем новый клиент дискорда
|
|
// [ ] Нездоровое получение параметров клиента из os..
|
|
var (
|
|
token = os.Getenv("DISCORD_TOKEN")
|
|
forumChannelID = os.Getenv("TASKS_CHANNEL")
|
|
)
|
|
|
|
s, err := discordgo.New("Bot " + token)
|
|
if err != nil {
|
|
return task, fmt.Errorf("unable to create discord session: %v", err)
|
|
// [ ] Что делать, если не получилось создать задачу?
|
|
}
|
|
|
|
if err := s.Open(); err != nil {
|
|
return task, fmt.Errorf("cannot open the session: %v", err)
|
|
// [ ] Что делать, если не получилось создать задачу?
|
|
}
|
|
|
|
msg := discordgo.MessageSend{
|
|
Content: task.NotStartedMessage(),
|
|
Components: []discordgo.MessageComponent{
|
|
discordgo.ActionsRow{
|
|
Components: []discordgo.MessageComponent{
|
|
discordgo.Button{
|
|
Label: "Start",
|
|
Style: discordgo.SuccessButton,
|
|
Disabled: false,
|
|
CustomID: "task_start",
|
|
},
|
|
discordgo.Button{
|
|
Label: "Close",
|
|
Style: discordgo.DangerButton,
|
|
Disabled: true,
|
|
CustomID: "task_close",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
th, err := s.ForumThreadStartComplex(
|
|
forumChannelID,
|
|
&discordgo.ThreadStart{
|
|
Name: "Task ID:" + strconv.Itoa(int(task.ID)),
|
|
},
|
|
&msg,
|
|
)
|
|
if err != nil {
|
|
return task, fmt.Errorf("unable to update channel: %v", err)
|
|
}
|
|
|
|
err = wc.q.UpdateTaskWithMessageID(context.TODO(), db.UpdateTaskWithMessageIDParams{
|
|
Messageid: pgtype.Text{String: th.ID, Valid: true},
|
|
ID: dbtask.ID,
|
|
})
|
|
if err != nil {
|
|
return task, fmt.Errorf("unable to set discord message to task: %v", err)
|
|
}
|
|
|
|
return task, nil
|
|
}
|
|
|
|
func (wc *WorkflowController) UpdateTask(id string, opt int, user string) (*TaskConvertable, error) {
|
|
var (
|
|
err error
|
|
dbtask db.Task
|
|
)
|
|
switch opt {
|
|
case 0:
|
|
dbtask, err = wc.q.StartTask(context.TODO(), db.StartTaskParams{
|
|
UpdatedAt: pgtype.Timestamptz{Time: time.Now(), InfinityModifier: 0, Valid: true},
|
|
Assignee: pgtype.Text{String: user, Valid: true},
|
|
Messageid: pgtype.Text{String: id, Valid: true},
|
|
})
|
|
return &TaskConvertable{&dbtask}, err
|
|
|
|
case 1:
|
|
dbtask, err = wc.q.CloseTask(context.TODO(), db.CloseTaskParams{
|
|
DeletedAt: pgtype.Timestamptz{Time: time.Now(), InfinityModifier: 0, Valid: true},
|
|
Assignee: pgtype.Text{String: user, Valid: true},
|
|
Messageid: pgtype.Text{String: id, Valid: true},
|
|
})
|
|
return &TaskConvertable{&dbtask}, err
|
|
}
|
|
|
|
return &TaskConvertable{&dbtask}, nil
|
|
}
|