329 lines
7.8 KiB
Go
329 lines
7.8 KiB
Go
package handlers
|
||
|
||
import (
|
||
"errors"
|
||
"fmt"
|
||
"github.com/Pena-Co-Ltd/amocrm_templategen_back/dal/model"
|
||
"github.com/Pena-Co-Ltd/amocrm_templategen_back/tools"
|
||
YaDisk "github.com/Pena-Co-Ltd/amocrm_templategen_back/yadisk"
|
||
"github.com/gorilla/schema"
|
||
"go.uber.org/zap"
|
||
"golang.org/x/oauth2"
|
||
"net/http"
|
||
"time"
|
||
)
|
||
|
||
type ReqYaDiskSaveToken struct {
|
||
AccessToken string `json:"access_token" schema:"access_token"`
|
||
Code string `json:"code"`
|
||
ExpiresIn int64 `json:"expires_in" schema:"expires_in"`
|
||
TokenType string `json:"token_type" schema:"token_type"`
|
||
RefreshToken string `json:"refresh_token"`
|
||
State string `json:"state" schema:"state"`
|
||
Scope string `json:"scope" schema:"scope"`
|
||
Error string `json:"error" schema:"error"`
|
||
}
|
||
|
||
func (h *Handlers) YaDiskSaveToken(w http.ResponseWriter, r *http.Request) {
|
||
var req ReqYaDiskSaveToken
|
||
err := r.ParseForm()
|
||
if err != nil {
|
||
h.reportError(w, err, http.StatusBadRequest)
|
||
return
|
||
}
|
||
|
||
err = schema.NewDecoder().Decode(&req, r.Form)
|
||
if err != nil {
|
||
h.reportError(w, err, http.StatusBadRequest)
|
||
return
|
||
}
|
||
|
||
if req.Error != "" {
|
||
err = errors.New("YaDiskErr:" + req.Error)
|
||
h.reportError(w, err, http.StatusBadRequest)
|
||
return
|
||
}
|
||
|
||
if req.AccessToken == "" && req.Code == "" {
|
||
err = errors.New("got empty token")
|
||
h.reportError(w, err, http.StatusBadRequest)
|
||
return
|
||
}
|
||
|
||
if req.State == "" {
|
||
err = errors.New("got empty state")
|
||
h.reportError(w, err, http.StatusBadRequest)
|
||
return
|
||
}
|
||
|
||
var state tools.StateToken
|
||
err = tools.DecryptTokenRC4(req.State, &state)
|
||
|
||
if err != nil {
|
||
h.reportError(w, err, http.StatusUnauthorized)
|
||
return
|
||
}
|
||
|
||
token := &oauth2.Token{
|
||
AccessToken: req.AccessToken,
|
||
TokenType: req.TokenType,
|
||
RefreshToken: req.RefreshToken,
|
||
Expiry: time.Now().Add(time.Duration(req.ExpiresIn) * time.Second),
|
||
}
|
||
|
||
yaDiskClient, err := h.YaDisk.NewClient(r.Context(), token, req.Code)
|
||
|
||
if err != nil {
|
||
h.reportError(w, err, http.StatusForbidden)
|
||
return
|
||
}
|
||
|
||
token = yaDiskClient.Token
|
||
|
||
yaUser, err := yaDiskClient.GetUser()
|
||
if err != nil {
|
||
h.reportError(w, err, http.StatusInternalServerError)
|
||
return
|
||
}
|
||
|
||
// Insert/Update token in DB
|
||
_, err = h.dal.YaDisk.InsertOrUpdate(r.Context(), &model.YaDisk{
|
||
UserID: state.UserID,
|
||
Name: fmt.Sprintf("Yandex Disk (%v)", yaUser.Login),
|
||
UID: yaUser.Uid,
|
||
Login: yaUser.Login,
|
||
DisplayName: yaUser.DisplayName,
|
||
AccessToken: token.AccessToken,
|
||
RefreshToken: token.RefreshToken,
|
||
ExpiresIn: token.Expiry,
|
||
TokenType: token.TokenType,
|
||
TemplateFolder: YaDisk.DEFAULT_TEMPLATE_FOLDER,
|
||
SaveFolder: YaDisk.DEFAULT_SAVE_FOLDER,
|
||
})
|
||
|
||
if err != nil {
|
||
h.reportError(w, err, 500)
|
||
return
|
||
}
|
||
|
||
// Make default directories in YandexDisk
|
||
_, err = yaDiskClient.PutResources(YaDisk.DEFAULT_FOLDER)
|
||
if err != nil {
|
||
h.logger.Error("ErrorHandler", zap.Error(err))
|
||
}
|
||
|
||
_, err = yaDiskClient.PutResources(YaDisk.DEFAULT_TEMPLATE_FOLDER)
|
||
if err != nil {
|
||
h.logger.Error("ErrorHandler", zap.Error(err))
|
||
}
|
||
|
||
_, err = yaDiskClient.PutResources(YaDisk.DEFAULT_SAVE_FOLDER)
|
||
if err != nil {
|
||
h.logger.Error("ErrorHandler", zap.Error(err))
|
||
}
|
||
|
||
http.Redirect(w, r, state.RedirectUrl, http.StatusPermanentRedirect)
|
||
}
|
||
|
||
type ReqYaDiskSetSettings struct {
|
||
ID string `json:"id"`
|
||
Name string `json:"name"`
|
||
TemplateFolder string `json:"template_folder"`
|
||
SaveFolder string `json:"save_folder"`
|
||
}
|
||
|
||
func (h *Handlers) YaDiskSetSettings(w http.ResponseWriter, r *http.Request) {
|
||
var req ReqYaDiskSetSettings
|
||
|
||
amoData := getAmoByJwt(r)
|
||
if amoData == nil {
|
||
h.reportError(w, ErrorUnauthorized, http.StatusUnauthorized)
|
||
return
|
||
}
|
||
|
||
err := decodePost(&req, r)
|
||
if err != nil {
|
||
h.reportError(w, err, http.StatusBadRequest)
|
||
return
|
||
}
|
||
|
||
if req.ID == "" {
|
||
h.reportError(w, errors.New("id required"), http.StatusBadRequest)
|
||
return
|
||
}
|
||
|
||
err = h.dal.YaDisk.Update(r.Context(), &model.YaDisk{
|
||
ID: req.ID,
|
||
Name: req.Name,
|
||
TemplateFolder: req.TemplateFolder,
|
||
SaveFolder: req.SaveFolder})
|
||
|
||
if err != nil {
|
||
h.reportError(w, err, http.StatusInternalServerError)
|
||
return
|
||
}
|
||
|
||
sendResponse(w, 200, nil)
|
||
}
|
||
|
||
// YaDiskGetList - возвращает список хранилищ Yandex закрепленных за пользователем по его UserID из JWT
|
||
func (h *Handlers) YaDiskGetList(w http.ResponseWriter, r *http.Request) {
|
||
amoData := getAmoByJwt(r)
|
||
|
||
if amoData == nil {
|
||
h.reportError(w, ErrorUnauthorized, http.StatusUnauthorized)
|
||
return
|
||
}
|
||
|
||
yadiskInfo, err := h.dal.YaDisk.GetListByUserID(r.Context(), amoData.UserID)
|
||
if err != nil {
|
||
h.reportError(w, err, http.StatusInternalServerError)
|
||
return
|
||
}
|
||
|
||
sendResponse(w, http.StatusOK, yadiskInfo)
|
||
}
|
||
|
||
type ReqYaDiskGetResource struct {
|
||
ID string `json:"id"` // ID хранилища
|
||
Path string `json:"path"` // Путь до папки. По умолчанию: disk:/
|
||
Limit int `json:"limit"` // По умолчанию: 20
|
||
Offset int `json:"offset"` // По умолчанию: 0
|
||
}
|
||
|
||
func (h *Handlers) YaDiskGetResources(w http.ResponseWriter, r *http.Request) {
|
||
var req ReqYaDiskGetResource
|
||
|
||
amoData := getAmoByJwt(r)
|
||
|
||
if amoData == nil {
|
||
h.reportError(w, ErrorUnauthorized, http.StatusUnauthorized)
|
||
return
|
||
}
|
||
|
||
err := decodePost(&req, r)
|
||
if err != nil {
|
||
h.reportError(w, err, http.StatusBadRequest)
|
||
return
|
||
}
|
||
|
||
if req.ID == "" {
|
||
h.reportError(w, errors.New("id required"), http.StatusBadRequest)
|
||
return
|
||
}
|
||
|
||
yadisk, err := h.dal.YaDisk.GetByID(r.Context(), req.ID)
|
||
if err != nil {
|
||
h.reportError(w, err, http.StatusInternalServerError)
|
||
return
|
||
}
|
||
|
||
client, err := h.YaDisk.NewClient(r.Context(), yadisk.Token(), "")
|
||
if err != nil {
|
||
h.reportError(w, err, http.StatusInternalServerError)
|
||
return
|
||
}
|
||
|
||
resource, err := client.GetResources(req.Path, req.Limit, req.Offset)
|
||
if err != nil {
|
||
h.reportError(w, err, http.StatusInternalServerError)
|
||
return
|
||
}
|
||
|
||
sendResponse(w, 200, resource)
|
||
}
|
||
|
||
type ReqYaDiskPutResources struct {
|
||
ID string `json:"id"`
|
||
Path string `json:"path"`
|
||
}
|
||
|
||
func (h *Handlers) YaDiskPutResources(w http.ResponseWriter, r *http.Request) {
|
||
var req ReqYaDiskPutResources
|
||
|
||
amoData := getAmoByJwt(r)
|
||
|
||
if amoData == nil {
|
||
h.reportError(w, ErrorUnauthorized, http.StatusUnauthorized)
|
||
return
|
||
}
|
||
|
||
err := decodePost(&req, r)
|
||
if err != nil {
|
||
h.reportError(w, err, http.StatusBadRequest)
|
||
return
|
||
}
|
||
|
||
if req.ID == "" {
|
||
h.reportError(w, errors.New("id required"), http.StatusBadRequest)
|
||
return
|
||
}
|
||
|
||
yadisk, err := h.dal.YaDisk.GetByID(r.Context(), req.ID)
|
||
if err != nil {
|
||
h.reportError(w, err, http.StatusInternalServerError)
|
||
return
|
||
}
|
||
|
||
client, err := h.YaDisk.NewClient(r.Context(), yadisk.Token(), "")
|
||
if err != nil {
|
||
h.reportError(w, err, http.StatusInternalServerError)
|
||
return
|
||
}
|
||
|
||
res, err := client.PutResources(req.Path)
|
||
if err != nil {
|
||
h.reportError(w, err, http.StatusInternalServerError)
|
||
return
|
||
}
|
||
|
||
sendResponse(w, http.StatusOK, res)
|
||
}
|
||
|
||
type ReqYaDiskDeleteResources struct {
|
||
ID string `json:"ID"`
|
||
Path string `json:"path"`
|
||
}
|
||
|
||
func (h *Handlers) YaDiskDeleteResources(w http.ResponseWriter, r *http.Request) {
|
||
var req ReqYaDiskPutResources
|
||
|
||
amoData := getAmoByJwt(r)
|
||
|
||
if amoData == nil {
|
||
h.reportError(w, ErrorUnauthorized, http.StatusUnauthorized)
|
||
return
|
||
}
|
||
|
||
err := decodePost(&req, r)
|
||
if err != nil {
|
||
h.reportError(w, err, http.StatusBadRequest)
|
||
return
|
||
}
|
||
|
||
if req.ID == "" {
|
||
h.reportError(w, errors.New("id required"), http.StatusBadRequest)
|
||
return
|
||
}
|
||
|
||
yadisk, err := h.dal.YaDisk.GetByID(r.Context(), req.ID)
|
||
if err != nil {
|
||
h.reportError(w, err, http.StatusInternalServerError)
|
||
return
|
||
}
|
||
|
||
client, err := h.YaDisk.NewClient(r.Context(), yadisk.Token(), "")
|
||
if err != nil {
|
||
h.reportError(w, err, http.StatusInternalServerError)
|
||
return
|
||
}
|
||
|
||
err = client.DeleteResources(req.Path)
|
||
if err != nil {
|
||
h.reportError(w, err, http.StatusInternalServerError)
|
||
return
|
||
}
|
||
|
||
w.WriteHeader(http.StatusOK)
|
||
}
|