package geth

/*
#include <stddef.h>
#include <stdbool.h>
extern bool StatusServiceSignalEvent( const char *jsonEvent );
*/
import "C"
import (
	"encoding/json"
	"time"

	"github.com/ethereum/go-ethereum/common"
	"github.com/ethereum/go-ethereum/crypto"
	"github.com/ethereum/go-ethereum/whisper"
)

var (
	whisperFilters []int
)

func onWhisperMessage(message *whisper.Message) {
	event := GethEvent{
		Type: "whisper",
		Event: WhisperMessageEvent{
			Payload: string(message.Payload),
			From:    common.ToHex(crypto.FromECDSAPub(message.Recover())),
			To:      common.ToHex(crypto.FromECDSAPub(message.To)),
			Sent:    message.Sent.Unix(),
			TTL:     int64(message.TTL / time.Second),
			Hash:    common.ToHex(message.Hash.Bytes()),
		},
	}
	body, _ := json.Marshal(&event)
	C.StatusServiceSignalEvent(C.CString(string(body)))
}

func AddWhisperFilter(args whisper.NewFilterArgs) int {
	whisperService, err := GetNodeManager().WhisperService()
	if err != nil {
		return -1
	}

	filter := whisper.Filter{
		To:     crypto.ToECDSAPub(common.FromHex(args.To)),
		From:   crypto.ToECDSAPub(common.FromHex(args.From)),
		Topics: whisper.NewFilterTopics(args.Topics...),
		Fn:     onWhisperMessage,
	}

	id := whisperService.Watch(filter)
	whisperFilters = append(whisperFilters, id)
	return id
}

func RemoveWhisperFilter(idFilter int) {
	whisperService, err := GetNodeManager().WhisperService()
	if err != nil {
		return
	}
	whisperService.Unwatch(idFilter)
}

func ClearWhisperFilters() {
	for _, idFilter := range whisperFilters {
		RemoveWhisperFilter(idFilter)
	}
	whisperFilters = nil
}