import lodashSnakecase from 'lodash.snakecase';
import { mapKeys } from './internal/utils';
import { isString } from './internal/validators';

/**
 * Snakelize string or keys of given object(s) if provided. Only top level keys are transformed when handling object(s).
 * Input strings / keys assumed to be camelCase.
 *
 * ```typescript
 * snakelize('CamelCase') // ~ camel_case
 * // dots are preserved
 * snakelize('FancyCars.LicenseNumber') // ~ fancy_cars.license_number
 * // handles objects, but no recursive transformation
 * snakelize([{snakelizeDis: {leaveThis: 1}}]) // [{snakelize_dis: {leaveThis: 1}}]
 * ```
 */
function snakelize(string: string): string;
function snakelize(object: { [key: string]: any }): { [key: string]: any };
function snakelize(array: Array<{ [key: string]: any }>): Array<{ [key: string]: any }>;
function snakelize(input: any) {
    if (isString(input)) {
        return input
            .split('.')
            .map(lodashSnakecase)
            .join('.');
    }
    return Array.isArray(input) ? input.map(o => mapKeys(o, snakelize)) : mapKeys(input, snakelize);
}

export default snakelize;