platform-ios/RainbowPlanet/RainbowPlanet/Modules/PublishModule/PublishViewController.swift

//
//  PublishViewController.swift
//  RainbowPlanet
//
//  Created by Christopher on 2019/6/18.
//  Copyright © 2019 RainbowPlanet. All rights reserved.
//

import UIKit
import JXSegmentedView

class PublishViewController: BaseViewController {
    
    // 视频选择页参数配置类
    var mediaConfig: AliyunMediaConfig {
        //默认配置
        let mediaConfig = AliyunMediaConfig.default()
        mediaConfig?.minDuration = 3.0
        mediaConfig?.maxDuration = 60.0
        mediaConfig?.fps = 25
        mediaConfig?.gop = 5
        mediaConfig?.cutMode = AliyunMediaCutMode.scaleAspectFill
        mediaConfig?.videoOnly = false
        mediaConfig?.backgroundColor = UIColor.black
        return mediaConfig!
    }
    // 裁剪页参数配置类
    var cropMediaConfig: AliyunMediaConfig {
        //默认配置
        let cropMediaConfig = AliyunMediaConfig.default()
        cropMediaConfig?.minDuration = 3.0
        cropMediaConfig?.maxDuration = 60.0
        cropMediaConfig?.fps = 25
        cropMediaConfig?.gop = 5
        cropMediaConfig?.cutMode = AliyunMediaCutMode.scaleAspectFill
        cropMediaConfig?.videoOnly = false
        cropMediaConfig?.backgroundColor = UIColor.black
        return cropMediaConfig!
    }
    
    override func viewDidLoad() {
        super.viewDidLoad()
        setupViews()
        setupLayouts()
        setupData()
    }
    
    override func setupViews() {        
        view.addSubview(listContainerView)
        view.addSubview(segmentedView)
    }
    
    override func setupLayouts() {        
        listContainerView.snp.makeConstraints { (make) in
            make.top.left.right.equalToSuperview()
            make.bottom.equalToSuperview()
        }
        segmentedView.snp.makeConstraints { (make) in
            make.height.equalTo(48)
            make.left.equalTo(0)
            make.right.equalTo(0)
            make.bottom.equalTo(-kSafeTabBarHeight)
        }
    }
    
    override func setupData() {
        
    }
    
    //1.初始化JXSegmentedViewx
    private lazy var segmentedView: JXSegmentedView = {
        let segmentedView = JXSegmentedView()
        segmentedView.delegate = self
        segmentedView.dataSource = segmentedDataSourceAlbum
        segmentedView.indicators = [indicator]
        segmentedView.contentScrollView = listContainerView.scrollView
        segmentedView.selectItemAt(index: 0)
        segmentedView.defaultSelectedIndex = 0
        return segmentedView
    }()
    
    //2.初始化dataSource
    private lazy var segmentedDataSourceAlbum: JXSegmentedTitleDataSource = {
        let segmentedDataSource = JXSegmentedTitleDataSource()
        segmentedDataSource.titles = ["相册","拍视频","拍照"]
        segmentedDataSource.isTitleColorGradientEnabled = true
        segmentedDataSource.isItemSpacingAverageEnabled = true
        segmentedDataSource.isTitleZoomEnabled = true
        segmentedDataSource.titleNormalColor = kbbbbbbColor
        segmentedDataSource.titleSelectedColor = k333333Color
        segmentedDataSource.titleNormalFont = kRegularFont14!
        segmentedDataSource.titleSelectedFont = kBoldFont22
        segmentedDataSource.isTitleStrokeWidthEnabled = true
        //reloadData(selectedIndex:)方法一定要调用，方法内部会刷新数据源数组
        segmentedDataSource.reloadData(selectedIndex: 0)
        
        return segmentedDataSource
    }()
    private lazy var segmentedDataSourceVideo: JXSegmentedTitleDataSource = {
        let segmentedDataSource = JXSegmentedTitleDataSource()
        segmentedDataSource.titles = ["相册","拍视频","拍照"]
        segmentedDataSource.isTitleColorGradientEnabled = true
        segmentedDataSource.isItemSpacingAverageEnabled = true
        segmentedDataSource.isTitleZoomEnabled = true
        segmentedDataSource.titleNormalColor = kffffffColor
        segmentedDataSource.titleSelectedColor = kffffffColor
        segmentedDataSource.titleNormalFont = kRegularFont14!
        segmentedDataSource.titleSelectedFont = kBoldFont22
        segmentedDataSource.isTitleStrokeWidthEnabled = true
        //reloadData(selectedIndex:)方法一定要调用，方法内部会刷新数据源数组
        
        return segmentedDataSource
    }()
    
