// Copyright (c) 2024, Přemysl Eric Janouch <p@janouch.name>
// SPDX-License-Identifier: 0BSD

package main

import (
	"bufio"
	"context"
	"encoding/binary"
	"flag"
	"fmt"
	"image/color"
	"io"
	"log"
	"net"
	"net/url"
	"os"
	"slices"
	"time"

	"fyne.io/fyne/v2"
	"fyne.io/fyne/v2/app"
	"fyne.io/fyne/v2/container"
	"fyne.io/fyne/v2/theme"
	"fyne.io/fyne/v2/widget"
)

var (
	debug          = flag.Bool("debug", false, "enable debug output")
	projectName    = "xA"
	projectVersion = "?"
)

// --- Theme -------------------------------------------------------------------

type customTheme struct{}

func convertColor(c int) color.Color {
	base16 := []uint16{
		0x000, 0x800, 0x080, 0x880, 0x008, 0x808, 0x088, 0xccc,
		0x888, 0xf00, 0x0f0, 0xff0, 0x00f, 0xf0f, 0x0ff, 0xfff,
	}
	if c < 16 {
		r := 0xf & uint8(base16[c]>>8)
		g := 0xf & uint8(base16[c]>>4)
		b := 0xf & uint8(base16[c])
		return color.RGBA{r * 0x11, g * 0x11, b * 0x11, 0xff}
	}
	if c >= 216 {
		return color.Gray{8 + uint8(c-216)*10}
	}

	var (
		i = uint8(c - 16)
		r = i / 36 >> 0
		g = (i / 6 >> 0) % 6
		b = i % 6
	)
	if r != 0 {
		r = 55 + 40*r
	}
	if g != 0 {
		g = 55 + 40*g
	}
	if b != 0 {
		b = 55 + 40*b
	}
	return color.RGBA{r, g, b, 0xff}
}

var ircColors = make(map[fyne.ThemeColorName]color.Color)

func ircColorName(color int) fyne.ThemeColorName {
	return fyne.ThemeColorName(fmt.Sprintf("irc%02x", color))
}

func init() {
	for color := 0; color < 256; color++ {
		ircColors[ircColorName(color)] = convertColor(color)
	}
}

func (t *customTheme) Color(
	name fyne.ThemeColorName, variant fyne.ThemeVariant) color.Color {
	// Fuck this low contrast shit, text must be black.
	if name == theme.ColorNameForeground &&
		variant == theme.VariantLight {
		return color.Black
	}

	// TODO(p): Consider constants for stuff like timestamps.
	if c, ok := ircColors[name]; ok {
		return c
	}
	return theme.DefaultTheme().Color(name, variant)
}

func (t *customTheme) Font(style fyne.TextStyle) fyne.Resource {
	return theme.DefaultTheme().Font(style)
}

func (t *customTheme) Icon(i fyne.ThemeIconName) fyne.Resource {
	return theme.DefaultTheme().Icon(i)
}

func (t *customTheme) Size(s fyne.ThemeSizeName) float32 {
	return theme.DefaultTheme().Size(s)
}

// --- Relay state -------------------------------------------------------------

type server struct {
	state     RelayServerState
	user      string
	userModes string
}

type bufferLineItem struct {
	format fyne.TextStyle
	// For RichTextStyle.ColorName.
	color fyne.ThemeColorName
	// XXX: Fyne's RichText doesn't support background colours.
	background fyne.ThemeColorName
	text       string
}

type bufferLine struct {
	/// Leaked from another buffer, but temporarily staying in another one.
	leaked bool

	isUnimportant bool
	isHighlight   bool
	rendition     RelayRendition
	when          time.Time
	items         []bufferLineItem
}

