import { h, Component } from 'preact';
import * as prettyBytes from 'pretty-bytes';

type FileContents = ArrayBuffer | Blob;

interface Props extends Pick<JSX.HTMLAttributes, Exclude<keyof JSX.HTMLAttributes, 'data'>> {
  file?: FileContents;
  compareTo?: FileContents;
  increaseClass?: string;
  decreaseClass?: string;
}

interface State {
  size?: number;
  sizeFormatted?: string;
  compareSize?: number;
  compareSizeFormatted?: string;
}

function calculateSize(data: FileContents): number {
  return data instanceof ArrayBuffer ? data.byteLength : data.size;
}

export default class FileSize extends Component<Props, State> {
  constructor(props: Props) {
    super(props);
    if (props.file) {
      this.computeSize('size', props.file);
    }
    if (props.compareTo) {
      this.computeSize('compareSize', props.compareTo);
    }
  }

  componentWillReceiveProps({ file, compareTo }: Props) {
    if (file !== this.props.file) {
      this.computeSize('size', file);
    }
    if (compareTo !== this.props.compareTo) {
      this.computeSize('compareSize', compareTo);
    }
  }

  componentDidMount() {
    this.applyStyles();
  }

  componentDidUpdate() {
    this.applyStyles();
  }

  applyStyles() {
    const { size, compareSize = 0 } = this.state;
    if (size != null && this.base) {
      const delta = size && compareSize ? (size - compareSize) / compareSize : 0;
      this.base.style.setProperty('--size', '' + size);
      this.base.style.setProperty('--size-delta', '' + Math.round(Math.abs(delta * 100)));
    }
  }

  computeSize(prop: keyof State, data?: FileContents) {
    const size = data ? calculateSize(data) : 0;
    const pretty = prettyBytes(size);
    this.setState({
      [prop]: size,
      [prop + 'Formatted']: pretty,
    });
  }

  render(
    { file, compareTo, increaseClass, decreaseClass, ...props }: Props,
    { size, sizeFormatted = '', compareSize }: State,
  ) {
    const delta = size && compareSize ? (size - compareSize) / compareSize : 0;
    return (
      <span {...props}>
        {sizeFormatted}
        {compareTo && (
          <span class={delta > 0 ? increaseClass : decreaseClass}>
            {delta > 0 && '+'}
            {Math.round(delta * 100)}%
          </span>
        )}
      </span>
    );
  }
}