package instagram

import (
	"crypto/sha256"
	"encoding/hex"
	"encoding/json"
	"errors"
	"io"
	"io/ioutil"
	"log"
	"os"

	"github.com/disintegration/imaging"

	"git.1750studios.com/AniNite/SocialDragon/config"
)

type InstagramListResponse struct {
	Tag struct {
		Name            string      `json:"name"`
		ContentAdvisory interface{} `json:"content_advisory"`
		Media           struct {
			Nodes []struct {
				CommentsDisabled bool   `json:"comments_disabled"`
				ID               string `json:"id"`
				Dimensions       struct {
					Height int `json:"height"`
					Width  int `json:"width"`
				} `json:"dimensions"`
				Owner struct {
					ID string `json:"id"`
				} `json:"owner"`
				ThumbnailSrc string `json:"thumbnail_src"`
				IsVideo      bool   `json:"is_video"`
				Code         string `json:"code"`
				Date         int    `json:"date"`
				DisplaySrc   string `json:"display_src"`
				Caption      string `json:"caption"`
				Comments     struct {
					Count int `json:"count"`
				} `json:"comments"`
				Likes struct {
					Count int `json:"count"`
				} `json:"likes"`
			} `json:"nodes"`
			Count    int `json:"count"`
			PageInfo struct {
				HasNextPage bool   `json:"has_next_page"`
				EndCursor   string `json:"end_cursor"`
			} `json:"page_info"`
		} `json:"media"`
		TopPosts struct {
			Nodes []struct {
				ID         string `json:"id"`
				Dimensions struct {
					Height int `json:"height"`
					Width  int `json:"width"`
				} `json:"dimensions"`
				Owner struct {
					ID string `json:"id"`
				} `json:"owner"`
				ThumbnailSrc string `json:"thumbnail_src"`
				IsVideo      bool   `json:"is_video"`
				Code         string `json:"code"`
				Date         int    `json:"date"`
				DisplaySrc   string `json:"display_src"`
				Caption      string `json:"caption"`
				Comments     struct {
					Count int `json:"count"`
				} `json:"comments"`
				Likes struct {
					Count int `json:"count"`
				} `json:"likes"`
			} `json:"nodes"`
		} `json:"top_posts"`
	} `json:"tag"`
}

type InstagramPostResponse struct {
	Graphql struct {
		ShortcodeMedia struct {
			Typename   string `json:"__typename"`
			ID         string `json:"id"`
			Shortcode  string `json:"shortcode"`
			Dimensions struct {
				Height int `json:"height"`
				Width  int `json:"width"`
			} `json:"dimensions"`
			DisplayURL            string `json:"display_url"`
			VideoURL              string `json:"video_url"`
			VideoViewCount        int    `json:"video_view_count"`
			IsVideo               bool   `json:"is_video"`
			EdgeMediaToTaggedUser struct {
				Edges []interface{} `json:"edges"`
			} `json:"edge_media_to_tagged_user"`
			EdgeMediaToCaption struct {
				Edges []struct {
					Node struct {
						Text string `json:"text"`
					} `json:"node"`
				} `json:"edges"`
			} `json:"edge_media_to_caption"`
			CaptionIsEdited    bool `json:"caption_is_edited"`
			EdgeMediaToComment struct {
				Count    int `json:"count"`
				PageInfo struct {
					HasNextPage bool   `json:"has_next_page"`
					EndCursor   string `json:"end_cursor"`
				} `json:"page_info"`
				Edges []struct {
					Node struct {
						ID        string `json:"id"`
						Text      string `json:"text"`
						CreatedAt int    `json:"created_at"`
						Owner     struct {
							ID            string `json:"id"`
							ProfilePicURL string `json:"profile_pic_url"`
							Username      string `json:"username"`
						} `json:"owner"`
					} `json:"node"`
				} `json:"edges"`
			} `json:"edge_media_to_comment"`
			CommentsDisabled     bool `json:"comments_disabled"`
			TakenAtTimestamp     int  `json:"taken_at_timestamp"`
			EdgeMediaPreviewLike struct {
				Count int `json:"count"`
				Edges []struct {
					Node struct {
						ID            string `json:"id"`
						ProfilePicURL string `json:"profile_pic_url"`
						Username      string `json:"username"`
					} `json:"node"`
				} `json:"edges"`
			} `json:"edge_media_preview_like"`
			EdgeMediaToSponsorUser struct {
				Edges []interface{} `json:"edges"`
			} `json:"edge_media_to_sponsor_user"`
			Location       interface{} `json:"location"`
			ViewerHasLiked bool        `json:"viewer_has_liked"`
			Owner          struct {
				ID                string `json:"id"`
				ProfilePicURL     string `json:"profile_pic_url"`
				Username          string `json:"username"`
				FollowedByViewer  bool   `json:"followed_by_viewer"`
				FullName          string `json:"full_name"`
				IsPrivate         bool   `json:"is_private"`
				RequestedByViewer bool   `json:"requested_by_viewer"`
				IsUnpublished     bool   `json:"is_unpublished"`
				BlockedByViewer   bool   `json:"blocked_by_viewer"`
				HasBlockedViewer  bool   `json:"has_blocked_viewer"`
			} `json:"owner"`
			IsAd                       bool `json:"is_ad"`
			EdgeWebMediaToRelatedMedia struct {
				Edges []interface{} `json:"edges"`
			} `json:"edge_web_media_to_related_media"`
		} `json:"shortcode_media"`
	} `json:"graphql"`
}

