package ui

import (
	"fmt"
	"go_dreamfactory/cmd/v2/lib/common"
	"go_dreamfactory/cmd/v2/service"
	"go_dreamfactory/cmd/v2/service/observer"
	"net"
	"strings"
	"sync"
	"time"

	"fyne.io/fyne/v2"
	"fyne.io/fyne/v2/container"
	"fyne.io/fyne/v2/theme"
	"fyne.io/fyne/v2/widget"
	"github.com/sirupsen/logrus"
	"github.com/spf13/cast"
)

type appPing struct {
	appAdapter
	resultCh chan int
}

func (this *appPing) LazyInit(ptService service.PttService, obs observer.Observer) error {
	this.tabItem = container.NewTabItemWithIcon(common.TOOLBAR_PING, theme.DownloadIcon(), nil)

	content := container.NewMax()
	content.Objects = []fyne.CanvasObject{}

	targetHost := widget.NewEntry()
	targetHost.PlaceHolder = "目标主机Ip"

	portEntry := widget.NewMultiLineEntry()
	portEntry.Text = "80,3306,6379"
	form := widget.NewForm(
		widget.NewFormItem("主机", targetHost),
		widget.NewFormItem("端口", portEntry),
	)
	// result
	result := widget.NewMultiLineEntry()

	form.OnSubmit = func() {
		result.Text = ""
		if portEntry.Text == "" {
			return
		}
		this.resultCh = make(chan int)
		ports := strings.Split(portEntry.Text, ",")
		this.ping(targetHost.Text, ports)

		var msgs []string

		for p := range this.resultCh {
			for _, v := range ports {
				if cast.ToInt(v) == p {
					msgs = append(msgs, fmt.Sprintf("端口:%d OK", p))
				} else {
					msgs = append(msgs, fmt.Sprintf("端口:%s ERR", v))
				}
				result.Text = strings.Join(msgs, "\n")
				result.Refresh()
			}
		}
	}
	form.Items[1].HintText = "多个端口使用英文,号分隔"

	result.OnChanged = func(s string) {
		result.Refresh()
	}
	content.Objects = append(content.Objects, container.NewBorder(form, nil, nil, nil, result))
	this.tabItem.Content = content
	return nil
}

func (this *appPing) ping(targetHost string, ports []string) {
	var wg sync.WaitGroup
	wg.Add(len(ports))
	for _, port := range ports {
		go func(p int) {
			defer wg.Done()
			_, err := net.DialTimeout("tcp", fmt.Sprintf("%s:%d", targetHost, p), time.Millisecond*500)
			if err == nil {
				this.resultCh <- p
			} else {
				logrus.Error(err)
			}
		}(cast.ToInt(port))
	}

	go func() {
		defer close(this.resultCh)
		wg.Wait()
	}()
}

func (a *appPing) GetAppName() string {
	return common.TOOLBAR_PING
}