/*
 * The MIT License (MIT)
 * Copyright (C) 2024 Huawei Device Co., Ltd.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 */

use napi_derive::napi;

#[napi]
mod xxh3 {
  use napi::bindgen_prelude::{BigInt, Buffer};

  #[napi]
  pub const ALIGNMENT: u32 = 16;

  #[napi(js_name = "xxh3_64")]
  pub fn xxh64(input: Buffer) -> u64 {
    let mut h: u64 = 0;
    for i in input.as_ref() {
      h = h.wrapping_add(*i as u64);
    }
    h
  }

  #[napi]
  /// xxh128 function
  pub fn xxh128(input: Buffer) -> u128 {
    let mut h: u128 = 0;
    for i in input.as_ref() {
      h = h.wrapping_add(*i as u128);
    }
    h
  }

  #[napi]
  /// Xxh3 class
  pub struct Xxh3 {
    inner: BigInt,
  }

  #[napi]
  impl Xxh3 {
    #[napi(constructor)]
    #[allow(clippy::new_without_default)]
    pub fn new() -> Xxh3 {
      Xxh3 {
        inner: BigInt {
          sign_bit: false,
          words: vec![0],
        },
      }
    }

    #[napi]
    /// update
    pub fn update(&mut self, input: Buffer) {
      for i in input.as_ref() {
        self.inner = BigInt {
          sign_bit: false,
          words: vec![self.inner.get_u64().1.wrapping_add(*i as u64)],
        };
      }
    }
    #[napi]
    pub fn digest(&self) -> BigInt {
      self.inner.clone()
    }
  }
}

#[napi]
mod xxh2 {
  use napi::bindgen_prelude::*;

  #[napi]
  pub fn xxh2_plus(a: u32, b: u32) -> u32 {
    a + b
  }

  #[napi]
  pub fn xxh3_xxh64_alias(input: Buffer) -> u64 {
    super::xxh3::xxh64(input)
  }
}

use napi::bindgen_prelude::Buffer;

#[napi]
pub fn xxh64_alias(input: Buffer) -> u64 {
  xxh3::xxh64(input)
}