/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.
 */

import org.apache.ofbiz.content.survey.SurveyWrapper

if (!survey) {
    surveyResponseId = parameters.surveyResponseId
    if (surveyResponseId) {
        surveyResponse = from("SurveyResponse").where("surveyResponseId", surveyResponseId).queryOne()
        if (surveyResponse) {
            surveyId = surveyResponse.surveyId
            if (surveyId) {
                survey = from("Survey").where("surveyId", surveyId).queryOne()
                context.survey = survey
                context.surveyId = surveyId
            }
        }
    }
}

if (survey) {
    surveyWrapper = new SurveyWrapper(delegator, surveyId)
    context.surveyWrapper = surveyWrapper
    context.sequenceSort = ['sequenceNum']
}
