package lmdb

import (
	"encoding/binary"
	"fmt"
	"log"

	"fiatjaf.com/nostr"
	"fiatjaf.com/nostr/eventstore/codec/betterbinary"
	"github.com/PowerDNS/lmdb-go/lmdb"
)

const (
	DB_VERSION byte = 'v'
)

const target = 2

func (b *LMDBBackend) migrate() error {
	return b.lmdbEnv.Update(func(txn *lmdb.Txn) error {
		val, err := txn.Get(b.settingsStore, []byte("version"))
		if err != nil && !lmdb.IsNotFound(err) {
			return fmt.Errorf("failed to get db version: %w", err)
		}

		var version uint16 = target
		if err == nil {
			version = binary.BigEndian.Uint16(val)
		}

		// do the migrations in increasing steps (there is no rollback)
		if version < target {
			log.Printf("[lmdb] migration %d: reindex everything\n", target)

			if err := txn.Drop(b.indexId, false); err != nil {
				return err
			}
			if err := txn.Drop(b.indexKind, false); err != nil {
				return err
			}
			if err := txn.Drop(b.indexPubkey, false); err != nil {
				return err
			}
			if err := txn.Drop(b.indexPubkeyKind, false); err != nil {
				return err
			}
			if err := txn.Drop(b.indexTag, false); err != nil {
				return err
			}
			if err := txn.Drop(b.indexTag32, false); err != nil {
				return err
			}
			if err := txn.Drop(b.indexTagAddr, false); err != nil {
				return err
			}
			if err := txn.Drop(b.indexPTagKind, false); err != nil {
				return err
			}

			cursor, err := txn.OpenCursor(b.rawEventStore)
			if err != nil {
				return fmt.Errorf("failed to open cursor in migration %d: %w", target, err)
			}
			defer cursor.Close()

			var idx, val []byte
			var evt nostr.Event

			for {
				idx, val, err = cursor.Get(nil, nil, lmdb.Next)
				if lmdb.IsNotFound(err) {
					break
				}
				if err != nil {
					return fmt.Errorf("failed to get next in migration %d: %w", target, err)
				}

				if err := betterbinary.Unmarshal(val, &evt); err != nil {
					log.Printf("failed to unmarshal event %x, skipping: %s", idx, err)
					continue
				}

				for key := range b.getIndexKeysForEvent(evt) {
					if err := txn.Put(key.dbi, key.key, idx, 0); err != nil {
						return fmt.Errorf("failed to save index %s for event %s (%v) on migration %d: %w",
							b.keyName(key), evt.ID, idx, target, err)
					}
				}
			}

			// bump version
			if err := b.setVersion(txn, target); err != nil {
				return err
			}
		}

		return nil
	})
}

func (b *LMDBBackend) setVersion(txn *lmdb.Txn, v uint16) error {
	var newVersion [2]byte
	binary.BigEndian.PutUint16(newVersion[:], v)
	return txn.Put(b.settingsStore, []byte("version"), newVersion[:], 0)
}