smithy4s.meta.ScalaImportsTrait Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of smithy4s-protocol Show documentation
protocol
There is a newer version: 0.19.0-41-91762fb
/*
 *  Copyright 2021-2024 Disney Streaming
 *
 *  Licensed under the Tomorrow Open Source Technology License, Version 1.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://disneystreaming.github.io/TOST-1.0.txt
 *
 *  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 smithy4s.meta;

import java.util.List;
import java.util.stream.Collectors;
import software.amazon.smithy.model.SourceException;
import software.amazon.smithy.model.node.Node;
import software.amazon.smithy.model.node.ArrayNode;
import software.amazon.smithy.model.shapes.ShapeId;
import software.amazon.smithy.model.traits.AbstractTrait;
import software.amazon.smithy.model.traits.AbstractTraitBuilder;
import software.amazon.smithy.model.traits.TraitService;
import software.amazon.smithy.utils.SmithyBuilder;
import software.amazon.smithy.utils.ToSmithyBuilder;

public final class ScalaImportsTrait extends AbstractTrait implements ToSmithyBuilder {


	public static final ShapeId ID = ShapeId.from("smithy4s.meta#scalaImports");

	private final List imports;

	private ScalaImportsTrait(ScalaImportsTrait.Builder builder) {
		super(ID, builder.getSourceLocation());
		this.imports = builder.imports;

		if (this.imports == null) {
			throw new SourceException("imports must be provided.", getSourceLocation());
		}
	}

	public List getImports() {
		return this.imports;
	}

	@Override
	protected Node createNode() {
		ArrayNode.Builder builder = ArrayNode.builder();
    getImports().forEach(s -> builder.withValue(s));
		return builder.build();
	}

	@Override
	public SmithyBuilder toBuilder() {
		return builder().imports(imports).sourceLocation(getSourceLocation());
	}

	/**
	 * @return Returns a new RefinedTrait builder.
	 */
	public static ScalaImportsTrait.Builder builder() {
		return new Builder();
	}

	public static final class Builder extends AbstractTraitBuilder {

		private List imports;

		public ScalaImportsTrait.Builder imports(List imports) {
			this.imports = imports;
			return this;
		}

		@Override
		public ScalaImportsTrait build() {
			return new ScalaImportsTrait(this);
		}
	}

	public static final class Provider implements TraitService {

		@Override
		public ShapeId getShapeId() {
			return ID;
		}

		@Override
		public ScalaImportsTrait createTrait(ShapeId target, Node value) {
			ArrayNode arrayNode = value.expectArrayNode();
			List imports = arrayNode.getElements().stream().map(node -> node.expectStringNode().getValue()).collect(Collectors.toList());
			return builder().sourceLocation(value).imports(imports).build();
		}

	}
}