package state

import (
	"fmt"
	"net/http"
	"strconv"
	"time"

	comatproto "github.com/bluesky-social/indigo/api/atproto"
	"github.com/bluesky-social/indigo/atproto/syntax"
	lexutil "github.com/bluesky-social/indigo/lex/util"
	indigoxrpc "github.com/bluesky-social/indigo/xrpc"

	"tangled.org/core/api/tangled"
	"tangled.org/core/appview/db"
	"tangled.org/core/appview/models"
	"tangled.org/core/appview/pages"
	"tangled.org/core/orm"
	"tangled.org/core/tid"
)

func (s *State) CommentBodyFragment(w http.ResponseWriter, r *http.Request) {
	l := s.logger.With("handler", "CommentBodyFragment")

	commentAt := r.URL.Query().Get("aturi")
	comment, err := db.GetComment(s.db, orm.FilterEq("at_uri", commentAt))
	if err != nil {
		l.Error("failed to fetch comment", "aturi", commentAt)
		http.Error(w, "Failed to fetch comment", http.StatusInternalServerError)
		return
	}

	err = s.pages.CommentBodyFragment(w, pages.CommentBodyFragmentParams{
		Comment: comment,
	})
	if err != nil {
		l.Error("failed to render")
	}
}

func (s *State) EditCommentFragment(w http.ResponseWriter, r *http.Request) {
	l := s.logger.With("handler", "EditCommentFragment")

	commentAt := r.URL.Query().Get("aturi")
	comment, err := db.GetComment(s.db, orm.FilterEq("at_uri", commentAt))
	if err != nil {
		l.Error("failed to fetch comment", "aturi", commentAt)
		http.Error(w, "Failed to fetch comment", http.StatusInternalServerError)
		return
	}

	err = s.pages.EditCommentFragment(w, pages.EditCommentFragmentParams{
		Comment: comment,
	})
	if err != nil {
		l.Error("failed to render")
	}
}

func (s *State) NewReplyCommentFragment(w http.ResponseWriter, r *http.Request) {
	s.pages.ReplyCommentFragment(w, pages.ReplyCommentFragmentParams{
		LoggedInUser: s.oauth.GetMultiAccountUser(r),
	})
}

func (s *State) ReplyPlaceholderFragment(w http.ResponseWriter, r *http.Request) {
	s.pages.ReplyPlaceholderFragment(w, pages.ReplyPlaceholderFragmentParams{
		LoggedInUser: s.oauth.GetMultiAccountUser(r),
	})
}

