299 lines
6.9 KiB
Go
299 lines
6.9 KiB
Go
package handlers
|
|
|
|
import (
|
|
"amocrm_templategen_back/amo"
|
|
"amocrm_templategen_back/dal/model"
|
|
"amocrm_templategen_back/templategen"
|
|
"errors"
|
|
"fmt"
|
|
"github.com/gorilla/schema"
|
|
docTemp "github.com/opencontrol/doc-template"
|
|
"golang.org/x/oauth2"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/url"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
type RespAmoSaveToken struct {
|
|
AccessToken string `json:"access_token" schema:"access_token"`
|
|
Code string `json:"code" schema:"code"`
|
|
ClientID string `json:"client_id" schema:"client_id"`
|
|
ExpiresIn int64 `json:"expires_in" schema:"expires_in"`
|
|
TokenType string `json:"token_type" schema:"token_type"`
|
|
RefreshToken string `json:"refresh_token" schema:"refresh_token"`
|
|
State string `json:"state" schema:"state"`
|
|
FromWidget string `json:"from_widget" schema:"from_widget"`
|
|
Referer string `json:"referer" schema:"referer"`
|
|
}
|
|
|
|
func (h *Handlers) AmoSaveToken(w http.ResponseWriter, r *http.Request) {
|
|
var resp RespAmoSaveToken
|
|
|
|
err := r.ParseForm()
|
|
if err != nil {
|
|
h.reportError(w, err, http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
err = schema.NewDecoder().Decode(&resp, r.Form)
|
|
if err != nil {
|
|
h.reportError(w, err, 500)
|
|
return
|
|
}
|
|
|
|
if resp.AccessToken == "" && resp.Code == "" {
|
|
err = errors.New("AmoErr: got empty token")
|
|
h.reportError(w, err, http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
token := &oauth2.Token{
|
|
AccessToken: resp.AccessToken,
|
|
RefreshToken: resp.RefreshToken,
|
|
TokenType: resp.TokenType,
|
|
Expiry: time.Now().Add(time.Duration(resp.ExpiresIn) * time.Second),
|
|
}
|
|
|
|
var amoClient *amo.Client
|
|
if resp.Code != "" {
|
|
amoClient = h.Amo.NewClient(resp.Referer, resp.Code)
|
|
token, err = amoClient.AuthCode()
|
|
if err != nil {
|
|
h.reportError(w, err, 500)
|
|
return
|
|
}
|
|
}
|
|
|
|
user := getJwtUser(r)
|
|
if user == nil {
|
|
h.reportError(w, ErrorUnauthorized, http.StatusUnauthorized)
|
|
return
|
|
}
|
|
|
|
amoAcc, err := amoClient.GetAccount()
|
|
|
|
if err != nil {
|
|
h.reportError(w, err, 200)
|
|
}
|
|
if amoAcc == nil {
|
|
return
|
|
} else {
|
|
fmt.Println(amoAcc)
|
|
}
|
|
|
|
// Insert/Update token in DB
|
|
_, err = h.dal.Amo.InsertOrUpdate(r.Context(), &model.Amo{
|
|
UserID: user.UserID,
|
|
AccountID: strconv.FormatInt(amoAcc.Id, 10),
|
|
AccessToken: token.AccessToken,
|
|
RefreshToken: token.RefreshToken,
|
|
Code: resp.Code,
|
|
FromWidget: resp.FromWidget,
|
|
Referer: resp.Referer,
|
|
Subdomain: amoAcc.Subdomain,
|
|
ExpiresIn: token.Expiry,
|
|
TokenType: token.TokenType,
|
|
})
|
|
|
|
if err != nil {
|
|
h.reportError(w, err, 500)
|
|
return
|
|
}
|
|
|
|
err = sendResponse(w, 200, map[string]string{"accessToken": resp.AccessToken})
|
|
if err != nil {
|
|
h.reportError(w, err, 500)
|
|
}
|
|
}
|
|
|
|
func (h *Handlers) AmoWebhook(w http.ResponseWriter, r *http.Request) {
|
|
fmt.Println(" -------------------- AMO WEBHOOK ------------------------")
|
|
resp := map[string]interface{}{}
|
|
|
|
user := getJwtUser(r)
|
|
if user != nil {
|
|
fmt.Println("USER:", user)
|
|
}
|
|
|
|
err := decodePost(&resp, r)
|
|
if err != nil {
|
|
h.reportError(w, err, 500)
|
|
}
|
|
|
|
fmt.Println("RESPONSE:")
|
|
fmt.Println(resp)
|
|
|
|
sendResponse(w, 200, nil)
|
|
}
|
|
|
|
func (h *Handlers) AmoWebhook2(w http.ResponseWriter, r *http.Request) {
|
|
fmt.Println(" -------------------- AMO WEBHOOK 2------------------------")
|
|
reqBody, err := ioutil.ReadAll(r.Body)
|
|
|
|
if err != nil {
|
|
h.reportError(w, err, 200)
|
|
return
|
|
}
|
|
|
|
p, err := url.ParseQuery(string(reqBody))
|
|
if err != nil {
|
|
h.reportError(w, err, 200)
|
|
return
|
|
}
|
|
|
|
leadId := p.Get("leads[status][0][id]")
|
|
subdomain := p.Get("account[subdomain]")
|
|
accId := p.Get("account[id]")
|
|
|
|
// Запрашиваем данные по аккаунту
|
|
amoData, err := h.dal.Amo.GetByAccountID(r.Context(), accId)
|
|
if err != nil {
|
|
h.reportError(w, err, 200)
|
|
return
|
|
}
|
|
|
|
amoClient := h.Amo.NewClient(subdomain+".amocrm.ru", amoData.Code)
|
|
|
|
amoClient.SetToken(&oauth2.Token{
|
|
AccessToken: amoData.AccessToken,
|
|
TokenType: amoData.TokenType,
|
|
RefreshToken: amoData.RefreshToken,
|
|
Expiry: amoData.ExpiresIn,
|
|
})
|
|
|
|
lead, err := amoClient.GetLeadById(leadId)
|
|
if err != nil {
|
|
h.reportError(w, err, 200)
|
|
return
|
|
}
|
|
|
|
if lead != nil {
|
|
// fmt.Printf("Lead RESP:%+v\r\n", lead)
|
|
} else {
|
|
fmt.Println("Lead empty")
|
|
}
|
|
|
|
dataTemplate := map[string]interface{}{}
|
|
|
|
// Добавляем Инфо Лида
|
|
for k, v := range templategen.AmoLeadFieldsToRuMap(lead) {
|
|
dataTemplate[k] = v
|
|
}
|
|
|
|
filename := fmt.Sprintf("%v", dataTemplate["Filename"])
|
|
|
|
// Добавялем инфо контактов
|
|
contacts := []amo.Contact{}
|
|
for _, data := range lead.Embedded.Contacts {
|
|
contact, err := amoClient.GetContactById(strconv.Itoa(data.Id))
|
|
if err == nil {
|
|
contacts = append(contacts, *contact)
|
|
} else {
|
|
fmt.Println("Something Wrong1:", err)
|
|
}
|
|
}
|
|
|
|
for k, v := range templategen.AmoContactsFieldsToRuMap(contacts) {
|
|
dataTemplate[k] = v
|
|
}
|
|
|
|
// Добавляем инфо компаний
|
|
companies := []amo.Company{}
|
|
for _, data := range lead.Embedded.Companies {
|
|
company, err := amoClient.GetCompanyById(strconv.Itoa(data.Id))
|
|
if err == nil {
|
|
companies = append(companies, *company)
|
|
} else {
|
|
fmt.Println("Something Wrong2:", err)
|
|
}
|
|
}
|
|
|
|
for k, v := range templategen.AmoCompaniesFieldsToRuMap(companies) {
|
|
dataTemplate[k] = v
|
|
}
|
|
|
|
//user := getJwtUser(r)
|
|
//if user != nil {
|
|
// //
|
|
//}
|
|
|
|
fmt.Printf("dataTemplate: %+v\r\n", dataTemplate)
|
|
|
|
user, err := h.dal.User.GetByID(r.Context(), amoData.UserID)
|
|
if err != nil {
|
|
h.reportError(w, err, 200)
|
|
}
|
|
|
|
// PARSING
|
|
yaDiskData, err := h.dal.YaDisk.GetByUserID(r.Context(), user.ID)
|
|
if err != nil {
|
|
h.reportError(w, err, http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
if yaDiskData == nil {
|
|
h.reportError(w, errors.New("YaDisk not found"), 200)
|
|
return
|
|
}
|
|
|
|
// download file
|
|
client := h.YaDisk.NewClient(yaDiskData.AccessToken)
|
|
|
|
res, err := client.GetResources(yaDiskData.TemplateFolder + "/" + filename)
|
|
if res == nil {
|
|
h.reportError(w, errors.New("resource in yandex disk not found"), http.StatusNotFound)
|
|
return
|
|
}
|
|
downloadFileName := fmt.Sprintf("./tmp/downloaded/%v_%v", user.ID, filename)
|
|
|
|
err = templategen.DownloadDocument(downloadFileName, res.File)
|
|
if err != nil {
|
|
h.reportError(w, err, 200)
|
|
return
|
|
}
|
|
|
|
// parsing file
|
|
dc, err := docTemp.GetTemplate(downloadFileName)
|
|
if err != nil {
|
|
h.reportError(w, err, 200)
|
|
return
|
|
}
|
|
|
|
dc.Parse()
|
|
|
|
saveFilename := fmt.Sprintf("%v_%v_%v", user.ID, time.Now().Unix(), filename)
|
|
err = dc.Execute("./tmp/parsed/"+saveFilename, dataTemplate)
|
|
if err != nil {
|
|
h.reportError(w, err, 200)
|
|
return
|
|
}
|
|
|
|
// Send to Yandex Disk
|
|
|
|
err = client.ResourcesUpload(yaDiskData.SaveFolder+"/"+saveFilename, "https://solweb.site/tmp/parsed/"+saveFilename)
|
|
if err != nil {
|
|
h.reportError(w, err, 200)
|
|
return
|
|
}
|
|
|
|
//err := decodePost(&resp, r)
|
|
//if err != nil {
|
|
// h.reportError(w, err, 200)
|
|
//}
|
|
|
|
//err := r.ParseForm()
|
|
//if err != nil {
|
|
// h.reportError(w, err, 200)
|
|
//}
|
|
//
|
|
//err = schema.NewDecoder().Decode(&resp, r.Form)
|
|
//if err != nil {
|
|
// h.reportError(w, err, 200)
|
|
//}
|
|
|
|
//fmt.Println("RESPONSE:", r.PostForm.Encode())
|
|
sendResponse(w, 200, nil)
|
|
}
|