type buffer struct {
	bufferName      string
	hideUnimportant bool
	kind            RelayBufferKind
	serverName      string
	lines           []bufferLine
	// TODO(p): Server by name or by pointer?

	// Channel:

	topic []bufferLineItem
	modes string

	// Stats:

	newMessages            int
	newUnimportantMessages int
	highlighted            bool

	// Input:

	input                string
	inputStart, inputEnd int
	history              []string
	historyAt            int
}

type callback func(err string, response *RelayResponseData)

var (
	backendAddress string
	backendContext context.Context
	backendCancel  context.CancelFunc

	// Connection state:

	commandSeq       uint32
	commandCallbacks = make(map[uint32]callback)

	buffers       []buffer
	bufferCurrent string
	bufferLast    string

	servers = make(map[string]*server)

	// Widgets:

	wRichText   *widget.RichText
	wRichScroll *container.Scroll
	wEntry      *widget.Entry
)

// -----------------------------------------------------------------------------

func defaultBufferLineItem() bufferLineItem { return bufferLineItem{} }

func convertItemFormatting(
	item RelayItemData, cf *bufferLineItem, inverse *bool) {
	switch data := item.Variant.(type) {
	case *RelayItemDataReset:
		*cf = defaultBufferLineItem()
	case *RelayItemDataFlipBold:
		cf.format.Bold = !cf.format.Bold
	case *RelayItemDataFlipItalic:
		cf.format.Italic = !cf.format.Italic
	case *RelayItemDataFlipUnderline:
		cf.format.Underline = !cf.format.Underline
	case *RelayItemDataFlipCrossedOut:
		// ¯\_(ツ)_/¯
	case *RelayItemDataFlipInverse:
		*inverse = !*inverse
	case *RelayItemDataFlipMonospace:
		cf.format.Monospace = !cf.format.Monospace
	case *RelayItemDataFgColor:
		if data.Color < 0 {
			cf.color = ""
		} else {
			cf.color = ircColorName(int(data.Color))
		}
	case *RelayItemDataBgColor:
		if data.Color < 0 {
			cf.background = ""
		} else {
			cf.background = ircColorName(int(data.Color))
		}
	}
}

func convertItems(items []RelayItemData) []bufferLineItem {
	result := []bufferLineItem{}
	cf, inverse := defaultBufferLineItem(), false
	for _, it := range items {
		text, ok := it.Variant.(*RelayItemDataText)
		if !ok {
			convertItemFormatting(it, &cf, &inverse)
			continue
		}

		item := cf
		item.text = text.Text
		if inverse {
			item.color, item.background = item.background, item.color
		}
		result = append(result, item)
	}
	return result
}

// -----------------------------------------------------------------------------

func bufferPrintLine(lines []bufferLine, index int) {
	// TODO(p): Date change.
	line := &lines[index]

	// TODO(p): Timestamp.
	line.when.Format("\n15:04:05")

	// TODO(p): Rendition.
	prefix := ""

	if line.leaked {
		// TODO(p): Similar as below, but everything greyed out.
		if prefix != "" {
		}
		for _, item := range line.items {
			_ = item
		}
	} else {
		// TODO(p): Render text.
		if prefix != "" {
		}
		for _, item := range line.items {
			_ = item
		}
	}
}

func bufferPrintSeparator() {
	// TODO(p)
	wRichText.Segments = append(wRichText.Segments,
		&widget.TextSegment{Style: widget.RichTextStyleParagraph},
		&widget.SeparatorSegment{})
}

func refreshBuffer(b *buffer) {
	// TODO(p): See xW, rewrite the whole buffer view.
}

func refreshBufferList() {
	// TODO(p): First off, add a buffer list, second, refresh it.
}

func refreshTopic(topic []bufferLineItem) {
	// TODO(p): First off, add a topic, second, refresh it.
}

func refreshStatus() {
	// TODO(p): First off, add a status, second, refresh it.
}

func refreshIcon() {
	// TODO(p): Can we have an icon at all?
}

func refreshPrompt() {
	// TODO(p): First off, add a prompt, second, refresh it.
}

