func putPage(c context.Context, params martini.Params, w http.ResponseWriter, r *http.Request) {
var jp model.JsonPage
if err := getRequestJson(w, r, &jp); err != nil {
handleError(c, w, err, http.StatusBadRequest)
return
}
keyIDStr := params["key"]
if keyIDStr == "" {
handleError(c, w, errors.New("pageID not found."), http.StatusBadRequest)
return
}
intID, err := strconv.ParseInt(keyIDStr, 10, 64)
if err != nil {
handleError(c, w, err, http.StatusBadRequest)
return
}
pageKey := model.NewPageKey(c, intID)
if err := savePage(c, pageKey, jp.Page, jp.Properties); err != nil {
if err == ErrPageAliasAlreadyExists {
handleError(c, w, err, http.StatusConflict)
} else {
handleError(c, w, err, http.StatusInternalServerError)
}
return
}
}
func postPage(params martini.Params, w http.ResponseWriter, r *http.Request) {
c := appengine.NewContext(r)
var jp model.JsonPage
if err := getRequestJson(w, r, &jp); err != nil {
handleError(c, w, err, http.StatusBadRequest)
return
}
keyIDStr := params["key"]
if keyIDStr != "" {
handleError(c, w,
errors.New(fmt.Sprintf("register page did not need pageID=%s", keyIDStr)), http.StatusInternalServerError)
return
}
pageKey, err := model.GeneratePageID(c)
if err != nil {
handleError(c, w, err, http.StatusInternalServerError)
return
}
if err := savePage(c, pageKey, jp.Page, jp.Properties); err != nil {
if err == ErrPageAliasAlreadyExists {
handleError(c, w, err, http.StatusConflict)
} else {
handleError(c, w, err, http.StatusInternalServerError)
}
return
}
}
func queryProperty(params martini.Params, w http.ResponseWriter, r *http.Request) {
c := appengine.NewContext(r)
pn := params["name"]
if pn != "" {
// get property from datastore
c := appengine.NewContext(r)
gpageKey := model.NewGlobalPageKey(c)
gpropKey := model.NewPagePropertyKey(c, pn, gpageKey)
var gpp model.PageProperty
err := ds.Get(c, gpropKey, &gpp)
if err != nil && errors.Root(err) != datastore.ErrNoSuchEntity {
handleError(c, w, err, http.StatusInternalServerError)
return
}
b, err := json.Marshal(gpp)
if err != nil {
handleError(c, w, err, http.StatusInternalServerError)
return
}
responseJson := string(b)
w.Header().Set("Content-Type", "application/json; charset=utf-8")
_, err = w.Write([]byte(responseJson))
if err != nil {
handleError(c, w, err, http.StatusInternalServerError)
return
}
} else {
handleError(c, w, errors.New("error: property name not found."), http.StatusInternalServerError)
return
}
return
}
func putRootPage(c context.Context, w http.ResponseWriter, r *http.Request) {
var rp model.RootPage
if err := getRequestJson(w, r, &rp); err != nil {
handleError(c, w, err, http.StatusBadRequest)
return
}
if rp.PageID == 0 {
handleError(c, w, errors.New("pageID not found."), http.StatusBadRequest)
return
}
rootPageKey := model.NewRootPageKey(c)
err := datastore.RunInTransaction(c, func(c context.Context) error {
var rp2put model.RootPage
if err := ds.Get(c, rootPageKey, &rp2put); err != nil {
if errors.Root(err) != datastore.ErrNoSuchEntity {
return err
} else {
rp2put = model.RootPage{}
rp2put.Key = rootPageKey
}
}
if rp2put.PageID != rp.PageID {
rp2put.PageID = rp.PageID
ds.Put(c, &rp2put)
}
return nil
}, nil)
if err != nil {
handleError(c, w, err, http.StatusInternalServerError)
return
}
}
func getPage(params martini.Params, w http.ResponseWriter, r *http.Request) {
c := appengine.NewContext(r)
if r.Method != "GET" {
handleError(c, w, errors.New("error: illegal access."), http.StatusInternalServerError)
return
}
keyIDStr := params["key"]
if keyIDStr != "" {
// get page from datastore
intID, err := strconv.ParseInt(keyIDStr, 10, 64)
if err != nil {
handleError(c, w, err, http.StatusInternalServerError)
return
}
key := model.NewPageKey(c, intID)
var p model.Page
if err := ds.Get(c, key, &p); err != nil {
handleError(c, w, err, http.StatusInternalServerError)
return
}
props, err := getPageProperties(c, key)
if err != nil {
handleError(c, w, err, http.StatusInternalServerError)
return
}
rpkey := model.NewRootPageKey(c)
var rp model.RootPage
err = ds.Get(c, rpkey, &rp)
if err != nil && errors.Root(err) != datastore.ErrNoSuchEntity {
handleError(c, w, err, http.StatusInternalServerError)
return
}
jsonPage := model.NewJsonPage(&p, props, intID == rp.PageID)
b, err := json.Marshal(jsonPage)
if err != nil {
handleError(c, w, err, http.StatusInternalServerError)
return
}
pageJson := string(b)
w.Header().Set("Content-Type", "application/json; charset=utf-8")
_, err = w.Write([]byte(pageJson))
if err != nil {
handleError(c, w, err, http.StatusInternalServerError)
return
}
} else {
handleError(c, w, errors.New("error: key string not found."), http.StatusInternalServerError)
return
}
return
}
func saveBlocks(params martini.Params, w http.ResponseWriter, r *http.Request) {
c := appengine.NewContext(r)
keyIDStr := params["key"]
if keyIDStr == "" {
handleError(c, w, errors.New("the pageID not found."), http.StatusInternalServerError)
return
}
var b map[string]interface{}
if err := getRequestJson(w, r, &b); err != nil {
handleError(c, w, err, http.StatusBadRequest)
return
}
intID, err := strconv.ParseInt(keyIDStr, 10, 64)
if err != nil {
handleError(c, w, err, http.StatusBadRequest)
return
}
pageKey := model.NewPageKey(c, intID)
var p model.Page
if err := ds.Get(c, pageKey, &p); err != nil {
handleError(c, w, err, http.StatusBadRequest)
return
}
err = datastore.RunInTransaction(c, func(c context.Context) error {
// TODO make async
for id, value := range b {
log.Infof(c, "saving block. id:%s, value:%s", id, value)
r := regexp.MustCompile(`^ctpl_block:(true|false):(\w+):(\d+)$`)
gr := r.FindStringSubmatch(id)
if gr == nil {
return errors.New("illegal block id:" + id)
}
global, err := strconv.ParseBool(gr[1])
if err != nil {
return err
}
areaID := gr[2]
strBlockID := gr[3]
blockID, err := strconv.ParseInt(strBlockID, 10, 64)
if err != nil {
return err
}
var pkey *datastore.Key
if global {
pkey = model.NewGlobalPageKey(c)
} else {
pkey = pageKey
}
akey := model.NewAreaKey(c, areaID, pkey)
bkey := model.NewHTMLBlockKey(c, blockID, akey)
var block model.HTMLBlock
if err := ds.Get(c, bkey, &block); err != nil {
return errors.WrapOr(err)
}
var ok bool
block.Value, ok = value.(string)
if !ok {
return errors.New(
fmt.Sprintf("illegal block value type :%T", value))
}
if err = ds.Put(c, &block); err != nil {
return errors.WrapOr(err)
}
// save backup entity when HTMLBlock saved.
blocks := []*model.HTMLBlock{&block}
backupHTMLBlockFunc.Call(c, uuid.New(), blocks)
}
return nil
}, &datastore.TransactionOptions{XG: true})
if err != nil {
handleError(c, w, err, http.StatusInternalServerError)
return
}
}
"caterpillar/model"
"code.google.com/p/go-uuid/uuid"
"github.com/go-martini/martini"
"github.com/knightso/base/errors"
"github.com/knightso/base/gae/ds"
"golang.org/x/net/context"
"google.golang.org/appengine"
"google.golang.org/appengine/datastore"
"google.golang.org/appengine/delay"
"google.golang.org/appengine/log"
)
var (
ErrPageAliasAlreadyExists = errors.New("caterpillar: Page alias already exists")
)
func queryLeaves(w http.ResponseWriter, r *http.Request) {
c := appengine.NewContext(r)
// TODO: cannot marshall array?
responseJson := "["
index := 1
for _, value := range leaves {
if index != 1 {
responseJson += ","
}
b, err := json.Marshal(*value)
if err != nil {
func renderPage(id string, u *user.User, c context.Context, w http.ResponseWriter, r *http.Request, edit bool) {
var pageKey *datastore.Key
pID, err := strconv.ParseInt(id, 10, 64)
if err != nil {
// check alias
paKey := model.NewPageAliasKey(c, id)
var pa model.PageAlias
if err := ds.Get(c, paKey, &pa); err != nil {
handleError(c, w, err, http.StatusNotFound)
return
}
pID = pa.PageKey.IntID()
pageKey = pa.PageKey
} else {
pageKey = model.NewPageKey(c, pID)
}
var p model.Page
if err := ds.Get(c, pageKey, &p); err != nil {
handleError(c, w, err, http.StatusNotFound)
return
}
if edit {
if u == nil {
loginURL, _ := user.LoginURL(c, p.URLBase()+common.EDIT_PAGE_EXT)
http.Redirect(w, r, loginURL, http.StatusFound)
return
} else if !u.Admin {
// TODO: prepare error page
http.Redirect(w, r, "/caterpillar/error/invalidUser", http.StatusFound)
return
}
}
leaf := leaves[p.Leaf]
if leaf == nil {
errmsg := fmt.Sprintf("leaf not found:" + p.Leaf)
handleError(c, w, errors.New(errmsg), http.StatusNotFound)
return
}
tparam := struct {
Properties map[string]interface{}
Areas map[string]interface{}
User *user.User
Edit bool
PageID int64
ViewURL string
EditURL string
PagesURL string
LogoutURL string
}{
make(map[string]interface{}),
make(map[string]interface{}),
u,
edit,
pID,
"",
"",
"",
"",
}
if u != nil {
tparam.PagesURL = "/caterpillar/static/mng/#/queryPage"
purl := p.URLBase()
tparam.ViewURL = purl + common.VIEW_PAGE_EXT
tparam.EditURL = purl + common.EDIT_PAGE_EXT
logoutURL, err := user.LogoutURL(c, tparam.ViewURL)
if err != nil {
// only log
applog.Warningf(c, "cannot get logoutURL. err:%v", err)
}
tparam.LogoutURL = logoutURL
}
futureProps := make(map[string]<-chan func() (*model.PageProperty, error))
futureAreas := make(map[string]<-chan func() (*model.Area, []model.Block, error))
for _, hole := range leaf.Wormholes {
var pkey *datastore.Key
if hole.Global {
pkey = model.NewGlobalPageKey(c)
} else {
pkey = pageKey
}
switch hole.Type {
case PROPERTY:
propkey := model.NewPagePropertyKey(c, hole.Name, pkey)
ch := getPagePropertyAsync(c, propkey)
futureProps[hole.Name] = ch
case AREA:
ch := getAreaAndBlocksAsync(c, pkey, hole.Name)
futureAreas[hole.Name] = ch
}
}
pageURLs := make(map[string]string)
for _, hole := range leaf.Wormholes {
switch hole.Type {
case PROPERTY:
prop, err := (<-futureProps[hole.Name])()
if err == nil {
tparam.Properties[hole.Name] = prop.Value
} else {
// TODO handle error
applog.Errorf(c, "%s", err)
}
case AREA:
area, blocks, err := (<-futureAreas[hole.Name])()
if err == nil {
areasrc := renderArea(hole.Global, area, blocks, edit)
if !edit {
futurePages := make(map[string]<-chan func() (*model.Page, error))
urls := pageUrlRegex.FindAllStringSubmatch(areasrc, -1)
for _, url := range urls {
purl := url[0]
if _, exists := pageURLs[purl]; exists {
continue
}
if _, exists := futurePages[purl]; exists {
continue
}
pageID, err := strconv.ParseInt(url[1], 10, 64)
if err != nil {
// TODO handle error
applog.Errorf(c, "%s", err)
continue
}
pkey := model.NewPageKey(c, pageID)
futurePages[purl] = getPageAsync(c, pkey)
}
for purl, ch := range futurePages {
page, err := (<-ch)()
if err != nil {
// TODO handle error
applog.Errorf(c, "%s", err)
continue
}
pageURLs[purl] = page.URLBase() + common.VIEW_PAGE_EXT
}
areasrc = pageUrlRegex.ReplaceAllStringFunc(areasrc, func(s string) string {
if r, true := pageURLs[s]; true {
return r
}
return s
})
}
tparam.Areas[hole.Name] = template.HTML(areasrc)
} else {
// TODO handle error
applog.Errorf(c, "%s", err)
}
}
}
// TODO: validate reserved page name property
// or put some prefix?
tparam.Properties["name"] = p.Name
// TODO: resolve charset from somewhere
w.Header().Set("Content-Type", "text/html; charset=utf-8")
if err = leaf.Template.Execute(w, tparam); err != nil {
handleError(c, w, err, http.StatusInternalServerError)
return
}
return
}
