func getPages(c context.Context) ([]*model.Page, error) {
q := datastore.NewQuery(model.KIND_PAGE)
var pages []*model.Page
if err := ds.ExecuteQuery(c, q, &pages); err != nil {
return nil, errors.WrapOr(err)
}
return pages, nil
}
func getPageProperties(c context.Context, pageKey *datastore.Key) ([]*model.PageProperty, error) {
q := datastore.NewQuery(model.KIND_PAGE_PROPERTY).Ancestor(pageKey)
var props []*model.PageProperty
if err := ds.ExecuteQuery(c, q, &props); err != nil {
return nil, errors.WrapOr(err)
}
return props, nil
}
func savePage(c context.Context, pageKey *datastore.Key, p *model.Page, props map[string]string) error {
err := datastore.RunInTransaction(c, func(c context.Context) error {
var p2 model.Page
if !pageKey.Incomplete() {
if err := ds.Get(c, pageKey, &p2); err != nil {
if errors.Root(err) != datastore.ErrNoSuchEntity {
return errors.WrapOr(err)
} else {
p2 = model.Page{}
p2.Key = pageKey
}
}
}
oldAlias := p2.Alias
p2.Name = p.Name
p2.Alias = p.Alias
p2.Leaf = p.Leaf
aliasChanged := oldAlias != p2.Alias
if err := ds.Put(c, &p2); err != nil {
return errors.WrapOr(err)
}
if aliasChanged && p2.Alias != "" {
newAliasKey := model.NewPageAliasKey(c, p2.Alias)
var pa model.PageAlias
if err := ds.Get(c, newAliasKey, &pa); err == nil {
return ErrPageAliasAlreadyExists
} else {
newAlias := model.PageAlias{}
newAlias.Key = newAliasKey
newAlias.PageKey = pageKey
if err := ds.Put(c, &newAlias); err != nil {
return errors.WrapOr(err)
}
}
}
if aliasChanged && oldAlias != "" {
oldAliasKey := model.NewPageAliasKey(c, oldAlias)
err := ds.Delete(c, oldAliasKey)
if err != nil {
return err
}
}
// TODO: put multi
for name, value := range props {
initial, _ := utf8.DecodeRune([]byte(name))
global := unicode.IsUpper(initial)
var pkey *datastore.Key
if global {
pkey = model.NewGlobalPageKey(c)
} else {
pkey = pageKey
}
propKey := model.NewPagePropertyKey(c, name, pkey)
var prop model.PageProperty
err := ds.Get(c, propKey, &prop)
if err == nil {
prop.Key = propKey
prop.Value = value
}
if err != nil {
if errors.Root(err) != datastore.ErrNoSuchEntity {
return errors.WrapOr(err)
} else {
prop = model.PageProperty{}
prop.Key = propKey
prop.Value = value
}
}
if err = ds.Put(c, &prop); err != nil {
return errors.WrapOr(err)
}
}
return nil
}, &datastore.TransactionOptions{XG: true})
if err != nil {
return err
}
return nil
}
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
}
}