func LoadList(token string) (*InstagramListResponse, error) {
	var listResponse InstagramListResponse
	var url string
	if token == "" {
		url = "https://www.instagram.com/explore/tags/" + config.C.Instagram.Tag + "/?__a=1"
	} else {
		url = "https://www.instagram.com/explore/tags/" + config.C.Instagram.Tag + "/?__a=1&max_id=" + token
	}
	res, err := GetHTTPResource(url)
	if err != nil {
		return nil, err
	}
	defer res.Body.Close()
	body, err := ioutil.ReadAll(res.Body)
	if err != nil {
		return nil, err
	}
	jerr := json.Unmarshal(body, &listResponse)
	if jerr != nil {
		return nil, jerr
	}
	return &listResponse, nil
}

func LoadPost(code string) (*InstagramPostResponse, error) {
	var postResponse InstagramPostResponse
	res, err := GetHTTPResource("https://www.instagram.com/p/" + code + "/?__a=1")
	if err != nil {
		return nil, err
	}
	defer res.Body.Close()
	body, err := ioutil.ReadAll(res.Body)
	if err != nil {
		return nil, err
	}
	jerr := json.Unmarshal(body, &postResponse)
	if jerr != nil {
		return nil, jerr
	}
	return &postResponse, nil
}

func ImageNameGenerator(seed string) (string, string) {
	seedBytes := []byte(seed)
	sha256Bytes := sha256.Sum256(seedBytes)
	hash := hex.EncodeToString(sha256Bytes[:])
	folders := config.C.ContentDirectory + "/" + hash[0:2] + "/" + hash[0:4] + "/"
	urls := config.C.ContentWebDirectory + "/" + hash[0:2] + "/" + hash[0:4] + "/"
	if err := os.MkdirAll(folders, 0775); err != nil {
		log.Fatalf("FAT Could not create ContentDirectory, error: %+v", err)
	}
	finalPath := folders + hash
	finalUrl := urls + hash
	return finalPath, finalUrl
}

func DownloadMedia(data io.Reader, path string, video bool) (string, error) {
	if video {
		ext := ".mp4"
		blob, err := ioutil.ReadAll(data)
		if err != nil {
			return "", err
		}
		if len(blob) == 0 {
			return "", errors.New("Empty response")
		}
		err = ioutil.WriteFile(path+ext, blob, 0644)
		if err != nil {
			return "", err
		}
		return ext, nil
	} else {
		ext := ".jpg"
		image, err := imaging.Decode(data)
		if err != nil {
			return "", err
		}
		err = imaging.Save(image, path+ext)
		if err != nil {
			return "", err
		}
		return ext, nil
	}
}