go-ayo/common을 gocommon으로 분리

2023-05-24 15:10:15 +09:00
commit e95efa06a6
21 changed files with 4389 additions and 0 deletions
--- a/azure/graph.go
+++ b/azure/graph.go
@ -0,0 +1,356 @@
+package azure
+
+import (
+	"bytes"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"io"
+	"net/http"
+	"net/url"
+	"os"
+	"path"
+	"time"
+)
+
+type Graph struct {
+	*accessToken
+	tenantId string
+	groupId  string
+}
+
+func NewGraphByEnv() *Graph {
+	tenantId := os.Getenv("GRAPH_TENANT_ID")
+	groupId := os.Getenv("GRAPH_GROUP_ID")
+	return NewGraph(tenantId, groupId)
+}
+
+func NewGraph(tenantId string, groupId string) *Graph {
+	return &Graph{
+		accessToken: &accessToken{
+			url: fmt.Sprintf("https://login.microsoftonline.com/%s/oauth2/v2.0/token", tenantId),
+			values: url.Values{
+				"grant_type":    {"client_credentials"},
+				"client_id":     {microsoftAppId()},
+				"scope":         {"https://graph.microsoft.com/.default"},
+				"client_secret": {microsoftAppPassword()},
+			},
+		},
+		tenantId: tenantId,
+		groupId:  groupId,
+	}
+}
+
+func (s Graph) GroupId() string {
+	return s.groupId
+}
+
+var errGroupIdMissing = errors.New("GRAPH_GROUP_ID is missing")
+
+func (s Graph) ReadFile(relativePath string) (io.ReadCloser, error) {
+	if len(s.groupId) == 0 {
+		return nil, errGroupIdMissing
+	}
+
+	token, err := s.getAuthoizationToken()
+	if err != nil {
+		return nil, err
+	}
+	requrl := fmt.Sprintf("https://graph.microsoft.com/v1.0/groups/%s/drive/root:/%s:/content", s.groupId, relativePath)
+	req, _ := http.NewRequest("GET", requrl, nil)
+
+	req.Header.Add("Authorization", token)
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return nil, err
+	}
+
+	if resp.StatusCode != http.StatusOK {
+		resp.Body.Close()
+		return nil, errors.New(resp.Status)
+	}
+
+	return resp.Body, nil
+}
+
+func (s Graph) GetWebURLOfFile(relativePath string) (string, error) {
+	if len(s.groupId) == 0 {
+		return "", errGroupIdMissing
+	}
+
+	token, err := s.getAuthoizationToken()
+	if err != nil {
+		return "", err
+	}
+	requrl := fmt.Sprintf("https://graph.microsoft.com/v1.0/groups/%s/drive/root:/%s:", s.groupId, relativePath)
+	req, _ := http.NewRequest("GET", requrl, nil)
+
+	req.Header.Add("Authorization", token)
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return "", err
+	}
+	defer func() {
+		io.Copy(io.Discard, resp.Body)
+		resp.Body.Close()
+	}()
+
+	itemdoc, _ := io.ReadAll(resp.Body)
+	var item map[string]interface{}
+	json.Unmarshal(itemdoc, &item)
+
+	url, ok := item["webUrl"]
+	if ok {
+		return url.(string), nil
+	}
+
+	return "", nil
+}
+
+func (s Graph) DownloadFile(relativePath string, outputFile string) error {
+	body, err := s.ReadFile(relativePath)
+	if err != nil {
+		return err
+	}
+	defer body.Close()
+
+	// Create the file
+	out, err := os.Create(outputFile)
+	if err != nil {
+		return err
+	}
+	defer out.Close()
+
+	// Write the body to file
+	_, err = io.Copy(out, body)
+	return err
+}
+
+func (s Graph) UploadStream(relativePath string, reader io.Reader) (string, error) {
+	if len(s.groupId) == 0 {
+		return "", errGroupIdMissing
+	}
+
+	token, err := s.getAuthoizationToken()
+	if err != nil {
+		return "", err
+	}
+
+	requrl := fmt.Sprintf("https://graph.microsoft.com/v1.0/groups/%s/drive/root:/%s:/content", s.groupId, relativePath)
+	req, _ := http.NewRequest("PUT", requrl, reader)
+
+	req.Header.Add("Authorization", token)
+	req.Header.Add("Content-Type", "text/plain")
+
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return "", err
+	}
+
+	defer func() {
+		io.Copy(io.Discard, resp.Body)
+		resp.Body.Close()
+	}()
+
+	body, _ := io.ReadAll(resp.Body)
+	unquot, _ := unescapeUnicodeCharactersInJSON(body)
+
+	var drive map[string]interface{}
+	err = json.Unmarshal([]byte(unquot), &drive)
+	if err != nil {
+		return "", err
+	}
+
+	if errnode, ok := drive["error"]; ok {
+		errmsg := errnode.(map[string]interface{})["message"].(string)
+		return "", errors.New(errmsg)
+	}
+
+	return drive["webUrl"].(string), nil
+}
+
+func (s Graph) UploadBytes(relativePath string, content []byte) (string, error) {
+	return s.UploadStream(relativePath, bytes.NewReader(content))
+}
+
+func (s Graph) GetChannelFilesFolderName(channel string) (string, error) {
+	if len(s.groupId) == 0 {
+		return "", errGroupIdMissing
+	}
+
+	token, err := s.getAuthoizationToken()
+	if err != nil {
+		return "", err
+	}
+
+	requrl := fmt.Sprintf("https://graph.microsoft.com/v1.0/teams/%s/channels/%s/filesFolder", s.groupId, channel)
+	req, _ := http.NewRequest("GET", requrl, nil)
+	req.Header.Add("Authorization", token)
+	req.Header.Add("Accept", "application/json")
+
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return "", err
+	}
+
+	defer func() {
+		io.Copy(io.Discard, resp.Body)
+		resp.Body.Close()
+	}()
+	body, _ := io.ReadAll(resp.Body)
+
+	unquot, _ := unescapeUnicodeCharactersInJSON(body)
+
+	var drive map[string]interface{}
+	err = json.Unmarshal([]byte(unquot), &drive)
+	if err != nil {
+		return "", err
+	}
+
+	if errnode, ok := drive["error"]; ok {
+		errmsg := errnode.(map[string]interface{})["message"].(string)
+		return "", errors.New(errmsg)
+	}
+
+	return drive["name"].(string), nil
+}
+
+func (s Graph) listChildren(path string) (map[string]interface{}, error) {
+	if len(s.groupId) == 0 {
+		return nil, errGroupIdMissing
+	}
+
+	token, err := s.getAuthoizationToken()
+	if err != nil {
+		return nil, err
+	}
+
+	// groupId := "612620b7-cd90-4b3f-a9bd-d34cddf52517"
+	var requrl string
+	if len(path) == 0 {
+		requrl = fmt.Sprintf("https://graph.microsoft.com/v1.0/groups/%s/drive/root/children", s.groupId)
+	} else {
+		requrl = fmt.Sprintf("https://graph.microsoft.com/v1.0/groups/%s/drive/root:/%s:/children", s.groupId, path)
+	}
+	req, _ := http.NewRequest("GET", requrl, nil)
+	req.Header.Add("Authorization", token)
+	req.Header.Add("Accept", "application/json")
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return nil, err
+	}
+
+	defer func() {
+		io.Copy(io.Discard, resp.Body)
+		resp.Body.Close()
+	}()
+	body, _ := io.ReadAll(resp.Body)
+
+	unquot, _ := unescapeUnicodeCharactersInJSON(body)
+
+	var drive map[string]interface{}
+	err = json.Unmarshal([]byte(unquot), &drive)
+	if err != nil {
+		return nil, err
+	}
+
+	if errnode, ok := drive["error"]; ok {
+		errmsg := errnode.(map[string]interface{})["message"].(string)
+		return nil, errors.New(errmsg)
+	}
+
+	return drive, nil
+}
+
+func (s Graph) ListFolders(parent string) ([]string, error) {
+	drive, err := s.listChildren(parent)
+	if err != nil {
+		return nil, err
+	}
+
+	var output []string
+	items := drive["value"].([]interface{})
+	for _, item := range items {
+		itemobj := item.(map[string]interface{})
+		if _, ok := itemobj["file"]; !ok {
+			output = append(output, itemobj["name"].(string))
+		}
+	}
+	return output, nil
+}
+
+type FileMeta struct {
+	Path         string    `json:"path"`
+	LastModified time.Time `json:"lastModified"`
+}
+
+func (s Graph) ListFiles(parent string) ([]FileMeta, error) {
+	drive, err := s.listChildren(parent)
+	if err != nil {
+		return nil, err
+	}
+
+	var output []FileMeta
+	items := drive["value"].([]interface{})
+	for _, item := range items {
+		itemobj := item.(map[string]interface{})
+		if _, ok := itemobj["file"]; ok {
+			modTime, _ := time.Parse(time.RFC3339, itemobj["lastModifiedDateTime"].(string))
+
+			output = append(output, FileMeta{
+				Path:         path.Join(parent, itemobj["name"].(string)),
+				LastModified: modTime,
+			})
+		}
+	}
+	return output, nil
+}
+
+func (s Graph) GetChannels() (map[string]string, error) {
+	if len(s.groupId) == 0 {
+		return nil, errGroupIdMissing
+	}
+
+	token, err := s.getAuthoizationToken()
+	if err != nil {
+		return nil, err
+	}
+
+	requrl := fmt.Sprintf("https://graph.microsoft.com/v1.0/teams/%s/channels", s.groupId)
+	req, _ := http.NewRequest("GET", requrl, nil)
+	req.Header.Add("Authorization", token)
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return nil, err
+	}
+	defer func() {
+		io.Copy(io.Discard, resp.Body)
+		resp.Body.Close()
+	}()
+	body, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return nil, err
+	}
+	var raw map[string]interface{}
+	if err = json.Unmarshal(body, &raw); err != nil {
+		return nil, err
+	}
+
+	if errnode, ok := raw["error"]; ok {
+		return nil, errors.New(errnode.(map[string]interface{})["message"].(string))
+	}
+
+	if r, ok := raw["value"]; ok {
+		valuenode := r.([]interface{})
+		output := make(map[string]string)
+		for _, v := range valuenode {
+			data := v.(map[string]interface{})
+			id := data["id"].(string)
+			displayname := data["displayName"].(string)
+			output[id] = displayname
+		}
+		return output, nil
+	}
+
+	return nil, errors.New("error")
+}