a26401efc599ed320e1c805428c6d7448da43d6a
[netconf.git] / netconf / sal-netconf-connector / src / main / java / org / opendaylight / netconf / sal / connect / util / NetconfTopologyRPCProvider.java
1 /*
2  * Copyright (c) 2017 Brocade Communication Systems 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.netconf.sal.connect.util;
9
10 import com.google.common.base.Preconditions;
11 import com.google.common.util.concurrent.CheckedFuture;
12 import com.google.common.util.concurrent.FutureCallback;
13 import com.google.common.util.concurrent.Futures;
14 import com.google.common.util.concurrent.SettableFuture;
15 import java.util.concurrent.Future;
16 import org.opendaylight.aaa.encrypt.AAAEncryptionService;
17 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
18 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
19 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
20 import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.AddNetconfNodeInput;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNode;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNodeBuilder;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNodeTopologyService;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.credentials.credentials.LoginPassword;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.credentials.credentials.LoginPasswordBuilder;
27 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
28 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
29 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TopologyId;
30 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
31 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
32 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
33 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
34 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
35 import org.opendaylight.yangtools.yang.common.RpcResult;
36 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
37 import org.slf4j.Logger;
38 import org.slf4j.LoggerFactory;
39
40 public class NetconfTopologyRPCProvider implements NetconfNodeTopologyService {
41     private final AAAEncryptionService encryptionService;
42     private final DataBroker dataBroker;
43     private final String topologyId;
44     private static final Logger LOG = LoggerFactory.getLogger(NetconfTopologyRPCProvider.class);
45
46     public NetconfTopologyRPCProvider(final DataBroker dataBroker,
47                                       final AAAEncryptionService encryptionService,
48                                       final String topologyId) {
49         this.dataBroker = dataBroker;
50         this.encryptionService = Preconditions.checkNotNull(encryptionService);
51         this.topologyId = Preconditions.checkNotNull(topologyId);
52     }
53
54     @Override
55     public Future<RpcResult<Void>> addNetconfNode(AddNetconfNodeInput input) {
56         NetconfNode node = this.encryptPassword(input);
57         final SettableFuture<RpcResult<Void>> futureResult = SettableFuture.create();
58         NodeId nodeId = new NodeId(input.getNodeId());
59         writeToConfigDS(node, nodeId, topologyId, futureResult);
60         return futureResult;
61     }
62
63     private NetconfNode encryptPassword(AddNetconfNodeInput input) {
64         NetconfNodeBuilder builder = new NetconfNodeBuilder();
65         builder.fieldsFrom(input);
66
67         boolean encrypt = input.isEncrypt();
68         LoginPassword loginPassword = (LoginPassword) input.getCredentials();
69         if (encrypt) {
70             String encryptedPassword = encryptionService.encrypt(loginPassword.getPassword());
71             LoginPassword newCreds = new LoginPasswordBuilder().setPassword(encryptedPassword)
72                     .setUsername(loginPassword.getUsername()).build();
73             builder.setCredentials(newCreds);
74         }
75
76         NetconfNode node = builder.build();
77         return node;
78     }
79
80     private void writeToConfigDS(NetconfNode node, NodeId nodeId, String topologyId,
81                                  final SettableFuture<RpcResult<Void>> futureResult) {
82
83         WriteTransaction writeTransaction = dataBroker.newWriteOnlyTransaction();
84         final InstanceIdentifier<NetworkTopology> networkTopologyId =
85                 InstanceIdentifier.builder(NetworkTopology.class).build();
86         final InstanceIdentifier<NetconfNode> niid = networkTopologyId.child(Topology.class,
87                 new TopologyKey(new TopologyId(topologyId))).child(Node.class,
88                 new NodeKey(nodeId)).augmentation(NetconfNode.class);
89         writeTransaction.merge(LogicalDatastoreType.CONFIGURATION, niid, node, true);
90         final CheckedFuture<Void, TransactionCommitFailedException> future = writeTransaction.submit();
91         Futures.addCallback(future, new FutureCallback<Void>() {
92
93             @Override
94             public void onSuccess(Void result) {
95                 LOG.info("add-netconf-node RPC: Added netconf node successfully.");
96                 futureResult.set(RpcResultBuilder.<Void>success().build());
97             }
98
99             @Override
100             public void onFailure(Throwable exception) {
101                 LOG.error("add-netconf-node RPC: Unable to add netconf node.", exception);
102                 futureResult.setException(exception);
103             }
104         });
105     }
106
107 }