1
0
mirror of https://github.com/aclindsa/moneygo.git synced 2024-10-31 16:00:05 -04:00
moneygo/reports.go

86 lines
1.9 KiB
Go
Raw Normal View History

package main
import (
"context"
"github.com/yuin/gopher-lua"
2017-01-27 11:04:39 -05:00
"log"
"net/http"
2017-02-07 20:59:52 -05:00
"path"
"time"
)
//type and value to store user in lua's Context
type key int
2017-01-27 11:04:39 -05:00
const (
userContextKey key = iota
accountsContextKey
securitiesContextKey
2017-01-27 21:50:02 -05:00
balanceContextKey
2017-01-27 11:04:39 -05:00
)
2017-01-27 11:04:39 -05:00
const luaTimeoutSeconds time.Duration = 5 // maximum time a lua request can run for
func ReportHandler(w http.ResponseWriter, r *http.Request) {
user, err := GetUserFromSession(r)
if err != nil {
WriteError(w, 1 /*Not Signed In*/)
return
}
if r.Method == "GET" {
2017-02-07 20:59:52 -05:00
var reportname string
n, err := GetURLPieces(r.URL.Path, "/report/%s", &reportname)
if err != nil || n != 1 {
WriteError(w, 3 /*Invalid Request*/)
return
}
reportpath := path.Join(baseDir, "reports", reportname+".lua")
// Create a new LState without opening the default libs for security
L := lua.NewState(lua.Options{SkipOpenLibs: true})
defer L.Close()
// Create a new context holding the current user with a timeout
ctx := context.WithValue(context.Background(), userContextKey, user)
ctx, cancel := context.WithTimeout(ctx, luaTimeoutSeconds*time.Second)
defer cancel()
L.SetContext(ctx)
for _, pair := range []struct {
n string
f lua.LGFunction
}{
{lua.LoadLibName, lua.OpenPackage}, // Must be first
{lua.BaseLibName, lua.OpenBase},
{lua.TabLibName, lua.OpenTable},
{lua.StringLibName, lua.OpenString},
{lua.MathLibName, lua.OpenMath},
} {
if err := L.CallByParam(lua.P{
Fn: L.NewFunction(pair.f),
NRet: 0,
Protect: true,
}, lua.LString(pair.n)); err != nil {
2017-02-07 20:59:52 -05:00
WriteError(w, 999 /*Internal Error*/)
log.Print(err)
return
}
}
2017-01-27 11:04:39 -05:00
luaRegisterAccounts(L)
luaRegisterSecurities(L)
2017-01-27 21:50:02 -05:00
luaRegisterBalances(L)
2017-01-29 21:09:23 -05:00
luaRegisterDates(L)
2017-01-27 11:04:39 -05:00
2017-02-07 20:59:52 -05:00
err = L.DoFile(reportpath)
if err != nil {
2017-02-07 20:59:52 -05:00
WriteError(w, 3 /*Invalid Request*/)
2017-01-27 11:04:39 -05:00
log.Print("lua:" + err.Error())
2017-02-07 20:59:52 -05:00
return
}
}
}