// Copyright 2014 PDFium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com

#include "xfa/fde/css/fde_csscache.h"

#include <algorithm>

#include "core/fxcrt/include/fx_ext.h"

FDE_CSSCacheItem::FDE_CSSCacheItem(IFDE_CSSStyleSheet* p)
    : pStylesheet(p), dwActivity(0) {
  FXSYS_assert(pStylesheet);
  pStylesheet->AddRef();
}
FDE_CSSCacheItem::~FDE_CSSCacheItem() {
  pStylesheet->Release();
}
IFDE_CSSStyleSheetCache* IFDE_CSSStyleSheetCache::Create() {
  return new CFDE_CSSStyleSheetCache;
}

CFDE_CSSStyleSheetCache::CFDE_CSSStyleSheetCache()
    : m_pFixedStore(NULL), m_iMaxItems(5) {}

CFDE_CSSStyleSheetCache::~CFDE_CSSStyleSheetCache() {
  for (const auto& pair : m_Stylesheets) {
    FXTARGET_DeleteWith(FDE_CSSCacheItem, m_pFixedStore, pair.second);
  }
  m_Stylesheets.clear();
  if (m_pFixedStore) {
    m_pFixedStore->Release();
  }
}
void CFDE_CSSStyleSheetCache::AddStyleSheet(const CFX_ByteStringC& szKey,
                                            IFDE_CSSStyleSheet* pStyleSheet) {
  FXSYS_assert(pStyleSheet != NULL);
  if (m_pFixedStore == NULL) {
    m_pFixedStore =
        FX_CreateAllocator(FX_ALLOCTYPE_Fixed, std::max(10, m_iMaxItems),
                           sizeof(FDE_CSSCacheItem));
    FXSYS_assert(m_pFixedStore != NULL);
  }
  auto it = m_Stylesheets.find(szKey);
  if (it != m_Stylesheets.end()) {
    FDE_CSSCacheItem* pItem = it->second;
    if (pItem->pStylesheet != pStyleSheet) {
      pItem->pStylesheet->Release();
      pItem->pStylesheet = pStyleSheet;
      pItem->pStylesheet->AddRef();
      pItem->dwActivity = 0;
    }
  } else {
    while (static_cast<int32_t>(m_Stylesheets.size()) >= m_iMaxItems) {
      RemoveLowestActivityItem();
    }
    m_Stylesheets[szKey] =
        FXTARGET_NewWith(m_pFixedStore) FDE_CSSCacheItem(pStyleSheet);
  }
}
IFDE_CSSStyleSheet* CFDE_CSSStyleSheetCache::GetStyleSheet(
    const CFX_ByteStringC& szKey) const {
  auto it = m_Stylesheets.find(szKey);
  if (it == m_Stylesheets.end()) {
    return nullptr;
  }
  FDE_CSSCacheItem* pItem = it->second;
  pItem->dwActivity++;
  pItem->pStylesheet->AddRef();
  return pItem->pStylesheet;
}
void CFDE_CSSStyleSheetCache::RemoveStyleSheet(const CFX_ByteStringC& szKey) {
  auto it = m_Stylesheets.find(szKey);
  if (it == m_Stylesheets.end()) {
    return;
  }
  FXTARGET_DeleteWith(FDE_CSSCacheItem, m_pFixedStore, it->second);
  m_Stylesheets.erase(it);
}
void CFDE_CSSStyleSheetCache::RemoveLowestActivityItem() {
  auto found = m_Stylesheets.end();
  for (auto it = m_Stylesheets.begin(); it != m_Stylesheets.end(); ++it) {
    switch (it->first.GetID()) {
      case FXBSTR_ID('#', 'U', 'S', 'E'):
      case FXBSTR_ID('#', 'A', 'G', 'E'):
        continue;
    }
    if (found == m_Stylesheets.end() ||
        it->second->dwActivity > found->second->dwActivity) {
      found = it;
    }
  }
  if (found != m_Stylesheets.end()) {
    FXTARGET_DeleteWith(FDE_CSSCacheItem, m_pFixedStore, found->second);
    m_Stylesheets.erase(found);
  }
}
FDE_CSSTagCache::FDE_CSSTagCache(FDE_CSSTagCache* parent,
                                 IFDE_CSSTagProvider* tag)
    : pTag(tag),
      pParent(parent),
      dwIDHash(0),
      dwTagHash(0),
      iClassIndex(0),
      dwClassHashs(1) {
  FXSYS_assert(pTag != NULL);
  CFX_WideStringC wsValue, wsName = pTag->GetTagName();
  dwTagHash =
      FX_HashCode_String_GetW(wsName.GetPtr(), wsName.GetLength(), TRUE);
  FX_POSITION pos = pTag->GetFirstAttribute();
  while (pos != NULL) {
    pTag->GetNextAttribute(pos, wsName, wsValue);
    FX_DWORD dwNameHash =
        FX_HashCode_String_GetW(wsName.GetPtr(), wsName.GetLength(), TRUE);
    static const FX_DWORD s_dwIDHash = FX_HashCode_String_GetW(L"id", 2, TRUE);
    static const FX_DWORD s_dwClassHash =
        FX_HashCode_String_GetW(L"class", 5, TRUE);
    if (dwNameHash == s_dwClassHash) {
      FX_DWORD dwHash =
          FX_HashCode_String_GetW(wsValue.GetPtr(), wsValue.GetLength());
      dwClassHashs.Add(dwHash);
    } else if (dwNameHash == s_dwIDHash) {
      dwIDHash = FX_HashCode_String_GetW(wsValue.GetPtr(), wsValue.GetLength());
    }
  }
}
FDE_CSSTagCache::FDE_CSSTagCache(const FDE_CSSTagCache& it)
    : pTag(it.pTag),
      pParent(it.pParent),
      dwIDHash(it.dwIDHash),
      dwTagHash(it.dwTagHash),
      iClassIndex(0),
      dwClassHashs(1) {
  if (it.dwClassHashs.GetSize() > 0) {
    dwClassHashs.Copy(it.dwClassHashs);
  }
}
void CFDE_CSSAccelerator::OnEnterTag(IFDE_CSSTagProvider* pTag) {
  FDE_CSSTagCache* pTop = GetTopElement();
  FDE_CSSTagCache item(pTop, pTag);
  m_Stack.Push(item);
}
void CFDE_CSSAccelerator::OnLeaveTag(IFDE_CSSTagProvider* pTag) {
  FXSYS_assert(m_Stack.GetTopElement());
  FXSYS_assert(m_Stack.GetTopElement()->GetTag() == pTag);
  m_Stack.Pop();
}