package levenshtein

import (
	"go.neonxp.ru/extra/math"
)

// Levenshtein возвращает последовательность правок для превращения последовательности элементов s1 в s2
// с учетом стоимостей операций возвращаемых функцией cost
// TODO в алгоритме не предусмотрена экономия памяти ¯\_(ツ)_/¯
func Levenshtein[T comparable](s1, s2 []T, cost EditCost[T]) []Edit {
	len1 := len(s1)
	len2 := len(s2)
	if cost == nil {
		cost = EditCost[T](func(t EditType, s1 *T, s2 *T) int {
			return 1
		})
	}

	dp := make([][]int, len1+1)
	for i := range dp {
		dp[i] = make([]int, len2+1)
	}
	dp[0][0] = 0
	for i := 0; i < len1; i++ {
		for j := 0; j < len2; j++ {
			if i == 0 && j == 0 {
				continue
			}
			switch {
			case i == 0:
				dp[0][j] = dp[0][j-1] + cost(Insert, nil, &s2[j])
			case j == 0:
				dp[i][0] = dp[i-1][0] + cost(Delete, &s1[i], nil)
			default:
				m1 := dp[i-1][j] + cost(Delete, &s1[i], nil)
				m2 := dp[i][j-1] + cost(Insert, nil, &s2[j])
				m3 := dp[i-1][j-1] + cost(Replace, &s1[i], &s2[j])
				dp[i][j] = math.MinScalar(m1, m2, m3)
			}
		}
	}
	i, j := len1-1, len2-1
	edits := []Edit{}
	for i > 0 && j > 0 {
		m1 := dp[i-1][j] + cost(Delete, &s1[i], nil)
		m2 := dp[i][j-1] + cost(Insert, nil, &s2[j])
		m3 := dp[i-1][j-1] + cost(Replace, &s1[i], &s2[j])
		min := math.MinScalar(m1, m2, m3)
		if min == m1 {
			// добавляем удаление символа S1[i]
			edits = append(edits, Edit{
				Type: Delete,
				Idx1: i,
			})
			i--
		}
		if min == m2 {
			// добавляем вставку символа S2[j] после S1[i]
			edits = append(edits, Edit{
				Type: Insert,
				Idx1: i,
				Idx2: j,
			})
			j--
		}
		if min == m3 {
			// добавляем замену S1[i] на S2[j]
			i--
			j--
			if s1[i] != s2[j] {
				edits = append(edits, Edit{
					Type: Replace,
					Idx1: i,
					Idx2: j,
				})
			}
		}
	}
	return edits
}

// EditCost функция возвращающая стоимость действия t над элементами from, to
type EditCost[T comparable] func(t EditType, from *T, to *T) int

// Edit редакторская правка описывающее одно действие над исходной последовательностью
type Edit struct {
	Type EditType // Тип правки: Вставка/Замена/Удаление
	Idx1 int      // Индекс элемента из первой последовательности
	Idx2 int      // Индекс элемента из второй последовательности
}

// EditType тип правки
type EditType int

const (
	Insert  EditType = iota // Вставка
	Replace                 // Замена
	Delete                  // Удаление
)