/*
 * 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.*
import org.apache.ofbiz.base.util.*

surveyResponseId = parameters.surveyResponseId
partyId = null

if (!surveyId && surveyResponseId) {
   surveyResponse = from("SurveyResponse").where("surveyResponseId", surveyResponseId).queryOne()
   surveyId = surveyResponse.surveyId
   context.surveyPartyId = surveyResponse.partyId
   context.surveyId = surveyId
}
surveyWrapper = new SurveyWrapper(delegator, surveyResponseId, partyId, surveyId, null)
surveyWrapper.setEdit(true)

templateUrl = UtilURL.fromOfbizHomePath("applications/content/template/survey/GenericSurvey.ftl")
if (templateUrl) {
    writer = new StringWriter()
    surveyWrapper.render(templateUrl, writer)
    context.surveyString = writer.toString()
}
