init commit
This commit is contained in:
naudachu
commit
03ac788f32
|
|
@ -0,0 +1,104 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
"os/signal"
|
||||||
|
"strings"
|
||||||
|
"syscall"
|
||||||
|
|
||||||
|
"ticket-creator/domain"
|
||||||
|
|
||||||
|
"github.com/joho/godotenv"
|
||||||
|
"github.com/mr-linch/go-tg"
|
||||||
|
"github.com/mr-linch/go-tg/tgb"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
env("../.env")
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
ctx, cancel := signal.NotifyContext(ctx, os.Interrupt, os.Kill, syscall.SIGTERM)
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
if err := run(ctx); err != nil {
|
||||||
|
fmt.Println(err)
|
||||||
|
defer os.Exit(1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func env(envFilePath string) {
|
||||||
|
err := godotenv.Load(envFilePath)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal("Error loading .env file")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func run(ctx context.Context) error {
|
||||||
|
client := tg.New(os.Getenv("TG_API"))
|
||||||
|
|
||||||
|
router := tgb.NewRouter().
|
||||||
|
Message(func(ctx context.Context, mu *tgb.MessageUpdate) error {
|
||||||
|
|
||||||
|
str := strings.Replace(mu.Text, "/new", "", 1)
|
||||||
|
if str == "" {
|
||||||
|
return errors.New("empty command provided")
|
||||||
|
}
|
||||||
|
issueKeyStr := workflow(str)
|
||||||
|
|
||||||
|
return mu.Answer(tg.HTML.Text(
|
||||||
|
tg.HTML.Line(
|
||||||
|
"🤘 Ticket ",
|
||||||
|
tg.HTML.Link(issueKeyStr, fmt.Sprintf("https://marlerino.youtrack.cloud/issue/%s", issueKeyStr)),
|
||||||
|
"has been created!",
|
||||||
|
),
|
||||||
|
)).ParseMode(tg.HTML).DoVoid(ctx)
|
||||||
|
}, tgb.TextHasPrefix("/new"))
|
||||||
|
|
||||||
|
return tgb.NewPoller(
|
||||||
|
router,
|
||||||
|
client,
|
||||||
|
).Run(ctx)
|
||||||
|
}
|
||||||
|
|
||||||
|
func workflow(name string) string {
|
||||||
|
yt := domain.NewYT(os.Getenv("YT_URL"), os.Getenv("YT_TOKEN"))
|
||||||
|
projects := yt.GetProjects()
|
||||||
|
issue := yt.CreateIssue(projects[0].ID, name)
|
||||||
|
if issue != nil {
|
||||||
|
git := createRepo(issue.Key, 0)
|
||||||
|
gitBuild := createRepo(issue.Key+"-build", 1)
|
||||||
|
folder := createFolder(issue.Key + " - " + issue.Summary)
|
||||||
|
updated := yt.UpdateIssue(issue, folder, git, gitBuild)
|
||||||
|
log.Print(updated)
|
||||||
|
}
|
||||||
|
return issue.Key
|
||||||
|
}
|
||||||
|
|
||||||
|
func createRepo(name string, param uint) string {
|
||||||
|
gb := domain.NewGitBucket(os.Getenv("GIT_BASE_URL"), os.Getenv("GIT_TOKEN"))
|
||||||
|
repo, _ := gb.NewRepo(name)
|
||||||
|
if repo != nil {
|
||||||
|
switch param {
|
||||||
|
case 0:
|
||||||
|
return repo.HtmlUrl
|
||||||
|
case 1:
|
||||||
|
return fmt.Sprintf("ssh://%s/%s.git", repo.SshUrl, repo.FullName)
|
||||||
|
default:
|
||||||
|
return repo.CloneUrl
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return "no-repo"
|
||||||
|
}
|
||||||
|
|
||||||
|
func createFolder(name string) string {
|
||||||
|
oc := domain.NewCloud(os.Getenv("CLOUD_BASE_URL"), os.Getenv("CLOUD_USER"), os.Getenv("CLOUD_PASS"))
|
||||||
|
cloud, _ := oc.CreateFolder(name)
|
||||||
|
if cloud != nil {
|
||||||
|
return cloud.FolderPath
|
||||||
|
}
|
||||||
|
return "no-folder"
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,56 @@
|
||||||
|
package domain
|
||||||
|
|
||||||
|
import (
|
||||||
|
"log"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/imroc/req/v3"
|
||||||
|
)
|
||||||
|
|
||||||
|
type cloud struct {
|
||||||
|
baseUrl string
|
||||||
|
client *req.Client
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewCloud(base, user, pass string) *cloud {
|
||||||
|
|
||||||
|
client := req.C().
|
||||||
|
SetTimeout(5*time.Second).
|
||||||
|
SetCommonBasicAuth(user, pass).
|
||||||
|
SetBaseURL(base)
|
||||||
|
return &cloud{
|
||||||
|
baseUrl: base,
|
||||||
|
client: client,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type Cloud struct {
|
||||||
|
FolderName string
|
||||||
|
FolderPath string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *cloud) CreateFolder(name string) (*Cloud, error) {
|
||||||
|
const (
|
||||||
|
HOMEPATH = "/remote.php/dav/files/naudachu/%23mobiledev/"
|
||||||
|
PATH = "/apps/files/?dir=/%23mobiledev/"
|
||||||
|
)
|
||||||
|
|
||||||
|
cloud := Cloud{
|
||||||
|
FolderName: name,
|
||||||
|
FolderPath: "",
|
||||||
|
}
|
||||||
|
|
||||||
|
resp, err := c.client.R().
|
||||||
|
Send("MKCOL", HOMEPATH+name)
|
||||||
|
|
||||||
|
// Check if request failed or response status is not Ok;
|
||||||
|
if !resp.IsSuccessState() || err != nil {
|
||||||
|
log.Print("bad status:", resp.Status)
|
||||||
|
log.Print(resp.Dump())
|
||||||
|
}
|
||||||
|
|
||||||
|
if resp.StatusCode == 201 {
|
||||||
|
cloud.FolderPath = c.baseUrl + PATH + name
|
||||||
|
}
|
||||||
|
return &cloud, err
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,67 @@
|
||||||
|
package domain
|
||||||
|
|
||||||
|
import (
|
||||||
|
"log"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/imroc/req/v3"
|
||||||
|
)
|
||||||
|
|
||||||
|
type gitbucket struct {
|
||||||
|
client *req.Client
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewGitBucket(base, token string) *gitbucket {
|
||||||
|
headers := map[string]string{
|
||||||
|
"Accept": "application/vnd.github+json",
|
||||||
|
"Authorization": "Token " + token,
|
||||||
|
"X-GitHub-Api-Version": "2022-11-28",
|
||||||
|
"Content-Type": "application/json",
|
||||||
|
}
|
||||||
|
|
||||||
|
client := req.C().
|
||||||
|
SetTimeout(5 * time.Second).
|
||||||
|
SetCommonHeaders(headers).
|
||||||
|
SetBaseURL(base)
|
||||||
|
return &gitbucket{
|
||||||
|
client: client,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type Repo struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
FullName string `json:"full_name"`
|
||||||
|
Private bool `json:"private"`
|
||||||
|
Url string `json:"url"`
|
||||||
|
CloneUrl string `json:"clone_url"`
|
||||||
|
HtmlUrl string `json:"Html_url"`
|
||||||
|
SshUrl string `json:"ssh_url"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (gb *gitbucket) NewRepo(name string) (*Repo, error) {
|
||||||
|
|
||||||
|
type request struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
Private bool `json:"private"`
|
||||||
|
}
|
||||||
|
|
||||||
|
payload := request{
|
||||||
|
Name: name,
|
||||||
|
Private: false,
|
||||||
|
}
|
||||||
|
|
||||||
|
var git Repo
|
||||||
|
|
||||||
|
resp, err := gb.client.R().
|
||||||
|
SetBody(&payload).
|
||||||
|
SetSuccessResult(&git).
|
||||||
|
Post("/user/repos")
|
||||||
|
|
||||||
|
// Check if request failed or response status is not Ok;
|
||||||
|
if !resp.IsSuccessState() || err != nil {
|
||||||
|
log.Print("bad status:", resp.Status)
|
||||||
|
log.Print(resp.Dump())
|
||||||
|
}
|
||||||
|
|
||||||
|
return &git, err
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,147 @@
|
||||||
|
package domain
|
||||||
|
|
||||||
|
import (
|
||||||
|
"log"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/imroc/req/v3"
|
||||||
|
)
|
||||||
|
|
||||||
|
type youtrack struct {
|
||||||
|
client *req.Client
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewYT(base, token string) *youtrack {
|
||||||
|
headers := map[string]string{
|
||||||
|
"Accept": "application/json",
|
||||||
|
"Content-Type": "application/json",
|
||||||
|
}
|
||||||
|
|
||||||
|
client := req.C().
|
||||||
|
SetTimeout(15 * time.Second).
|
||||||
|
SetCommonHeaders(headers).
|
||||||
|
SetBaseURL(base).
|
||||||
|
SetCommonBearerAuthToken(token)
|
||||||
|
|
||||||
|
return &youtrack{
|
||||||
|
client: client,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type Project struct {
|
||||||
|
ID string `json:"id"`
|
||||||
|
ShortName string `json:"shortName"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetProjects
|
||||||
|
// provides an array of existing projects;
|
||||||
|
func (yt *youtrack) GetProjects() []Project {
|
||||||
|
|
||||||
|
var projects []Project
|
||||||
|
|
||||||
|
resp, err := yt.client.R().
|
||||||
|
EnableDump().
|
||||||
|
SetQueryParam("fields", "id,name,shortName").
|
||||||
|
SetSuccessResult(&projects).
|
||||||
|
Get("/admin/projects")
|
||||||
|
|
||||||
|
if !resp.IsSuccessState() || err != nil {
|
||||||
|
log.Print("bad status:", resp.Status)
|
||||||
|
log.Print(resp.Dump())
|
||||||
|
}
|
||||||
|
return projects
|
||||||
|
}
|
||||||
|
|
||||||
|
type ProjectID struct {
|
||||||
|
ID string `json:"id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type IssueCreateRequest struct {
|
||||||
|
ProjectID ProjectID `json:"project"`
|
||||||
|
Key string `json:"idReadable"`
|
||||||
|
ID string `json:"id"`
|
||||||
|
Summary string `json:"summary"`
|
||||||
|
Description string `json:"description"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// CreateIssue
|
||||||
|
// example: newIssue := yt.CreateIssue("0-2", "Summary", "Description");
|
||||||
|
func (yt *youtrack) CreateIssue(projectID, name string) *IssueCreateRequest {
|
||||||
|
|
||||||
|
// Create an issue with the provided:, Project ID, Name, Description;
|
||||||
|
issue := IssueCreateRequest{
|
||||||
|
ProjectID: ProjectID{
|
||||||
|
ID: projectID, //"id":"0-2"
|
||||||
|
},
|
||||||
|
Summary: name,
|
||||||
|
//Description: description,
|
||||||
|
}
|
||||||
|
|
||||||
|
// Push issue to the YT;
|
||||||
|
resp, err := yt.client.R().
|
||||||
|
SetQueryParam("fields", "idReadable,id").
|
||||||
|
SetBody(&issue).
|
||||||
|
SetSuccessResult(&issue).
|
||||||
|
Post("/issues")
|
||||||
|
|
||||||
|
// Check if request failed or response status is not Ok;
|
||||||
|
if !resp.IsSuccessState() || err != nil {
|
||||||
|
log.Print("bad status:", resp.Status)
|
||||||
|
log.Print(resp.Dump())
|
||||||
|
}
|
||||||
|
|
||||||
|
return &issue
|
||||||
|
}
|
||||||
|
|
||||||
|
type IssueUpdateRequest struct {
|
||||||
|
IssueCreateRequest
|
||||||
|
CustomFields []CustomField `json:"customFields"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type CustomFields struct {
|
||||||
|
List []CustomField `json:"customFields"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type CustomField struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
Type string `json:"$type"`
|
||||||
|
Value string `json:"value"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (yt *youtrack) UpdateIssue(issue *IssueCreateRequest, folder, git, gitBuild string) *IssueUpdateRequest {
|
||||||
|
// Set Folder, Git, GitBuild to the Issue:
|
||||||
|
update := IssueUpdateRequest{
|
||||||
|
IssueCreateRequest: *issue,
|
||||||
|
CustomFields: []CustomField{
|
||||||
|
{
|
||||||
|
Name: "Директория графики",
|
||||||
|
Type: "SimpleIssueCustomField",
|
||||||
|
Value: folder,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "Репо проекта",
|
||||||
|
Type: "SimpleIssueCustomField",
|
||||||
|
Value: git,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "Репо iOS сборки",
|
||||||
|
Type: "SimpleIssueCustomField",
|
||||||
|
Value: gitBuild,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
// Push issue update to YT
|
||||||
|
resp, err := yt.client.R().
|
||||||
|
SetBody(&update).
|
||||||
|
SetSuccessResult(&issue).
|
||||||
|
Post("/issues/" + issue.Key)
|
||||||
|
|
||||||
|
if !resp.IsSuccessState() || err != nil {
|
||||||
|
log.Print("bad status:", resp.Status)
|
||||||
|
log.Print(resp.Dump())
|
||||||
|
}
|
||||||
|
|
||||||
|
return &update
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,14 @@
|
||||||
|
# Основное:
|
||||||
|
- [ ] Сохранять правильную ссылку на Git;
|
||||||
|
- [ ] Сохранять правильную ссылку на GitBuild;
|
||||||
|
- [ ] Делать запросы в Git, ownCloud параллельно;
|
||||||
|
- [x] Сделать бота в Telegram;
|
||||||
|
|
||||||
|
|
||||||
|
# Под звёздочкой:
|
||||||
|
- [ ] Сохранять короткую ссылку на графику;
|
||||||
|
- [ ] Сохранять внешнюю ссылку на графику;
|
||||||
|
- [ ] Сделать бота в Discord;
|
||||||
|
- [ ] Подумать над нормальной обработкой ошибок, сейчас достаточно всрато;
|
||||||
|
- [ ] Складывать в описание репозитория ссылку на тикет;
|
||||||
|
- [ ] Сделать базулю с достойными пользователями;
|
||||||
Loading…
Reference in New Issue