container_builder.go

package threads

import (
	"encoding/json"
	"net/url"
	"strings"
)

// ContainerBuilder helps build container creation parameters
type ContainerBuilder struct {
	params url.Values
}

// NewContainerBuilder creates a new container builder
func NewContainerBuilder() *ContainerBuilder {
	return &ContainerBuilder{
		params: url.Values{},
	}
}

// SetMediaType sets the media type.
// If a non-TEXT type is set, any previously set is_ghost_post flag is cleared
// since ghost posts are only supported for TEXT.
func (b *ContainerBuilder) SetMediaType(mediaType string) *ContainerBuilder {
	b.params.Set("media_type", mediaType)
	if mediaType != "" && mediaType != MediaTypeText {
		b.params.Del("is_ghost_post")
	}
	return b
}

// SetText sets the text content
func (b *ContainerBuilder) SetText(text string) *ContainerBuilder {
	if strings.TrimSpace(text) != "" {
		b.params.Set("text", text)
	}
	return b
}

// SetImageURL sets the image URL for image posts
func (b *ContainerBuilder) SetImageURL(imageURL string) *ContainerBuilder {
	if imageURL != "" {
		b.params.Set("image_url", imageURL)
	}
	return b
}

// SetVideoURL sets the video URL for video posts
func (b *ContainerBuilder) SetVideoURL(videoURL string) *ContainerBuilder {
	if videoURL != "" {
		b.params.Set("video_url", videoURL)
	}
	return b
}

// SetAltText sets the alt text for media
func (b *ContainerBuilder) SetAltText(altText string) *ContainerBuilder {
	if altText != "" {
		b.params.Set("alt_text", altText)
	}
	return b
}

// SetReplyControl sets who can reply to the post
func (b *ContainerBuilder) SetReplyControl(replyControl ReplyControl) *ContainerBuilder {
	if replyControl != "" {
		b.params.Set("reply_control", string(replyControl))
	}
	return b
}

// SetReplyTo sets the ID of the post being replied to
func (b *ContainerBuilder) SetReplyTo(replyToID string) *ContainerBuilder {
	if replyToID != "" {
		b.params.Set("reply_to_id", replyToID)
	}
	return b
}

// SetTopicTag sets the topic tag
func (b *ContainerBuilder) SetTopicTag(tag string) *ContainerBuilder {
	if tag != "" {
		b.params.Set("topic_tag", tag)
	}
	return b
}

// SetLocationID sets the location ID
func (b *ContainerBuilder) SetLocationID(locationID string) *ContainerBuilder {
	if locationID != "" {
		b.params.Set("location_id", locationID)
	}
	return b
}

// SetQuotePostID sets the quoted post ID
func (b *ContainerBuilder) SetQuotePostID(quotePostID string) *ContainerBuilder {
	if quotePostID != "" {
		b.params.Set("quote_post_id", quotePostID)
	}
	return b
}

// SetLinkAttachment sets the link attachment
func (b *ContainerBuilder) SetLinkAttachment(linkURL string) *ContainerBuilder {
	if linkURL != "" {
		b.params.Set("link_attachment", linkURL)
	}
	return b
}

// SetPollAttachment sets the poll attachment
func (b *ContainerBuilder) SetPollAttachment(poll *PollAttachment) *ContainerBuilder {
	if poll != nil {
		pollJSON, err := json.Marshal(poll)
		if err == nil {
			b.params.Set("poll_attachment", string(pollJSON))
		}
	}
	return b
}

// SetAllowlistedCountryCodes sets geo-gating country codes
func (b *ContainerBuilder) SetAllowlistedCountryCodes(codes []string) *ContainerBuilder {
	for _, code := range codes {
		b.params.Add("allowlisted_country_codes", code)
	}
	return b
}

// AddChild adds a child container ID (for carousel posts)
func (b *ContainerBuilder) AddChild(childID string) *ContainerBuilder {
	if childID == "" {
		return b
	}
	existing := b.params.Get("children")
	if existing != "" {
		b.params.Set("children", existing+","+childID)
	} else {
		b.params.Set("children", childID)
	}
	return b
}