func (s *State) NewComment(w http.ResponseWriter, r *http.Request) {
	l := s.logger.With("handler", "NewComment")
	user := s.oauth.GetMultiAccountUser(r)

	noticeId := "comment-error"
	ctx := r.Context()

	body := r.FormValue("body")
	if body == "" {
		s.pages.Notice(w, noticeId, "Body is required")
		return
	}

	// TODO(boltless): normalize markdown body
	normalizedBody := body
	_, references := s.mentionsResolver.Resolve(ctx, body)

	markdownBody := tangled.MarkupMarkdown{
		Text:     normalizedBody,
		Original: &body,
		Blobs:    nil,
	}

	subjectUri, err := syntax.ParseATURI(r.FormValue("subject-uri"))
	if err != nil {
		l.Warn("invalid subject uri", "err", err)
		s.pages.Notice(w, noticeId, "Subject URI should be valid AT-URI")
		return
	}
	l = l.With("subject.uri", subjectUri)

	// ingest CID of subject record on-demand.
	// TODO(boltless): appview should ingest CID of all atproto records
	var subjectCid syntax.CID
	if subjectCidRaw := r.FormValue("subject-cid"); subjectCidRaw != "" {
		subjectCid, err = syntax.ParseCID(subjectCidRaw)
		if err != nil {
			l.Warn("invalid subject cid", "err", err)
			s.pages.Notice(w, noticeId, "Subject URI should be valid AT-URI")
			return
		}
	} else {
		l.Debug("ingesting subject record CID")
		subjectCid, err = func(uri syntax.ATURI) (syntax.CID, error) {
			ident, err := s.idResolver.ResolveIdent(ctx, uri.Authority().String())
			if err != nil {
				return "", err
			}

			xrpcc := indigoxrpc.Client{Host: ident.PDSEndpoint()}
			out, err := comatproto.RepoGetRecord(ctx, &xrpcc, "", uri.Collection().String(), ident.DID.String(), uri.RecordKey().String())
			if err != nil {
				return "", err
			}
			if out.Cid == nil {
				return "", fmt.Errorf("record CID is empty")
			}

			cid, err := syntax.ParseCID(*out.Cid)
			if err != nil {
				return "", err
			}

			return cid, nil
		}(subjectUri)
		if err != nil {
			l.Error("failed to backfill subject record", "err", err)
			s.pages.Notice(w, noticeId, "failed to backfill subject record")
			return
		}
	}
	l = l.With("subject.cid", subjectCid)

	subject := comatproto.RepoStrongRef{
		Uri: subjectUri.String(),
		Cid: subjectCid.String(),
	}

	var pullRoundIdx *int
	if pullRoundIdxRaw := r.FormValue("pull-round-idx"); pullRoundIdxRaw != "" {
		roundIdx, err := strconv.Atoi(pullRoundIdxRaw)
		if err != nil {
			l.Warn("invalid round idx", "err", err)
			s.pages.Notice(w, noticeId, "pull round index should be valid integer")
			return
		}
		pullRoundIdx = &roundIdx
	}

	var replyTo *comatproto.RepoStrongRef
	replyToUriRaw := r.FormValue("reply-to-uri")
	replyToCidRaw := r.FormValue("reply-to-cid")
	if replyToUriRaw != "" && replyToCidRaw != "" {
		uri, err := syntax.ParseATURI(replyToUriRaw)
		if err != nil {
			s.pages.Notice(w, noticeId, "reply-to-uri should be valid AT-URI")
			return
		}
		cid, err := syntax.ParseCID(replyToCidRaw)
		if err != nil {
			s.pages.Notice(w, noticeId, "reply-to-cid should be valid CID")
			return
		}
		replyTo = &comatproto.RepoStrongRef{
			Uri: uri.String(),
			Cid: cid.String(),
		}
	}

	comment := models.Comment{
		Did:        syntax.DID(user.Active.Did),
		Collection: tangled.FeedCommentNSID,
		Rkey:       syntax.RecordKey(tid.TID()),

		Subject:      subject,
		Body:         markdownBody,
		Created:      time.Now(),
		ReplyTo:      replyTo,
		PullRoundIdx: pullRoundIdx,
	}
	if err = comment.Validate(); err != nil {
		l.Error("failed to validate comment", "err", err)
		s.pages.Notice(w, noticeId, "Failed to create comment.")
		return
	}

	client, err := s.oauth.AuthorizedClient(r)
	if err != nil {
		l.Error("failed to get authorized client", "err", err)
		s.pages.Notice(w, noticeId, "Failed to create comment.")
		return
	}

	// create a record first
	out, err := comatproto.RepoPutRecord(ctx, client, &comatproto.RepoPutRecord_Input{
		Collection: comment.Collection.String(),
		Repo:       comment.Did.String(),
		Rkey:       comment.Rkey.String(),
		Record:     &lexutil.LexiconTypeDecoder{Val: comment.AsRecord()},
	})
	if err != nil {
		l.Error("failed to create comment", "err", err)
		s.pages.Notice(w, noticeId, "Failed to create comment.")
		return
	}

	comment.Cid = syntax.CID(out.Cid)

	tx, err := s.db.Begin()
	if err != nil {
		l.Error("failed to start transaction", "err", err)
		s.pages.Notice(w, noticeId, "Failed to create comment, try again later.")
		return
	}
	defer tx.Rollback()

	err = db.PutComment(tx, &comment, references)
	if err != nil {
		l.Error("failed to create comment", "err", err)
		s.pages.Notice(w, noticeId, "Failed to create comment.")
		return
	}

	err = tx.Commit()
	if err != nil {
		l.Error("failed to commit transaction", "err", err)
		s.pages.Notice(w, noticeId, "Failed to create comment, try again later.")
		return
	}

	// TODO: return comment or reply-comment fragment
	// onattach, htmx-callback to focus on comment.
	s.pages.HxRefresh(w)
}