func relayReadMessage(r io.Reader) (m RelayEventMessage, ok bool) {
	var length uint32
	if err := binary.Read(r, binary.BigEndian, &length); err != nil {
		log.Println("Event receive failed: " + err.Error())
		return
	}
	b := make([]byte, length)
	if _, err := io.ReadFull(r, b); err != nil {
		log.Println("Event receive failed: " + err.Error())
		return
	}

	if after, ok2 := m.ConsumeFrom(b); !ok2 {
		log.Println("Event deserialization failed")
		return
	} else if len(after) != 0 {
		log.Println("Event deserialization failed: trailing data")
		return
	}

	if *debug {
		log.Printf("<? %v\n", b)

		j, err := m.MarshalJSON()
		if err != nil {
			log.Println("Event marshalling failed: " + err.Error())
			return
		}

		log.Printf("<- %s\n", j)
	}
	return m, true
}

func relayMakeReceiver(
	ctx context.Context, conn net.Conn) <-chan RelayEventMessage {
	// The usual event message rarely gets above 1 kilobyte,
	// thus this is set to buffer up at most 1 megabyte or so.
	p := make(chan RelayEventMessage, 1000)
	r := bufio.NewReaderSize(conn, 65536)
	go func() {
		defer close(p)
		for {
			m, ok := relayReadMessage(r)
			if !ok {
				return
			}
			select {
			case p <- m:
			case <-ctx.Done():
				return
			}
		}
	}()
	return p
}

func relayWriteMessage(conn net.Conn, data RelayCommandData) bool {
	m := RelayCommandMessage{
		CommandSeq: commandSeq,
		Data:       data,
	}
	commandSeq++

	b, ok := m.AppendTo(make([]byte, 4))
	if !ok {
		log.Println("Command serialization failed")
		return false
	}
	binary.BigEndian.PutUint32(b[:4], uint32(len(b)-4))
	if _, err := conn.Write(b); err != nil {
		log.Println("Command send failed: " + err.Error())
		return false
	}

	if *debug {
		log.Printf("-> %v\n", b)
	}
	return true
}

func convertBufferLine(m *RelayEventDataBufferLine) bufferLine {
	return bufferLine{
		items:         convertItems(m.Items),
		isUnimportant: m.IsUnimportant,
		isHighlight:   m.IsHighlight,
		rendition:     m.Rendition,
		when:          time.UnixMilli(int64(m.When)),
	}
}

func relayProcessBufferLine(b *buffer, m *RelayEventDataBufferLine) {
	line := convertBufferLine(m)

	// Initial sync: skip all other processing, let highlights be.
	bc := bufferByName(bufferCurrent)
	if bc == nil {
		b.lines = append(b.lines, line)
		return
	}

	// Retained mode is complicated.
	display := (!m.IsUnimportant || !bc.hideUnimportant) &&
		(b.bufferName == bufferCurrent || m.LeakToActive)
	toBottom := display            // && bufferAtBottom()
	visible := display && toBottom // && ...
	separate := display &&
		!visible && bc.newMessages == 0 && bc.newUnimportantMessages == 0

	b.lines = append(b.lines, line)
	if !(visible || m.LeakToActive) ||
		b.newMessages != 0 || b.newUnimportantMessages != 0 {
		if line.isUnimportant || m.LeakToActive {
			b.newUnimportantMessages++
		} else {
			b.newMessages++
		}
	}

	if m.LeakToActive {
		leakedLine := line
		leakedLine.leaked = true
		bc.lines = append(bc.lines, leakedLine)

		if !visible || bc.newMessages != 0 || bc.newUnimportantMessages != 0 {
			if line.isUnimportant {
				bc.newUnimportantMessages++
			} else {
				bc.newMessages++
			}
		}
	}

	if separate {
		// TODO(p)
	}
	if display {
		bufferPrintLine(bc.lines, len(bc.lines)-1)
	}
	if toBottom {
		// TODO(p)
	}

	if line.isHighlight || (!visible && !line.isUnimportant &&
		b.kind == RelayBufferKindPrivateMessage) {
		// TODO(p): beep(), probably using https://github.com/ebitengine/oto

		if !visible {
			b.highlighted = true
			refreshIcon()
		}
	}

	refreshBufferList()
}

