bixi/src/app/shared/components/theme-btn/theme-btn.component.ts

企业级中后台前端解决方案

import { Platform } from '@angular/cdk/platform';
import { ChangeDetectionStrategy, Component, OnDestroy, OnInit, Renderer2 } from '@angular/core';
import { AppService, SiteTheme } from '@core/app.service';

@Component({
  selector: 'theme-btn',
  templateUrl: './theme-btn.component.html',
  host: {
    '[class.theme-btn]': `true`
  },
  changeDetection: ChangeDetectionStrategy.OnPush
})
export class ThemeBtnComponent implements OnInit, OnDestroy {
  private el: HTMLLinkElement;
  theme: SiteTheme = 'default';

  constructor(
    private appService: AppService,
    private renderer: Renderer2,
    private platform: Platform
  ) { }

  ngOnInit(): void {
    if (!this.platform.isBrowser) {
      return;
    }
    this.initTheme();
  }

  private initTheme(): void {
    this.theme = (localStorage.getItem('site-theme') as SiteTheme) || 'default';
    this.onThemeChange(this.theme);
  }

  onThemeChange(theme: SiteTheme): void {
    this.theme = theme;
    this.appService.setTheme(theme);
    this.renderer.setAttribute(document.body, 'data-theme', theme);
    const dom = document.getElementById('site-theme');
    if (dom) {
      dom.remove();
    }
    localStorage.removeItem('site-theme');
    if (theme !== 'default') {
      const el = (this.el = document.createElement('link'));
      el.type = 'text/css';
      el.rel = 'stylesheet';
      el.id = 'site-theme';
      el.href = `assets/${theme}.css`;

      localStorage.setItem('site-theme', theme);
      document.body.append(el);
    }
  }

  ngOnDestroy(): void {
    if (this.el) {
      document.body.removeChild(this.el);
    }
  }
}