<?php
if ( ! defined( 'ABSPATH' ) ) {
	exit; // Exit if accessed directly
}

// Allow some additional file types for upload.
function streamcast_mime_types( $mimes ) {
	// New allowed mime types.
	$mimes['pls']  = 'audio/x-scpls';
	$mimes['m3u8'] = 'application/vnd.apple.mpegurl';

	return $mimes;
}
add_filter( 'upload_mimes', 'streamcast_mime_types' );

function streamcast_add_allow_upload_extension_exception( $data, $file, $filename, $mimes, $real_mime = null ) {
	// If file extension is 2 or more.
	$f_sp        = explode( '.', $filename );
	$f_exp_count = count( $f_sp );

	if ( $f_exp_count <= 1 ) {
		return $data;
	} else {
		$f_name = $f_sp[0];
		$ext    = $f_sp[ $f_exp_count - 1 ];
	}

	if ( 'pls' === $ext ) {
		$type            = 'audio/x-scpls';
		$proper_filename = '';
		return compact( 'ext', 'type', 'proper_filename' );
	} elseif ( 'm3u8' === $ext ) {
		$type            = 'application/vnd.apple.mpegurl';
		$proper_filename = '';
		return compact( 'ext', 'type', 'proper_filename' );
	} else {
		return $data;
	}
}

// It's different arguments between WordPress 5.1 and previous versions.
global $wp_version;
if ( version_compare( $wp_version, '5.1' ) >= 0 ) {
	add_filter( 'wp_check_filetype_and_ext', 'streamcast_add_allow_upload_extension_exception', 10, 5 );
} else {
	add_filter( 'wp_check_filetype_and_ext', 'streamcast_add_allow_upload_extension_exception', 10, 4 );
}