// Licensed to the Apache Software Foundation (ASF) under one

// or more contributor license agreements.  See the NOTICE file

// distributed with this work for additional information

// regarding copyright ownership.  The ASF licenses this file

// to you under the Apache License, Version 2.0 (the

// "License"); you may not use this file except in compliance

// with the License.  You may obtain a copy of the License at

//

//   http://www.apache.org/licenses/LICENSE-2.0

//

// Unless required by applicable law or agreed to in writing,

// software distributed under the License is distributed on an

// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY

// KIND, either express or implied.  See the License for the

// specific language governing permissions and limitations

// under the License.

#pragma once

#include <fmt/compile.h>

#include <cstdint>

#include <cstring>

#include <iostream>

#include <string>

namespace doris {

// 24bit int type, used to store date type in storage

struct uint24_t {

public:

    uint24_t() = default;

    uint24_t(const uint32_t& value) {

        data[0] = static_cast<uint8_t>(value);

        data[1] = static_cast<uint8_t>(value >> 8);

        data[2] = static_cast<uint8_t>(value >> 16);

    }

    uint24_t& operator=(const uint32_t value) {

        data[0] = static_cast<uint8_t>(value);

        data[1] = static_cast<uint8_t>(value >> 8);

        data[2] = static_cast<uint8_t>(value >> 16);

        return *this;

    }

    uint24_t& operator=(const unsigned __int128& value) {

        data[0] = static_cast<uint8_t>(value);

        data[1] = static_cast<uint8_t>(value >> 8);

        data[2] = static_cast<uint8_t>(value >> 16);

        return *this;

    }

    uint24_t& operator=(const uint64_t value) {

        data[0] = static_cast<uint8_t>(value);

        data[1] = static_cast<uint8_t>(value >> 8);

        data[2] = static_cast<uint8_t>(value >> 16);

        return *this;

    }

    uint24_t& operator+=(const uint24_t& value) {

        *this = static_cast<int>(*this) + static_cast<int>(value);

        return *this;

    }

    uint24_t& operator>>=(const int bits) {

        *this = static_cast<unsigned int>(*this) >> bits;

        return *this;

    }

    uint24_t& operator|=(const uint24_t& value) {

        *this = static_cast<int>(*this) | static_cast<int>(value);

        return *this;

    }

    operator uint32_t() const {

        uint32_t value = static_cast<uint8_t>(data[0]);

        value += (static_cast<uint32_t>(static_cast<uint8_t>(data[1]))) << 8;

        value += (static_cast<uint32_t>(static_cast<uint8_t>(data[2]))) << 16;

        return value;

    }

    uint24_t& operator=(const int value) {

        data[0] = static_cast<uint8_t>(value);

        data[1] = static_cast<uint8_t>(value >> 8);

        data[2] = static_cast<uint8_t>(value >> 16);

        return *this;

    }

    uint24_t& operator=(const int64_t value) {

        data[0] = static_cast<uint8_t>(value);

        data[1] = static_cast<uint8_t>(value >> 8);

        data[2] = static_cast<uint8_t>(value >> 16);

        return *this;

    }

    bool operator==(const uint24_t& value) const { return cmp(value) == 0; }

    bool operator!=(const uint24_t& value) const { return cmp(value) != 0; }

    bool operator<(const uint24_t& value) const { return cmp(value) < 0; }

    bool operator<=(const uint24_t& value) const { return cmp(value) <= 0; }

    bool operator>(const uint24_t& value) const { return cmp(value) > 0; }

    bool operator>=(const uint24_t& value) const { return cmp(value) >= 0; }

    int32_t cmp(const uint24_t& other) const {

        if (data[2] > other.data[2]) {

            return 1;

        } else if (data[2] < other.data[2]) {

            return -1;

        }

        if (data[1] > other.data[1]) {

            return 1;

        } else if (data[1] < other.data[1]) {

            return -1;

        }

        if (data[0] > other.data[0]) {

            return 1;

        } else if (data[0] < other.data[0]) {

            return -1;

        }

        return 0;

    }

    std::string to_string() const {

        int value = *reinterpret_cast<const uint24_t*>(data);

        int mday = value & 31;

        int mon = value >> 5 & 15;

        int year = value >> 9;

        return fmt::format(FMT_COMPILE("{:04d}-{:02d}-{:02d}"), year, mon, mday);

    }

    const uint8_t* get_data() const { return data; }

private:

    uint8_t data[3];

} __attribute__((packed));

static_assert(std::is_trivial_v<uint24_t>, "uint24_t should be a POD type");

inline std::ostream& operator<<(std::ostream& os, const uint24_t& val) {

    os << val.to_string();

    return os;

}

} // namespace doris