fix: minor improvement

This commit is contained in:
Zhicheng Wang 2017-08-16 10:44:00 +08:00
parent 70e9685c4f
commit 255c410d5d

View File

@ -103,13 +103,13 @@ export class DocViewerComponent implements DoCheck, OnDestroy {
private addTitleAndToc(docId: string) {
this.tocService.reset();
let title = '';
const translatedTitleEl = this.hostElement.querySelector('h1[translation-result]');
const originalTitleEl = this.hostElement.querySelector('h1[translation-origin]');
const translatedTitleEl = this.hostElement.querySelector('h1[translation-result]') as HTMLElement;
const originalTitleEl = this.hostElement.querySelector('h1[translation-origin]') as HTMLElement;
const titleEl = translatedTitleEl || originalTitleEl;
// Only create TOC for docs with an <h1> title
// If you don't want a TOC, add "no-toc" class to <h1>
if (titleEl) {
title = (titleEl.textContent).trim();
title = (titleEl.innerText || titleEl.textContent).trim();
if (!/(no-toc|notoc)/i.test(titleEl.className)) {
this.tocService.genToc(this.hostElement, docId);
(originalTitleEl || translatedTitleEl).insertAdjacentHTML('afterend', '<aio-toc class="embedded"></aio-toc>');