lua: Bind after_message_stored and before_mail_accepted (#322)

Signed-off-by: James Hillyerd <james@hillyerd.com>

Signed-off-by: James Hillyerd <james@hillyerd.com>

pkg/extension/luahost/lua_test.go

@@ -1,12 +1,16 @@
 package luahost_test
 
 import (
+	"net/mail"
 	"strings"
 	"testing"
+	"time"
 
 	"github.com/inbucket/inbucket/pkg/extension"
+	"github.com/inbucket/inbucket/pkg/extension/event"
 	"github.com/inbucket/inbucket/pkg/extension/luahost"
 	"github.com/stretchr/testify/require"
+	lua "github.com/yuin/gopher-lua"
 )
 
 func TestEmptyScript(t *testing.T) {
@@ -16,3 +20,96 @@ func TestEmptyScript(t *testing.T) {
 	_, err := luahost.NewFromReader(extHost, strings.NewReader(script), "test.lua")
 	require.NoError(t, err)
 }
+
+func TestAfterMessageStored(t *testing.T) {
+	// Register lua event listener, setup notify channel.
+	script := `
+		local test_ok = true
+
+		function assert_eq(got, want)
+			if got ~= want then
+				-- Incorrect value, schedule test to fail.
+				print("got '" .. got .. "', wanted '" .. want .. "'")
+				test_ok = false
+			end
+		end
+
+		function after_message_stored(msg)
+			assert_eq(msg:mailbox(), "mb1")
+			assert_eq(msg:id(), "id1")
+			assert_eq(msg:subject(), "subj1")
+			assert_eq(msg:size(), 42)
+
+			assert_eq(msg:from():name(), "name1")
+			assert_eq(msg:from():address(), "addr1")
+
+			assert_eq(table.getn(msg:to()), 1)
+			assert_eq(msg:to()[1]:name(), "name2")
+			assert_eq(msg:to()[1]:address(), "addr2")
+
+			assert_eq(msg:date(), 981173106)
+
+			notify:send(test_ok)
+		end
+	`
+	extHost := extension.NewHost()
+	luaHost, err := luahost.NewFromReader(extHost, strings.NewReader(script), "test.lua")
+	require.NoError(t, err)
+	notify := luaHost.CreateChannel("notify")
+
+	// Send event, check channel response is true.
+	msg := &event.MessageMetadata{
+		Mailbox: "mb1",
+		ID:      "id1",
+		From:    &mail.Address{Name: "name1", Address: "addr1"},
+		To:      []*mail.Address{{Name: "name2", Address: "addr2"}},
+		Date:    time.Date(2001, time.February, 3, 4, 5, 6, 0, time.UTC),
+		Subject: "subj1",
+		Size:    42,
+	}
+	go extHost.Events.AfterMessageStored.Emit(msg)
+	assertNotified(t, notify)
+}
+
+func TestBeforeMailAccepted(t *testing.T) {
+	// Register lua event listener.
+	script := `
+		function before_mail_accepted(localpart, domain)
+			return localpart == "from" and domain == "test"
+		end
+	`
+	extHost := extension.NewHost()
+	_, err := luahost.NewFromReader(extHost, strings.NewReader(script), "test.lua")
+	require.NoError(t, err)
+
+	// Send event to be accepted.
+	addr := &event.AddressParts{Local: "from", Domain: "test"}
+	got := extHost.Events.BeforeMailAccepted.Emit(addr)
+	want := true
+	require.NotNil(t, got)
+	if *got != want {
+		t.Errorf("Got %v, wanted %v for addr %v", *got, want, addr)
+	}
+
+	// Send event to be denied.
+	addr = &event.AddressParts{Local: "reject", Domain: "me"}
+	got = extHost.Events.BeforeMailAccepted.Emit(addr)
+	want = false
+	require.NotNil(t, got)
+	if *got != want {
+		t.Errorf("Got %v, wanted %v for addr %v", *got, want, addr)
+	}
+}
+
+func assertNotified(t *testing.T, notify chan lua.LValue) {
+	t.Helper()
+	select {
+	case reslv := <-notify:
+		// Lua function received event.
+		if lua.LVIsFalse(reslv) {
+			t.Error("Lua responsed with false, wanted true")
+		}
+	case <-time.After(2 * time.Second):
+		t.Fatal("Lua did not respond to event within timeout")
+	}
+}