    //3.初始化指示器indicator
    private lazy var indicator: JXSegmentedIndicatorRainbowLineView = {
        let indicator = JXSegmentedIndicatorRainbowLineView()
        indicator.indicatorColors = [k62CC74Color,.white,.white]
        indicator.indicatorHeight = 3
        indicator.indicatorWidth = 20
        indicator.verticalOffset = 8
        return indicator
    }()
    
    //4.初始化JXSegmentedListContainerView
    private lazy var listContainerView: JXSegmentedListContainerView = {
        let listContainerView = JXSegmentedListContainerView(dataSource: self)
        listContainerView.didAppearPercent = 0.01
        listContainerView.defaultSelectedIndex = 0
        listContainerView.scrollView.isScrollEnabled = false
        return listContainerView
    }()
    
}

// MARK: - JXSegmentedViewDelegate
extension PublishViewController : JXSegmentedViewDelegate {
    //点击选中或者滚动选中都会调用该方法。适用于只关心选中事件，而不关心具体是点击还是滚动选中的情况。
    func segmentedView(_ segmentedView: JXSegmentedView, didSelectedItemAt index: Int) {
        switch index {
        case 0:
            self.segmentedView.backgroundColor = .white
            self.segmentedDataSourceAlbum.reloadData(selectedIndex: 0)
            self.segmentedView.dataSource = segmentedDataSourceAlbum
            self.segmentedView.reloadData()
            statusBarStyle = .default
        case 1:
            self.segmentedView.backgroundColor = .clear
            self.segmentedDataSourceVideo.reloadData(selectedIndex: 1)
            self.segmentedView.dataSource = segmentedDataSourceVideo
            self.segmentedView.reloadData()
            statusBarStyle = .lightContent
        case 2:
            self.segmentedView.backgroundColor = .clear
            self.segmentedDataSourceVideo.reloadData(selectedIndex: 2)
            self.segmentedView.dataSource = segmentedDataSourceVideo
            self.segmentedView.reloadData()
            statusBarStyle = .lightContent
        default:
            break
        }
    }
    
    // 点击选中的情况才会调用该方法
    func segmentedView(_ segmentedView: JXSegmentedView, didClickSelectedItemAt index: Int) {
        //传递didClickSelectedItemAt事件给listContainerView，必须调用！！！
        listContainerView.didClickSelectedItem(at: index)
    }
    
    // 滚动选中的情况才会调用该方法(未支持滚动选中)
    func segmentedView(_ segmentedView: JXSegmentedView, didScrollSelectedItemAt index: Int) {
        print("\n------滚动选中\(index)")
    }
    
    // 正在滚动中的回调
    func segmentedView(_ segmentedView: JXSegmentedView, scrollingFrom leftIndex: Int, to rightIndex: Int, percent: CGFloat) {
        self.segmentedView.backgroundColor = .clear

        //传递scrolling事件给listContainerView，必须调用！！！
        listContainerView.segmentedViewScrolling(from: leftIndex, to: rightIndex, percent: percent, selectedIndex: segmentedView.selectedIndex)
    }
    
    /// 是否允许点击选中目标index的item
    func segmentedView(_ segmentedView: JXSegmentedView, canClickItemAt index: Int) -> Bool {
        return true
    }
}

// MARK: - JXSegmentedListContainerViewDataSource
extension PublishViewController :JXSegmentedListContainerViewDataSource {
    
    func numberOfLists(in listContainerView: JXSegmentedListContainerView) -> Int {
        if let titleDataSource = segmentedView.dataSource as? JXSegmentedBaseDataSource {
            return titleDataSource.dataSource.count
        }
        return 0
    }
    func listContainerView(_ listContainerView: JXSegmentedListContainerView, initListAt index: Int) -> JXSegmentedListContainerViewListDelegate {
        switch index {
        case 0:
            let mediaVc = KSMediaPickerController(maxVideoItemCount: 1, maxPictureItemCount: 9)
            mediaVc.dismissClosure = {
                [weak self] in
                self?.dismiss(animated: true, completion: nil)
            }
            mediaVc.cropClosure = {
                [weak self] (vItemModel) in
                self?.getAvUrlAssetAndCrop(vItemModel)
            }
            mediaVc.pubImgClosure = {
                [weak self] (imageArray) in
                let pubVc = PublishEditController()
                pubVc.imageArr = imageArray
                self?.navigationController?.pushViewController(pubVc, animated: true)
            }
            return mediaVc
        case 1:
            let videoVc = AliyunMagicCameraViewController()
            videoVc.quVideo = mediaConfig
            videoVc.dismissBlock = {
                [weak self] in
                self?.dismiss(animated: true, completion: nil)
            }
            videoVc.editBlock = {
                [weak self] in
                let editVc = AlivcShortVideoRoute.shared().alivcViewController(with: AlivcViewControlType.edit)
                self?.navigationController?.pushViewController(editVc, animated: true)
            }
            videoVc.hideSegmentBlock = {
                [weak self] (isHidden) in
                self?.segmentedView.isHidden = isHidden
            }
            return videoVc
            
            
//            let videoVc = PublishVideoRecorderController()
//            videoVc.dismissClosure = {
//                [weak self] in
//                self?.dismiss(animated: true, completion: nil)
//            }
//            videoVc.editClosure = {
//                [weak self] in
//                let editVc = PublishEditController()
//                self?.navigationController?.pushViewController(editVc, animated: true)
//            }
//            videoVc.hideSegmentClosure = {
//                [weak self] (isHidden) in
//                self?.segmentedView.isHidden = isHidden
//            }
//            return videoVc
        default:
            let photoVc = PublishTakePhotoController()
            photoVc.dismissClosure = {
                [weak self] in
                self?.dismiss(animated: true, completion: nil)
            }
            photoVc.pubImgTransClosure = {
                [weak self] (imageArray) in
                let pubVc = PublishEditController()
                pubVc.imageArr = imageArray
                self?.navigationController?.pushViewController(pubVc, animated: true)
            }
            return photoVc
        }
    }
}

