|
|
@@ -1,7 +1,7 @@
|
|
|
-// Author: Matthew Shiel
|
|
|
package main
|
|
|
|
|
|
import (
|
|
|
+ "bytes"
|
|
|
"embed"
|
|
|
"encoding/json"
|
|
|
"fmt"
|
|
|
@@ -12,7 +12,9 @@ import (
|
|
|
"strings"
|
|
|
|
|
|
"git.riomhaire.com/gremlin/elizaservice/eliza"
|
|
|
+ "github.com/gofrs/uuid"
|
|
|
"github.com/jaffee/commandeer"
|
|
|
+
|
|
|
"github.com/labstack/echo/v4"
|
|
|
"github.com/labstack/echo/v4/middleware"
|
|
|
)
|
|
|
@@ -26,11 +28,17 @@ var templates embed.FS
|
|
|
//go:embed bots/*
|
|
|
var bots embed.FS
|
|
|
|
|
|
+/***********************************************************************************************
|
|
|
+*
|
|
|
+ ***********************************************************************************************/
|
|
|
type Main struct {
|
|
|
BotName string `help:"What This Bot is Called."`
|
|
|
Port int `help:"What port bot is listening too"`
|
|
|
}
|
|
|
|
|
|
+/***********************************************************************************************
|
|
|
+*
|
|
|
+ ***********************************************************************************************/
|
|
|
func NewMain() *Main {
|
|
|
return &Main{
|
|
|
BotName: "Eliza",
|
|
|
@@ -38,6 +46,18 @@ func NewMain() *Main {
|
|
|
}
|
|
|
}
|
|
|
|
|
|
+/***********************************************************************************************
|
|
|
+* This at the moment ... but it could be one passed in via LTI if available
|
|
|
+ ***********************************************************************************************/
|
|
|
+func createSession(c echo.Context) string {
|
|
|
+ guid, _ := uuid.NewV4()
|
|
|
+ sessionID := guid.String()
|
|
|
+ return sessionID
|
|
|
+}
|
|
|
+
|
|
|
+/***********************************************************************************************
|
|
|
+*
|
|
|
+ ***********************************************************************************************/
|
|
|
func findPersonality(name string) (personality eliza.Personality, err error) {
|
|
|
filename := "bots/" + name + ".json"
|
|
|
jsonFile, err := bots.Open(filename)
|
|
|
@@ -61,10 +81,14 @@ func findPersonality(name string) (personality eliza.Personality, err error) {
|
|
|
return
|
|
|
}
|
|
|
|
|
|
+/***********************************************************************************************
|
|
|
+*
|
|
|
+ ***********************************************************************************************/
|
|
|
func chantboInteractiontEndpoint(c echo.Context) error {
|
|
|
// Extract question from GET request
|
|
|
question := c.QueryParam("value")
|
|
|
botname := c.QueryParam("bot")
|
|
|
+ sessionID := c.QueryParam("session")
|
|
|
|
|
|
// default to eliza
|
|
|
if len(botname) == 0 {
|
|
|
@@ -73,20 +97,24 @@ func chantboInteractiontEndpoint(c echo.Context) error {
|
|
|
|
|
|
personality, err := findPersonality(strings.ToLower(botname)) // files are in lowercase
|
|
|
if err != nil {
|
|
|
- fmt.Fprintf(c.Response(), "%s", err.Error())
|
|
|
- return nil
|
|
|
+ return c.String(http.StatusOK, err.Error())
|
|
|
}
|
|
|
|
|
|
// Return Eliza's response's so long as the user doesn't give a quit statement
|
|
|
character := eliza.NewBotPersonality(&personality)
|
|
|
answer := character.ReplyTo(question)
|
|
|
|
|
|
+ fmt.Println("Session ID [", sessionID, "] Question [", question, "] Answer [", answer, "]")
|
|
|
+
|
|
|
// Return Eliza's answer
|
|
|
- fmt.Fprintf(c.Response(), "%s", answer)
|
|
|
- return nil
|
|
|
+ return c.String(http.StatusOK, answer)
|
|
|
}
|
|
|
|
|
|
+/***********************************************************************************************
|
|
|
+*
|
|
|
+ ***********************************************************************************************/
|
|
|
func chantbotEndpoint(c echo.Context) error {
|
|
|
+
|
|
|
c.Response().WriteHeader(http.StatusOK)
|
|
|
c.Response().Header().Add("Content-Type", "text/html")
|
|
|
|
|
|
@@ -96,23 +124,36 @@ func chantbotEndpoint(c echo.Context) error {
|
|
|
if err != nil {
|
|
|
log.Fatal(err)
|
|
|
}
|
|
|
- // respond with the output of template execution
|
|
|
- t.Execute(c.Response(), struct {
|
|
|
+
|
|
|
+ sessionID := createSession(c)
|
|
|
+
|
|
|
+ fmt.Println("Session ID [", sessionID, "] Created")
|
|
|
+ var tBuffer bytes.Buffer
|
|
|
+
|
|
|
+ data := struct {
|
|
|
+ SessionID string
|
|
|
Id string
|
|
|
GivenName string
|
|
|
FamilyName string
|
|
|
Name string
|
|
|
Bot string
|
|
|
}{
|
|
|
+ SessionID: sessionID,
|
|
|
Id: c.QueryParam("user_id"),
|
|
|
GivenName: c.QueryParam("given_name"),
|
|
|
FamilyName: c.QueryParam("family_name"),
|
|
|
Name: c.QueryParam("name"),
|
|
|
Bot: strings.Title(strings.ToLower(personality)),
|
|
|
- })
|
|
|
- return nil
|
|
|
+ }
|
|
|
+ // respond with the output of template execution
|
|
|
+ t.Execute(&tBuffer, data)
|
|
|
+ page := tBuffer.String()
|
|
|
+ return c.String(http.StatusOK, page)
|
|
|
}
|
|
|
|
|
|
+/***********************************************************************************************
|
|
|
+*
|
|
|
+ ***********************************************************************************************/
|
|
|
func (m *Main) Run() error {
|
|
|
|
|
|
fmt.Println(`
|
|
|
@@ -127,18 +168,22 @@ func (m *Main) Run() error {
|
|
|
fmt.Println()
|
|
|
|
|
|
e := echo.New()
|
|
|
+ e.Use(middleware.CORS())
|
|
|
|
|
|
var contentHandler = echo.WrapHandler(http.FileServer(http.FS(web)))
|
|
|
var contentRewrite = middleware.Rewrite(map[string]string{"/*": "/web/$1"})
|
|
|
|
|
|
+ e.GET("/*", contentHandler, contentRewrite)
|
|
|
e.GET("/user-input", chantboInteractiontEndpoint)
|
|
|
e.GET("/chatbot/:personality", chantbotEndpoint)
|
|
|
- e.GET("/*", contentHandler, contentRewrite)
|
|
|
e.Logger.Fatal(e.Start(fmt.Sprintf(":%d", m.Port)))
|
|
|
|
|
|
return nil
|
|
|
}
|
|
|
|
|
|
+/***********************************************************************************************
|
|
|
+*
|
|
|
+ ***********************************************************************************************/
|
|
|
func main() {
|
|
|
err := commandeer.Run(NewMain())
|
|
|
if err != nil {
|
gremlin