#include "clang/ExtractAPI/AvailabilityInfo.h"

#include "clang/AST/Attr.h"

#include "llvm/ADT/STLExtras.h"

using namespace clang;

using namespace extractapi;

AvailabilitySet::AvailabilitySet(const Decl *Decl) {

  // Collect availability attributes from all redeclrations.

  for (const auto *RD : Decl->redecls()) {

    if (const auto *A = RD->getAttr<UnavailableAttr>()) {

      if (!A->isImplicit()) {

        this->Availabilities.clear();

        UnconditionallyUnavailable = true;

      }

    }

    if (const auto *A = RD->getAttr<DeprecatedAttr>()) {

      if (!A->isImplicit()) {

        this->Availabilities.clear();

        UnconditionallyDeprecated = true;

      }

    }

    for (const auto *Attr : RD->specific_attrs<AvailabilityAttr>()) {

      StringRef Domain = Attr->getPlatform()->getName();

      auto *Availability =

          llvm::find_if(Availabilities, [Domain](const AvailabilityInfo &Info) {

            return Domain.equals(Info.Domain);

          });

      if (Availability != Availabilities.end()) {

        // Get the highest introduced version for all redeclarations.

        if (Availability->Introduced < Attr->getIntroduced())

          Availability->Introduced = Attr->getIntroduced();

        // Get the lowest deprecated version for all redeclarations.

        if (Availability->Deprecated > Attr->getDeprecated())

          Availability->Deprecated = Attr->getDeprecated();

        // Get the lowest obsoleted version for all redeclarations.

        if (Availability->Obsoleted > Attr->getObsoleted())

          Availability->Obsoleted = Attr->getObsoleted();

      } else {

        Availabilities.emplace_back(Domain, Attr->getIntroduced(),

                                    Attr->getDeprecated(), Attr->getObsoleted(),

                                    Attr->getUnavailable());

      }

    }

  }

}