1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
/*
* Copyright 2016-2017 Nokia Corporation
*
* 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
*
* 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.
*/
package com.nokia.cbam.swagger;
import com.google.common.collect.Sets;
import io.swagger.codegen.*;
import io.swagger.models.Model;
import io.swagger.models.ModelImpl;
import java.io.File;
import java.util.*;
import io.swagger.codegen.languages.JavaClientCodegen;
import static com.google.common.collect.Sets.newHashSet;
public class CbamJavaClientCodegen extends JavaClientCodegen {
public CbamJavaClientCodegen() {
supportsInheritance = false;
for (CliOption cliOption : this.cliOptions) {
if(cliOption.getType().equals("useRxJava")){
cliOption.setDefault("true");
}
System.err.println(cliOption.getOpt() + cliOption.getDefault());
}
}
@Override
public CodegenModel fromModel(String name, Model model, Map<String, Model> allDefinitions) {
CodegenModel m = super.fromModel(name, model, allDefinitions);
//remove the fields inherited from parent objects
remove(m, newHashSet("OPENSTACK_V3_INFO", "OPENSTACK_V2_INFO", "OTHER_VIM_INFO", "VMWARE_VCLOUD_INFO"), newHashSet("id"));
remove(m, newHashSet("OtherNotification", "VnfIdentifierCreationNotification", "VnfIdentifierDeletionNotification", "VnfInfoAttributeValueChangeNotification", "VnfLifecycleChangeNotification"), newHashSet("notificationType", "subscriptionId", "timestamp", "vnfInstanceId"));
if(model instanceof ModelImpl && (!m.isEnum) && !m.hasVars) {
typeMapping.put(m.classname, typeMapping.get(((ModelImpl)model).getType()));
}
return m;
}
public void remove(CodegenModel m, Set<String> classNames, Set<String> fieldNames){
System.err.println("**** Processing " + m.name);
if(classNames.contains(m.name)){
System.err.println("**** Processing " + m.name);
Iterator<CodegenProperty> iterator = m.vars.iterator();
while(iterator.hasNext()){
CodegenProperty prop = iterator.next();
if(fieldNames.contains(prop.name)){
System.err.println("**** Removing: " + prop.name + " from " + m.name);
iterator.remove();
}
System.err.println("********* Keeping " + prop.name + " from " + m.name);
}
}
}
@Override
public Map<String, Object> postProcessAllModels(Map<String, Object> objs) {
Map<String, Object> m = super.postProcessAllModels(objs);
for(Map.Entry<String, Object> entry: m.entrySet()) {
CodegenModel cm = ((HashMap<String, CodegenModel>)((List)((HashMap)entry.getValue()).get("models")).get(0)).get("model");
for (CodegenProperty var : cm.vars) {
if (typeMapping.containsKey(var.datatype)) {
var.dataFormat = var.datatypeWithEnum = typeMapping.get(var.datatype);
}
}
}
return m;
}
}
|