func (s *State) EditComment(w http.ResponseWriter, r *http.Request) {
	l := s.logger.With("handler", "EditComment")
	user := s.oauth.GetMultiAccountUser(r)

	noticeId := "comment-error"
	ctx := r.Context()

	commentAt := r.FormValue("aturi")
	comment, err := db.GetComment(s.db, orm.FilterEq("at_uri", commentAt))
	if err != nil {
		l.Error("failed to fetch comment", "aturi", commentAt, "err", err)
		s.pages.Notice(w, noticeId, "Failed to fetch comment")
		return
	}

	if comment.Did.String() != user.Active.Did {
		l.Error("unauthorized comment edit", "expectedDid", comment.Did, "gotDid", user.Active.Did)
		s.pages.Notice(w, noticeId, "You are not the author of this comment")
		return
	}

	body := r.FormValue("body")
	if body == "" {
		s.pages.Notice(w, noticeId, "Body is required")
		return
	}

	// TODO(boltless): normalize markdown body
	normalizedBody := body
	_, references := s.mentionsResolver.Resolve(ctx, body)

	now := time.Now()
	newComment := comment
	newComment.Body = tangled.MarkupMarkdown{
		Text:     normalizedBody,
		Original: &body,
		Blobs:    nil,
	}
	newComment.Edited = &now
	if err := newComment.Validate(); err != nil {
		l.Error("failed to validate comment", "err", err)
		s.pages.Notice(w, noticeId, "Failed to update comment.")
		return
	}

	client, err := s.oauth.AuthorizedClient(r)
	if err != nil {
		l.Error("failed to get authorized client", "err", err)
		s.pages.Notice(w, noticeId, "Failed to create comment. try again later.")
		return
	}

	// update the record first
	exCid := comment.Cid.String()
	out, err := comatproto.RepoPutRecord(ctx, client, &comatproto.RepoPutRecord_Input{
		Collection: newComment.Collection.String(),
		Repo:       newComment.Did.String(),
		Rkey:       newComment.Rkey.String(),
		SwapRecord: &exCid,
		Record: &lexutil.LexiconTypeDecoder{
			Val: newComment.AsRecord(),
		},
	})
	if err != nil {
		l.Error("failed to update comment", "err", err)
		s.pages.Notice(w, noticeId, "Failed to update comment, try again later.")
		return
	}

	newComment.Cid = syntax.CID(out.Cid)

	tx, err := s.db.Begin()
	if err != nil {
		l.Error("failed to start transaction", "err", err)
		s.pages.Notice(w, noticeId, "Failed to update comment, try again later.")
		return
	}
	defer tx.Rollback()

	err = db.PutComment(tx, &newComment, references)
	if err != nil {
		l.Error("failed to perform update-description query", "err", err)
		s.pages.Notice(w, noticeId, "Failed to update comment, try again later.")
		return
	}
	err = tx.Commit()
	if err != nil {
		l.Error("failed to commit transaction", "err", err)
		s.pages.Notice(w, noticeId, "Failed to update comment, try again later.")
		return
	}

	// TODO: return full comment fragment so we can update comment header too
	s.pages.CommentBodyFragment(w, pages.CommentBodyFragmentParams{
		Comment: newComment,
	})
}

func (s *State) DeleteComment(w http.ResponseWriter, r *http.Request) {
	l := s.logger.With("handler", "DeleteComment")
	user := s.oauth.GetMultiAccountUser(r)

	noticeId := "comment"
	ctx := r.Context()

	commentAt := r.URL.Query().Get("aturi")
	comment, err := db.GetComment(s.db, orm.FilterEq("at_uri", commentAt))
	if err != nil {
		l.Error("failed to fetch comment", "aturi", commentAt)
		s.pages.Notice(w, noticeId, "Failed to fetch comment.")
		return
	}

	if comment.Did.String() != user.Active.Did {
		l.Error("unauthorized action", "expectedDid", comment.Did, "gotDid", user.Active.Did)
		s.pages.Notice(w, noticeId, "you are not the author of this comment")
		return
	}

	if comment.Deleted != nil {
		s.pages.Notice(w, noticeId, "Comment already deleted")
		return
	}

	client, err := s.oauth.AuthorizedClient(r)
	if err != nil {
		l.Error("failed to get authorized client", "err", err)
		s.pages.Notice(w, "comment", "Failed to delete comment.")
		return
	}
	_, err = comatproto.RepoDeleteRecord(ctx, client, &comatproto.RepoDeleteRecord_Input{
		Collection: comment.Collection.String(),
		Repo:       comment.Did.String(),
		Rkey:       comment.Rkey.String(),
	})
	if err != nil {
		l.Error("failed to delete from PDS", "err", err)
		s.pages.Notice(w, noticeId, "Failed to delete comment, try again later.")
		return
	}

	// optimistic update for htmx response
	now := time.Now()
	comment.Body = tangled.MarkupMarkdown{}
	comment.Deleted = &now

	s.pages.CommentBodyFragment(w, pages.CommentBodyFragmentParams{
		Comment: comment,
	})
}