SystemImplicitVariableProvider.java

package act.view;

/*-
 * #%L
 * ACT Framework
 * %%
 * Copyright (C) 2014 - 2017 ActFramework
 * %%
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * #L%
 */

import act.app.ActionContext;
import act.app.App;
import act.conf.AppConfig;
import act.mail.MailerContext;
import act.security.CSRF;
import act.util.ActContext;
import org.osgl.http.H;
import org.osgl.util.C;
import org.rythmengine.utils.RawData;
import org.rythmengine.utils.S;

import javax.mail.internet.InternetAddress;
import java.util.List;
import java.util.Locale;
import java.util.Map;

/**
 * Define system implicit variables
 */
public class SystemImplicitVariableProvider extends ImplicitVariableProvider {
    @Override
    public List<ActionViewVarDef> implicitActionViewVariables() {
        return actionViewVarDefs;
    }

    @Override
    public List<MailerViewVarDef> implicitMailerViewVariables() {
        return mailerViewVarDefs;
    }

    private List<ActionViewVarDef> actionViewVarDefs = C.listOf(
            new ActionViewVarDef("_app", App.class) {
                @Override
                public Object eval(ActionContext context) {
                    return context.app();
                }
            },
            new ActionViewVarDef("_conf", AppConfig.class) {
                @Override
                public Object eval(ActionContext context) {
                    return context.config();
                }
            },
            new ActionViewVarDef("_ctx", ActContext.class) {
                @Override
                public Object eval(ActionContext context) {
                    return context;
                }
            },
            new ActionViewVarDef("_action", ActionContext.class) {
                @Override
                public Object eval(ActionContext context) {
                    return context;
                }
            },
            new ActionViewVarDef("_req", H.Request.class) {
                @Override
                public Object eval(ActionContext context) {
                    return context.req();
                }
            },
            new ActionViewVarDef("_resp", H.Response.class) {
                @Override
                public Object eval(ActionContext context) {
                    return context.resp();
                }
            },
            new ActionViewVarDef("_mailer", MailerContext.class) {
                @Override
                public Object eval(ActionContext context) {
                    return null;
                }
            },
            new ActionViewVarDef("_session", H.Session.class) {
                @Override
                public Object eval(ActionContext context) {
                    return context.session();
                }
            },
            new ActionViewVarDef("_flash", H.Flash.class) {
                @Override
                public Object eval(ActionContext context) {
                    return context.flash();
                }
            },
            new ActionViewVarDef("_params", Map.class) {
                @Override
                public Object eval(ActionContext context) {
                    return context.allParams();
                }
            },
            new ActionViewVarDef("_csrf", String.class) {
                @Override
                public Object eval(ActionContext context) {
                    return CSRF.token(context);
                }
            },
            new ActionViewVarDef("_csrfField", RawData.class) {
                @Override
                public Object eval(ActionContext context) {
                    return S.raw(CSRF.formField(context));
                }
            },
            new ActionViewVarDef("_dspToken", String.class) {
                @Override
                public Object eval(ActionContext context) {
                    return dspToken(context);
                }
            },
            new ActionViewVarDef("_dspField", RawData.class) {
                @Override
                public Object eval(ActionContext context) {
                    String dsp = dspToken(context);
                    return S.raw(new StringBuilder("<a type='hidden' name='").append(context.config().dspToken())
                            .append("' value='").append(dsp).append("'>"));
                }
            },
            new ActionViewVarDef("_lang", String.class) {
                @Override
                public Object eval(ActionContext context) {
                    Locale locale = context.locale(true);
                    return locale.toString().replace('_', '-');
                }
            }
    );

    private String dspToken(ActionContext context) {
        String dsp = context.attribute("__dsp__");
        if (null == dsp) {
            dsp = S.random(6);
            context.attribute("__dsp__", dsp);
        }
        return dsp;
    }

    private List<MailerViewVarDef> mailerViewVarDefs = C.listOf(
            new MailerViewVarDef("_app", App.class) {
                @Override
                public Object eval(MailerContext context) {
                    return context.app();
                }
            },
            new MailerViewVarDef("_conf", AppConfig.class) {
                @Override
                public Object eval(MailerContext context) {
                    return context.config();
                }
            },
            new MailerViewVarDef("_ctx", ActContext.class) {
                @Override
                public Object eval(MailerContext context) {
                    return context;
                }
            },
            new MailerViewVarDef("_mailer", MailerContext.class) {
                @Override
                public Object eval(MailerContext context) {
                    return context;
                }
            },
            new MailerViewVarDef("_action", ActionContext.class) {
                @Override
                public Object eval(MailerContext context) {
                    return null;
                }
            },
            new MailerViewVarDef("_from", InternetAddress.class) {
                @Override
                public Object eval(MailerContext context) {
                    return context.from();
                }
            },
            new MailerViewVarDef("_to", List.class) {
                @Override
                public Object eval(MailerContext context) {
                    return context.to();
                }
            },
            new MailerViewVarDef("_cc", List.class) {
                @Override
                public Object eval(MailerContext context) {
                    return context.cc();
                }
            },
            new MailerViewVarDef("_bcc", List.class) {
                @Override
                public Object eval(MailerContext context) {
                    return context.bcc();
                }
            }
    );
}