b7174fa5b14ad5bed17f543cc56f17ff69072ce8
[yangtools.git] / yang / yang-parser-impl / src / main / java / org / opendaylight / yangtools / yang / parser / repo / SemVerDependencyResolver.java
1 /*
2  * Copyright (c) 2016 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.yangtools.yang.parser.repo;
9
10 import com.google.common.base.Optional;
11 import java.util.Collection;
12 import java.util.Map;
13 import org.opendaylight.yangtools.concepts.SemVer;
14 import org.opendaylight.yangtools.yang.common.QName;
15 import org.opendaylight.yangtools.yang.model.api.ModuleImport;
16 import org.opendaylight.yangtools.yang.model.repo.api.SemVerSourceIdentifier;
17 import org.opendaylight.yangtools.yang.model.repo.api.SourceIdentifier;
18 import org.opendaylight.yangtools.yang.parser.impl.util.YangModelDependencyInfo;
19
20 final class SemVerDependencyResolver extends DependencyResolver {
21
22     protected SemVerDependencyResolver(final Map<SourceIdentifier, YangModelDependencyInfo> depInfo) {
23         super(depInfo);
24     }
25
26     protected static SourceIdentifier findCompatibleVersion(final Iterable<SourceIdentifier> haystack,
27             final ModuleImport mi) {
28         final String requestedModuleName = mi.getModuleName();
29         for (SourceIdentifier r : haystack) {
30             if (requestedModuleName.equals(r.getName())
31                     && isCompatible(((SemVerSourceIdentifier) r).getSemanticVersion(), mi.getSemanticVersion())) {
32                 return r;
33             }
34         }
35
36         return null;
37     }
38
39     private static boolean isCompatible(final SemVer moduleSemVer, final SemVer importSemVer) {
40         return moduleSemVer.getMajor() == importSemVer.getMajor() && moduleSemVer.compareTo(importSemVer) >= 0;
41     }
42
43     @Override
44     protected boolean isKnown(final Collection<SourceIdentifier> haystack, final ModuleImport mi) {
45         final String rev = mi.getRevision() != null ? QName.formattedRevision(mi.getRevision()) : null;
46         final SemVerSourceIdentifier msi = SemVerSourceIdentifier.create(mi.getModuleName(), Optional.fromNullable(rev),
47             mi.getSemanticVersion());
48
49         // Quick lookup
50         if (haystack.contains(msi)) {
51             return true;
52         }
53
54         // Slow revision-less walk
55         return findCompatibleVersion(haystack, mi) != null;
56     }
57
58     public static SemVerDependencyResolver create(final Map<SourceIdentifier, YangModelDependencyInfo> depInfo) {
59         return new SemVerDependencyResolver(depInfo);
60     }
61 }