/*
 * Copyright 2024-2026 the original author or authors.
 *
 * 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
 *
 *      https://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.
 */
package com.alibaba.cloud.ai.graph.node.code.python3;

import com.alibaba.cloud.ai.graph.node.code.TemplateTransformer;
import com.alibaba.cloud.ai.graph.node.code.entity.CodeStyle;

/**
 * @author HeYQ
 * @since 2025-01-06 22:19
 */
public class Python3TemplateTransformer extends TemplateTransformer {

	@Override
	public String getRunnerScript(CodeStyle style) {
		return switch (style) {
			case EXPLICIT_PARAMETERS -> String.format("""
					# declare main function
					%s

					import json
					from base64 import b64decode
					inputs_obj = json.loads(b64decode('%s').decode('utf-8'))
					output_obj = main(**inputs_obj)
					output_json = json.dumps(output_obj, indent=4)
					result = f'''%s{output_json}%s'''
					print(result)
					""", CODE_PLACEHOLDER, INPUTS_PLACEHOLDER, RESULT_TAG, RESULT_TAG);
			case GLOBAL_DICTIONARY -> String.format("""
					import json
					from base64 import b64decode
					params = json.loads(b64decode('%s').decode('utf-8'))

					# declare main function
					%s

					output_obj = main()
					output_json = json.dumps(output_obj, indent=4)
					result = f'''%s{output_json}%s'''
					print(result)
					""", INPUTS_PLACEHOLDER, CODE_PLACEHOLDER, RESULT_TAG, RESULT_TAG);
		};
	}

}
