WordPressのDB（$table_prefix）をURLパラメーターに応じて切り替えたい

/**
 * WordPress データベーステーブルの接頭辞
 *
 * それぞれにユニーク (一意) な接頭辞を与えることで一つのデータベースに複数の WordPress を
 * インストールすることができます。半角英数字と下線のみを使用してください。
 */
$table_prefix = 'wp_';
$ref = filter_input( INPUT_GET, 'ref' );

// A株式会社の場合
if (
    $ref === 'company-a'
    || filter_input( INPUT_COOKIE, 'company_id' ) === 'company-a'
) {
    $table_prefix = 'ca_';
}

// A株式会社以外の場合
if (
    $ref !== 'company-a'
    $ref !== null
    && $ref !== false
) {
    $table_prefix = 'wp_';
}

/**
 * テーマが読み込まれた時の処理
 */
add_action( 'after_setup_theme', function() {
    // 会社情報をCookieに保存
    set_cookie_company();
} );

/**
 * 会社情報を取得
 *
 * @return array
 */
function get_company() {
    $COMPANY = [
        [
            'id' => 'company-a',
            'ref' => 'company-a',
            'name' => 'A株式会社',
        ],
    ];

    return $COMPANY;
}

/**
 * 会社IDを取得
 * 
 * @return string|bool
 */
function get_company_id() {
    // 会社情報
    $COMPANY = get_company();

    // 参照元
    $ref = filter_input( INPUT_GET, 'ref' );

    // 会社ID
    $id = false;

    // Cookie名
    $cookie_name = 'company_id';

    // Cookieに保存されている会社ID
    $cookie_company_id = filter_input( INPUT_COOKIE, $cookie_name );

    // Cookieに会社IDが保存されていて、かつ会社情報に一致するIDの会社がある場合
    if ( $cookie_company_id ) {
        foreach ( $COMPANY as $key => $val ) {
            if ( $cookie_company_id === $val['id'] ) {
                $id = $COMPANY[ $key ]['id'];
                break;
            }
        }
    }

    // パラメーターで会社が指定されていて、かつ会社情報がある場合
    if ( get_company_id_from_params( $ref ) ) {
        $id = get_company_id_from_params( $ref );
    }

    return $id;
}

/**
 * 会社情報をCookieに保存
 *
 * @return string|bool
 */
function set_cookie_company() {
    // Cookie名
    $cookie_name = 'company_id';

    // Cookieに保存されている会社ID
    $cookie_company_id = filter_input( INPUT_COOKIE, $cookie_name );

    // 会社ID
    $id = get_company_id();

    if ( $id === false ) {
        return false;
    }

    // 会社IDのCookieの値
    $cookie_val = $id;

    if ( ! $cookie_company_id || $cookie_company_id !== $cookie_val ) {
        setcookie( $cookie_name, $cookie_val, 0, '/' );
    }
}

/**
 * パラメーターから会社IDを取得
 *
 * @param array $ref パラメーター
 * @return string|bool
 */
function get_company_id_from_params( $ref ) {
    if ( empty( $ref ) ) {
        return false;
    }

    // 会社情報
    $COMPANY = get_company();

    // 会社のキー
    $key = array_search( $ref, array_column( $COMPANY, 'ref' ) );

    if (
        $ref !== null
        && $key !== false
    ) {
        return $COMPANY[ $key ]['id'];
    }

    return false;
}

/**
 * 管理画面用JavaScript読み込み
 */
add_action( 'admin_footer', function() {
    echo "<script>";
    echo "(function() {";
        echo "if (location.pathname === '/wp-admin/' || location.pathname === '/wp-admin/index.php') {";
            echo "alert('A株式会社用の管理画面にアクセスしています。');";
        echo "}";
    echo "})();";
    echo "</script>";
} );