func relayProcessCallbacks(
	commandSeq uint32, err string, response *RelayResponseData) {
	if handler, ok := commandCallbacks[commandSeq]; !ok {
		// TODO(p): Warn about an unawaited response.
	} else {
		delete(commandCallbacks, commandSeq)
		if handler != nil {
			handler(err, response)
		}
	}

	// We don't particularly care about wraparound issues.
	for cs, handler := range commandCallbacks {
		if cs <= commandSeq {
			delete(commandCallbacks, cs)
			if handler != nil {
				handler("No response", nil)
			}
		}
	}
}

func bufferByName(name string) *buffer {
	for i := range buffers {
		if buffers[i].bufferName == name {
			return &buffers[i]
		}
	}
	return nil
}

func relayProcessMessage(m *RelayEventMessage) {
	switch data := m.Data.Variant.(type) {
	case *RelayEventDataError:
		relayProcessCallbacks(data.CommandSeq, data.Error, nil)
	case *RelayEventDataResponse:
		relayProcessCallbacks(data.CommandSeq, "", &data.Data)

	case *RelayEventDataPing:
		// TODO(p): Send the command.
		_ = RelayCommandData{
			Variant: &RelayCommandDataPingResponse{EventSeq: m.EventSeq},
		}

	case *RelayEventDataBufferLine:
		b := bufferByName(data.BufferName)
		if b == nil {
			return
		}

		relayProcessBufferLine(b, data)
	case *RelayEventDataBufferUpdate:
		b := bufferByName(data.BufferName)
		if b == nil {
			buffers = append(buffers, buffer{})
			b = &buffers[len(buffers)-1]
			refreshBufferList()
		}

		hidingToggled := b.hideUnimportant != data.HideUnimportant
		b.hideUnimportant = data.HideUnimportant
		b.kind = data.Context.Variant.Kind()
		b.serverName = ""
		switch context := data.Context.Variant.(type) {
		case *RelayBufferContextServer:
			b.serverName = context.ServerName
		case *RelayBufferContextChannel:
			b.serverName = context.ServerName
			b.modes = context.Modes
			b.topic = convertItems(context.Topic)
		case *RelayBufferContextPrivateMessage:
			b.serverName = context.ServerName
		}

		if b.bufferName == bufferCurrent {
			refreshTopic(b.topic)
			refreshStatus()

			if hidingToggled {
				refreshBuffer(b)
			}
		}
	case *RelayEventDataBufferStats:
		b := bufferByName(data.BufferName)
		if b == nil {
			return
		}

		b.newMessages = int(data.NewMessages)
		b.newUnimportantMessages = int(data.NewUnimportantMessages)
		b.highlighted = data.Highlighted

		refreshIcon()
	case *RelayEventDataBufferRename:
		b := bufferByName(data.BufferName)
		if b == nil {
			return
		}

		b.bufferName = data.New

		refreshBufferList()
		if data.BufferName == bufferCurrent {
			bufferCurrent = data.New
			refreshStatus()
		}
		if data.BufferName == bufferLast {
			bufferLast = data.New
		}
	case *RelayEventDataBufferRemove:
		buffers = slices.DeleteFunc(buffers, func(b buffer) bool {
			return b.bufferName == data.BufferName
		})

		refreshBufferList()
		refreshIcon()
	case *RelayEventDataBufferActivate:
		// TODO(p): Process all remaining message kinds.
	case *RelayEventDataBufferInput:
		b := bufferByName(data.BufferName)
		if b == nil {
			return
		}

		if b.historyAt == len(b.history) {
			b.historyAt++
		}
		b.history = append(b.history, data.Text)
	case *RelayEventDataBufferClear:
		b := bufferByName(data.BufferName)
		if b == nil {
			return
		}

		b.lines = nil
		if b.bufferName == bufferCurrent {
			refreshBuffer(b)
		}

	case *RelayEventDataServerUpdate:
		s, existed := servers[data.ServerName]
		if !existed {
			s = &server{}
			servers[data.ServerName] = s
		}

		s.state = data.Data.Variant.State()
		switch state := data.Data.Variant.(type) {
		case *RelayServerDataRegistered:
			s.user = state.User
			s.userModes = state.UserModes
		default:
			s.user = ""
			s.userModes = ""
		}

		refreshPrompt()
	case *RelayEventDataServerRename:
		servers[data.New] = servers[data.ServerName]
		delete(servers, data.ServerName)
	case *RelayEventDataServerRemove:
		delete(servers, data.ServerName)
	}
}

