Convert DocumentationGenerator to FileGenerator
[mdsal.git] / binding / maven-sal-api-gen-plugin / src / main / java / org / opendaylight / mdsal / binding / yang / wadl / generator / WadlGenerator.java
1 /*
2  * Copyright (c) 2014 Cisco Systems, Inc. and others.  All rights reserved.
3  *
4  * This program and the accompanying materials are made available under the
5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6  * and is available at http://www.eclipse.org/legal/epl-v10.html
7  */
8 package org.opendaylight.mdsal.binding.yang.wadl.generator;
9
10 import com.google.common.collect.ImmutableTable;
11 import com.google.common.collect.Table;
12 import java.util.Set;
13 import org.opendaylight.yangtools.plugin.generator.api.FileGenerator;
14 import org.opendaylight.yangtools.plugin.generator.api.GeneratedFile;
15 import org.opendaylight.yangtools.plugin.generator.api.GeneratedFileLifecycle;
16 import org.opendaylight.yangtools.plugin.generator.api.GeneratedFilePath;
17 import org.opendaylight.yangtools.plugin.generator.api.GeneratedFileType;
18 import org.opendaylight.yangtools.plugin.generator.api.ModuleResourceResolver;
19 import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
20 import org.opendaylight.yangtools.yang.model.api.Module;
21
22 final class WadlGenerator implements FileGenerator {
23     @Override
24     public Table<GeneratedFileType, GeneratedFilePath, GeneratedFile> generateFiles(final EffectiveModelContext context,
25             final Set<Module> localModules, final ModuleResourceResolver moduleResourcePathResolver) {
26         final var result = ImmutableTable.<GeneratedFileType, GeneratedFilePath, GeneratedFile>builder();
27
28         for (Module module : localModules) {
29             final CharSequence body = new WadlTemplate(context, module).body();
30             if (body != null) {
31                 result.put(GeneratedFileType.RESOURCE, GeneratedFilePath.ofPath(module.getName() + ".wadl"),
32                     GeneratedFile.of(GeneratedFileLifecycle.TRANSIENT, body));
33             }
34         }
35         return result.build();
36     }
37 }