// MARK: - AliyunCropViewControllerDelegate
extension PublishViewController: AliyunCropViewControllerDelegate {
    func cropViewControllerExit() {
        self.navigationController?.popViewController(animated: true)
    }
    
    func cropViewControllerFinish(_ mediaInfo: AliyunMediaConfig!, viewController controller: UIViewController!) {
        print("----didFinishCrop")
    }
}

// MARK: -
extension PublishViewController {
    // 先将PHAsset类型转换为AVURLAsset，并填充参数
    func getAvUrlAssetAndCrop(_ videoItem: KSMediaPickerItemModel) {
        let options = PHVideoRequestOptions()
        options.isNetworkAccessAllowed = true
        options.version = PHVideoRequestOptionsVersion.current
        
        let phAsset = videoItem.asset
        PHImageManager.default().requestAVAsset(forVideo: phAsset, options: options, resultHandler: {
            [weak self] avAsset, audioMix, info in
            let cpInfo = AliyunCompositionInfo()
            cpInfo.duration = avAsset!.avAssetVideoTrackDuration()
            let urlAsset = avAsset as? AVURLAsset
            cpInfo.asset = urlAsset
            cpInfo.sourcePath = urlAsset?.url.path
            cpInfo.type = AliyunCompositionInfoTypeVideo
            
            DispatchQueue.main.async {
                self?.jumpToCropVideoVc(cpInfo)
            }
        })
    }
    
    // 跳转至裁剪页
    func jumpToCropVideoVc(_ info: AliyunCompositionInfo) {
        var path: String? = nil
        var url: URL? = nil
        url = info.asset.url
        let root: String = AliyunPathManager.compositionRootDir()
        path = root.appending(AliyunPathManager.randomString())
        path = path?.appending(".mp4")
        
        handleOriginalSize(info)
        
        let cutConfig = AliyunMediaConfig.cut(withOutputPath: path, outputSize: cropMediaConfig.outputSize, minDuration: cropMediaConfig.minDuration, maxDuration: info.duration, cutMode: cropMediaConfig.cutMode, videoQuality: cropMediaConfig.videoQuality, fps: cropMediaConfig.fps, gop: cropMediaConfig.gop)
        cutConfig?.bitrate = cropMediaConfig.bitrate
        cutConfig?.sourcePath = url?.path
        cutConfig?.phImage = info.phImage
        cutConfig?.encodeMode = cropMediaConfig.encodeMode  //编码
        cutConfig?.backgroundColor = cropMediaConfig.backgroundColor
        cutConfig?.cutMode = cropMediaConfig.cutMode
        cutConfig?.gpuCrop = cropMediaConfig.gpuCrop
        
        let cropVc = AliyunCropViewController()
        cropVc.cutInfo = cutConfig
        cropVc.delegate = self
        cropVc.fakeCrop = true
        self.navigationController?.pushViewController(cropVc, animated: true)
    }
    
    // 获取原比例的视频尺寸
    func handleOriginalSize(_ info: AliyunCompositionInfo?) {
        var size = CGSize.zero
        if info?.asset != nil {
            size = info?.asset.avAssetNaturalSize() ?? CGSize.zero
        } else {
            print("#Wrong:视频信息为nil")
        }
        
        let ratio = size.width / size.height
        if ratio > 0 {
            let height = cropMediaConfig.outputSize.width / ratio
            cropMediaConfig.outputSize = CGSize(width: cropMediaConfig.outputSize.width, height: height)
        }
        
        cropMediaConfig.outputSize = cropMediaConfig.fixedSize()
    }
}