// Copyright 2023 Greptime Team
//
// Licensed 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.

mod build_index_table;
mod discard_unflushed_data;
mod flush_compact_region;
mod flush_compact_table;
mod gc;
mod migrate_region;
#[cfg(feature = "enterprise")]
mod purge_table;
mod reconcile_catalog;
mod reconcile_database;
mod reconcile_table;

use flush_compact_region::{CompactRegionFunction, FlushRegionFunction};
use flush_compact_table::{CompactTableFunction, FlushTableFunction};
use gc::{GcRegionsFunction, GcTableFunction};
use migrate_region::MigrateRegionFunction;
#[cfg(feature = "enterprise")]
use purge_table::PurgeTableFunction;
use reconcile_catalog::ReconcileCatalogFunction;
use reconcile_database::ReconcileDatabaseFunction;
use reconcile_table::ReconcileTableFunction;

use crate::admin::build_index_table::BuildIndexFunction;
use crate::admin::discard_unflushed_data::DiscardUnflushedDataFunction;
use crate::flush_flow::FlushFlowFunction;
use crate::function_registry::FunctionRegistry;

/// Administration functions
pub(crate) struct AdminFunction;

impl AdminFunction {
    /// Register all admin functions to [`FunctionRegistry`].
    pub fn register(registry: &FunctionRegistry) {
        registry.register(MigrateRegionFunction::factory());
        registry.register(FlushRegionFunction::factory());
        registry.register(CompactRegionFunction::factory());
        registry.register(FlushTableFunction::factory());
        registry.register(CompactTableFunction::factory());
        registry.register(GcRegionsFunction::factory());
        registry.register(GcTableFunction::factory());
        registry.register(BuildIndexFunction::factory());
        registry.register(FlushFlowFunction::factory());
        registry.register(ReconcileCatalogFunction::factory());
        registry.register(ReconcileDatabaseFunction::factory());
        registry.register(ReconcileTableFunction::factory());
    }

    /// Register functions that must only be resolved by an ADMIN statement.
    pub fn register_admin_only(registry: &FunctionRegistry) {
        registry.register(DiscardUnflushedDataFunction::factory());
        #[cfg(feature = "enterprise")]
        registry.register(PurgeTableFunction::factory());
    }
}