// SetChildren sets all children container IDs at once (for carousel posts)
func (b *ContainerBuilder) SetChildren(childIDs []string) *ContainerBuilder {
	if len(childIDs) == 0 {
		b.params.Del("children")
		return b
	}
	b.params.Set("children", strings.Join(childIDs, ","))
	return b
}

// SetAutoPublishText sets whether to auto-publish text posts
func (b *ContainerBuilder) SetAutoPublishText(autoPublish bool) *ContainerBuilder {
	if autoPublish {
		b.params.Set("auto_publish_text", "true")
	}
	return b
}

// SetIsCarouselItem marks this as a carousel item
func (b *ContainerBuilder) SetIsCarouselItem(isCarouselItem bool) *ContainerBuilder {
	if isCarouselItem {
		b.params.Set("is_carousel_item", "true")
	}
	return b
}

// SetTextEntities adds text spoiler entities
// Marks specific text ranges as spoilers using offset and length
// Max 10 entities per post
func (b *ContainerBuilder) SetTextEntities(entities []TextEntity) *ContainerBuilder {
	if len(entities) > 0 {
		entitiesJSON, err := json.Marshal(entities)
		if err == nil {
			b.params.Set("text_entities", string(entitiesJSON))
		}
	}
	return b
}

// SetIsSpoilerMedia marks media (IMAGE, VIDEO, CAROUSEL) as spoilers
// For CAROUSEL media type, this marks ALL media in the carousel as spoilers
func (b *ContainerBuilder) SetIsSpoilerMedia(isSpoilerMedia bool) *ContainerBuilder {
	if isSpoilerMedia {
		b.params.Set("is_spoiler_media", "true")
	}
	return b
}

// SetTextAttachment adds a text attachment to the post
// Can only be used with TEXT-only posts (not with polls or other media)
// Max 10,000 characters in the plaintext field
func (b *ContainerBuilder) SetTextAttachment(textAttachment *TextAttachment) *ContainerBuilder {
	if textAttachment != nil {
		attachmentJSON, err := json.Marshal(textAttachment)
		if err == nil {
			b.params.Set("text_attachment", string(attachmentJSON))
		}
	}
	return b
}

// SetGIFAttachment adds a GIF attachment to the post
// Can only be used with TEXT-only posts (not with image, video, or carousel posts)
// Supported providers: TENOR (deprecated, sunset March 31, 2026) and GIPHY
func (b *ContainerBuilder) SetGIFAttachment(gifAttachment *GIFAttachment) *ContainerBuilder {
	if gifAttachment != nil {
		attachmentJSON, err := json.Marshal(gifAttachment)
		if err == nil {
			b.params.Set("gif_attachment", string(attachmentJSON))
		}
	}
	return b
}

// SetIsGhostPost marks the post as a ghost post (text-only, expires in 24h, no replies allowed).
// Ghost posts are only supported for TEXT media type. If the builder's media_type is already set
// to a non-TEXT value this call is silently ignored; if a non-TEXT type is set after this call,
// SetMediaType will automatically clear the flag.
func (b *ContainerBuilder) SetIsGhostPost(isGhostPost bool) *ContainerBuilder {
	if isGhostPost {
		// Ghost posts are only allowed for TEXT media type
		mediaType := b.params.Get("media_type")
		if mediaType != "" && mediaType != MediaTypeText {
			// Silently ignore - ghost post flag is not applicable to non-text posts
			return b
		}
		b.params.Set("is_ghost_post", "true")
	} else {
		b.params.Del("is_ghost_post")
	}
	return b
}

// SetEnableReplyApprovals enables reply approvals on the post
func (b *ContainerBuilder) SetEnableReplyApprovals(enable bool) *ContainerBuilder {
	if enable {
		b.params.Set("enable_reply_approvals", "true")
	}
	return b
}

// Build returns the built parameters
func (b *ContainerBuilder) Build() url.Values {
	return b.params
}