func relayRun() {
	// TODO(p): Maybe reset state, and indicate in the UI that we're connecting.

	backendContext, backendCancel = context.WithCancel(context.Background())
	defer backendCancel()

	conn, err := net.Dial("tcp", backendAddress)
	if err != nil {
		log.Println("Connection failed: " + err.Error())
		// TODO(p): Display errors to the user.
		return
	}
	defer conn.Close()

	// TODO(p): How to send messages?
	//  - It would probably make the most sense to have either a chan
	//    (which makes code synchronize), or a locked slice.
	//  - But I also need to wake the sender up somehow,
	//    so maybe use a channel after all.
	//     - Or maybe use a channel just for the signalling.
	//  - Sending (semi-)synchronously is also an option, perhaps.
	// TODO(p): Handle any errors here.
	_ = relayWriteMessage(conn, RelayCommandData{
		Variant: &RelayCommandDataHello{Version: RelayVersion},
	})

	relayMessages := relayMakeReceiver(backendContext, conn)
	for {
		select {
		case m, ok := <-relayMessages:
			if !ok {
				break
			}
			relayProcessMessage(&m)
		default:
			break
		}
	}

	// TODO(p): Indicate in the UI that we're no longer connected.
}

func main() {
	flag.Usage = func() {
		fmt.Fprintf(flag.CommandLine.Output(),
			"Usage: %s [OPTION...] CONNECT\n\n", os.Args[0])
		flag.PrintDefaults()
	}

	flag.Parse()
	if flag.NArg() < 1 || flag.NArg() > 1 {
		flag.Usage()
		os.Exit(1)
	}

	backendAddress = flag.Arg(0)

	a := app.New()
	a.Settings().SetTheme(&customTheme{})
	w := a.NewWindow(projectName)

	// TODO(p): There should also be a widget.NewLabel() next to the entry.
	//  - Probably another Border, even though this seems odd.
	wRichText = widget.NewRichText()
	wRichScroll = container.NewVScroll(wRichText)
	wEntry = widget.NewMultiLineEntry()
	w.SetContent(container.NewBorder(nil, wEntry, nil, nil, wRichScroll))

	testURL, _ := url.Parse("https://x.com")
	wRichText.Segments = []widget.RichTextSegment{
		&widget.ParagraphSegment{Texts: []widget.RichTextSegment{
			&widget.TextSegment{Text: "Test"},
			&widget.HyperlinkSegment{Text: "X", URL: testURL},
			&widget.TextSegment{
				Text:  " is a website, certainly",
				Style: widget.RichTextStyleInline,
			},
		}},
		&widget.TextSegment{Style: widget.RichTextStyleParagraph},
		&widget.SeparatorSegment{},
		&widget.TextSegment{Text: "Paragraph"},
	}
	wRichText.Wrapping = fyne.TextWrapWord
	wRichText.Refresh()

	go relayRun()

	w.ShowAndRun()
}