package swextreload

import (
	"context"
	"fmt"
	"log"
	"strings"
	"time"

	"github.com/chromedp/cdproto/target"
	"github.com/chromedp/chromedp"
)

// isDebug controls whether debug printing is enabled.
var isDebug = false

// SetDebugOn turns on debug printing.
func SetDebugOn() {
	isDebug = true
}

// Reload reloads the extensions in extensionIDs.
func Reload(
	url string,
	extensionIDs []string,
	shouldReloadTab bool,
) error {
	for _, extensionID := range extensionIDs {
		return reloadExtension(
			url,
			extensionID,
			shouldReloadTab,
		)
	}

	return nil
}

// reloadExtension reloads the extension extensionID. If shouldReloadTab is
// true, also reload the current tab.
func reloadExtension(
	url string,
	extensionID string,
	shouldReloadTab bool,
) error {
	allocatorContext, cancel := chromedp.NewRemoteAllocator(
		context.Background(),
		url,
	)
	defer cancel()

	ctx, cancel := chromedp.NewContext(allocatorContext)
	defer cancel()

	targets, err := chromedp.Targets(ctx)
	if err != nil {
		return fmt.Errorf("swextreload: can't get targets: %v", err)
	}

	if isDebug {
		log.Printf("Targets: %#v", targets)
	}

	extensionURL := "chrome-extension://" + extensionID + "/"

	var targetID target.ID
	for _, target := range targets {
		if strings.HasPrefix(target.URL, extensionURL) {
			if isDebug {
				log.Printf("Target: %#v", target)
			}

			targetID = target.TargetID
			break
		}
	}

	targetCtx, cancel := chromedp.NewContext(ctx, chromedp.WithTargetID(targetID))
	defer cancel()

	var runtimeResp []byte
	err = chromedp.Run(
		targetCtx,
		chromedp.Evaluate(`chrome.runtime.reload();`, nil),
	)
	if err != nil {
		return fmt.Errorf(
			"swextreload: error reloading extension '%s': %v",
			extensionID,
			err,
		)
	}

	if isDebug {
		log.Printf("Runtime: %v", string(runtimeResp))
	}

	if shouldReloadTab {
		time.Sleep(200 * time.Millisecond)

		targets, err = chromedp.Targets(ctx)
		if err != nil {
			return fmt.Errorf(
				"swextreload: can't get targets for '%s' tab reload: %v",
				extensionID,
				err,
			)
		}

		if isDebug {
			log.Printf("Targets: %#v", targets)
		}

		for _, target := range targets {
			if strings.HasPrefix(target.URL, extensionURL) {
				if isDebug {
					log.Printf("Target: %#v", target)
				}

				targetID = target.TargetID
				break
			}
		}

		targetCtx, cancel = chromedp.NewContext(ctx, chromedp.WithTargetID(targetID))
		defer cancel()

		var tabsResp []byte
		err = chromedp.Run(
			targetCtx,
			chromedp.Evaluate(`chrome.tabs.reload();`, nil),
		)
		if err != nil {
			return fmt.Errorf(
				"swextreload: error reloading tab '%s': %v",
				extensionID,
				err,
			)
		}

		if isDebug {
			log.Printf("Tabs: %v", string(tabsResp))
		}
	}

	return nil
}