package com.example.demo;

import com.aliyuncs.imagesearch.model.v20180120.SearchItemResponse;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.LineIterator;
import org.apache.commons.io.filefilter.DirectoryFileFilter;
import org.apache.commons.io.filefilter.IOFileFilter;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.io.*;
import java.net.URL;
import java.net.URLDecoder;
import java.net.URLEncoder;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.CopyOnWriteArrayList;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.atomic.AtomicReference;


/**
 * Created by jack on 2018/4/18.
 */
@Service
public class ShowSer {

    @Autowired
    PhotoSer aliPhotoSer;

    public ConcurrentHashMap<String, Integer> currentIndexMap = new ConcurrentHashMap<>();

    public ConcurrentHashMap<String, List<Integer>> sknDirsListMap = new ConcurrentHashMap<String, List<Integer>>();

    public ConcurrentHashMap<String, Integer> samplesMap = new ConcurrentHashMap<>();

    /*public Set<String> updateSet =Collections.synchronizedSet(new HashSet<>());*/




    public PhotoViewObj fetchCurentPhotos(String currentDir, int skip) {
        Integer index = currentIndexMap.get(currentDir);
        if (null == index) {
            index = 0;
        }
        index = index + skip;
        if (index >= sknDirsListMap.get(currentDir).size()) {
            index = 0;
        }
        currentIndexMap.put(currentDir, index);
        int skn = sknDirsListMap.get(currentDir).get(index);
        return fetchPhotos(currentDir, skn);
    }


    public PhotoViewObj fetchPrevPhotos(String currentDir) {
        int index = currentIndexMap.get(currentDir);
        index--;
        if (index < 0) {
            int last = sknDirsListMap.get(currentDir).size() - 1;

            index = last;
        }
        currentIndexMap.put(currentDir, index);
        int skn = sknDirsListMap.get(currentDir).get(index);
        return fetchPhotos(currentDir, skn);

    }

    public PhotoViewObj fetchSknPhotos(String currentDir, String skn) {
        return fetchPhotos(currentDir, Integer.valueOf(skn));
    }


    public void loadDir(String curDir) {
        List<Integer> sknDirList = new ArrayList<>();
        curDir = StringUtils.trim(curDir);
        if (curDir.endsWith("/")) {
            curDir = StringUtils.stripEnd(curDir, "/");
        }
        File file = new File(curDir);
        File[] childFiles = file.listFiles();
        List<String> childDirs = new ArrayList<>();
        for (File oneSknDir : childFiles) {
            if(!oneSknDir.isDirectory()|| oneSknDir.getName().startsWith(".")){
                continue;
            }
            sknDirList.add(Integer.valueOf(oneSknDir.getName()));
        }
        sknDirList.sort(new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                return o1.compareTo(o2);
            }
        });
        sknDirsListMap.put(curDir, sknDirList);
    }

    private PhotoViewObj fetchPhotos(String curDir, int skn) {
        PhotoViewObj photoViewObj = new PhotoViewObj();
        try {
            photoViewObj.setMainPic("/show/read?curDir=" + URLEncoder.encode(curDir) + "&skn=" + skn + "&order=0");
            for (int i = 1; i < 20; i++) {
                photoViewObj.getSimPics().add("/show/read?curDir=" + URLEncoder.encode(curDir) + "&skn=" + skn + "&order=" + i);
                photoViewObj.getSimPicFlag().add(false);
            }

            String sknDir = curDir + "/" + skn + "/";
            Iterator<File> photoIterator = FileUtils.iterateFiles(new File(sknDir), null, false);
            String resourcePath = "";

            while (photoIterator.hasNext()) {
                File onePhoto = photoIterator.next();
                String onePhotoName = onePhoto.getName();
                if (onePhotoName.startsWith("0-")) {
                    resourcePath = onePhoto.getAbsolutePath();
                     break;
                }
            }
            if (StringUtils.isNotBlank(resourcePath)) {
                SearchItemResponse response = aliPhotoSer.searchPhoto(resourcePath, 20);
                if (null != response && null != response.getAuctions()) {
                    List<SearchItemResponse.Auction> auctions = response.getAuctions();
                    if (0 < auctions.size()) {
                        for (SearchItemResponse.Auction oneAuction : auctions) {
                            String oneUrl = "/show/readAli?file=" + URLEncoder.encode("/home/downloaded_imgs-maxsort/" + oneAuction.getPicName());
                            photoViewObj.getAliPics().add(oneUrl);
                        }
                    }
                }

            }

            //读取skn.txt文件内容，展示是否被移除的标记
            String fileName =  curDir + "/" + skn + ".txt";
            File file = new File(fileName);
            if(file.exists()){//逐行读取文件
                LineIterator lineIterator =  FileUtils.lineIterator(new File(fileName), "UTF-8");
                while(lineIterator.hasNext()){
                    String content = lineIterator.next();
                    if(StringUtils.isNotEmpty(content)){
                        String imageName = StringUtils.split(content, ",")[1];
                        String indexStr = StringUtils.split(imageName, "-")[0];
                        int imageIndex = Integer.valueOf(indexStr) - 1;
                        photoViewObj.getSimPicFlag().set(imageIndex, true);//加入被移除的标记
                    }

                }
                lineIterator.close();//关闭

            }

        } catch (Exception e) {
            e.printStackTrace();
        }

        return photoViewObj;
    }


    public Object readPhotoBytes(String curDir, String skn, String order) throws IOException {
        String photoFilePath = "";
        String sknDir = curDir + "/" + skn + "/";
        Iterator<File> photoIterator = FileUtils.iterateFiles(new File(sknDir), null, false);
        Map<String, String> indexMap = new HashMap<>();
        while (photoIterator.hasNext()) {
            File onePhoto = photoIterator.next();
            String onePhotoName = onePhoto.getName();
            String[] names = StringUtils.split(onePhotoName, "-");
            if (null == names || 0 == names.length) continue;
            if (StringUtils.equals(order, names[0])) {
                photoFilePath = sknDir + onePhotoName;
                break;
            }
        }
        if (StringUtils.isNotBlank(photoFilePath))
            return FileUtils.readFileToByteArray(new File(photoFilePath));
        else
            return null;
    }

    public Object readPhotoBytes(String filePath) throws IOException {
        return FileUtils.readFileToByteArray(new File(filePath));
    }

    //获取样本信息---分页
    public List<SampleViewObj> fetchSamplesByPage(String sampleDir, int startIndex, int pageSize)  {
        List<SampleViewObj> pageList = new ArrayList<>();
        LineIterator qInfo = null;
        LineIterator pInfo = null;
        LineIterator nInfo = null;
        try {
            qInfo = FileUtils.lineIterator(new File(sampleDir+"/q.info"), "UTF-8");
            pInfo = FileUtils.lineIterator(new File(sampleDir+"/p.info"), "UTF-8");
            nInfo = FileUtils.lineIterator(new File(sampleDir+"/n.info"), "UTF-8");
            int endIndex = startIndex + pageSize - 1;
            List<String> qList = readFileByIndex(qInfo,startIndex,endIndex);
            List<String> pList = readFileByIndex(pInfo,startIndex,endIndex);
            List<String> nList = readFileByIndex(nInfo,startIndex,endIndex);
            String readUrl = "/show/readAli?file=";
            String updateFilePath = sampleDir+"/update.info";
            Set<Integer> updateSet = getUpdateFileInfo(updateFilePath);
            for(int i = 0 ; i< qList.size(); i++){
                int itemIndex = startIndex + i;//所在的行 以0为起始
                boolean flag = false;
                if(updateSet.contains(itemIndex)){
                    flag = true;
                }
                SampleViewObj sampleViewObj = new SampleViewObj(
                        readUrl+ URLEncoder.encode(qList.get(i)),
                        readUrl+ URLEncoder.encode(pList.get(i)),
                        readUrl+ URLEncoder.encode(nList.get(i)),
                        itemIndex,
                        flag);
                pageList.add(sampleViewObj);
            }
        } catch (IOException e) {
            e.printStackTrace();
        }finally {
            LineIterator.closeQuietly(qInfo);
            LineIterator.closeQuietly(pInfo);
            LineIterator.closeQuietly(nInfo);
        }
        return pageList;

    }

    //读取updateFile文件
    private Set<Integer> getUpdateFileInfo(String updateFilePath) {
        Set<Integer> set = new HashSet<>();
        LineIterator lineIterator = null;
        try {
            lineIterator = FileUtils.lineIterator(new File(updateFilePath));
            while (lineIterator.hasNext()){
                String s = lineIterator.next();
                if(StringUtils.isEmpty(s))continue;
                int value = Integer.valueOf(s);
                set.add(value);
            }
        } catch (IOException e) {
            e.printStackTrace();
        }finally {
            LineIterator.closeQuietly(lineIterator);
        }

        return set;
    }

    //读取文件
    private List<String> readFileByIndex(LineIterator lineIterator, int startIndex, int endIndex) {
        List<String> info = new ArrayList<>();
        int currentNum = -1;
        while (lineIterator.hasNext()) {
            currentNum++;
            String s = lineIterator.next();
            if(currentNum < startIndex)continue;
            if(currentNum > endIndex)break;
            if(StringUtils.isNotEmpty(s)){
                info.add(s);
            }
        }
        return info;
    }



    public void loadSampleInfo(String sampleDir) {
        LineIterator qInfo = null;
        try {
            qInfo = FileUtils.lineIterator(new File(sampleDir+"/q.info"), "UTF-8");
            samplesMap.put(sampleDir, getFileLines(qInfo));
        } catch (IOException e) {
            e.printStackTrace();
        }finally {
            LineIterator.closeQuietly(qInfo);
        }
    }

    private int getFileLines(LineIterator lineIterator) {
        int count= 0;
        while (lineIterator.hasNext()) {
            count ++;
            lineIterator.next();
        }
        return count;
    }

    /**
     * 更新 update.info 文件
     *
     * @param file
     * @param flag --true ,当前行号写入文件 false --当前行 移出文件
     */


    /*public synchronized  boolean  modifyFile(File file, int content, boolean flag) {
       if(flag){
            try {//追加写入文件
                FileUtils.writeLines(file,"utf-8",Arrays.asList(content), true);
            } catch (IOException e) {
                e.printStackTrace();
                return false;
            }
        };

       return true;
    }*/

    public synchronized  boolean  insertLine(File file, String content) {
            try {//追加写入文件
                FileUtils.writeLines(file,"utf-8", Arrays.asList(content), true);
            } catch (IOException e) {
                e.printStackTrace();
                return false;
            }
        return true;
    }
    /**
     * 更新update文件
     * @param file
     * @param contentListStr
     * @param flag  ---true ：表示写入文件； false: 表示从文件中删除
     * @return
     */
    public synchronized  boolean  modifyFile(File file, String contentListStr, boolean flag) {
        String[] strings = StringUtils.split(contentListStr, ",");
        if(flag){
            try {//追加写入文件
                FileUtils.writeLines(file,"utf-8",Arrays.asList(strings), true);
            } catch (IOException e) {
                e.printStackTrace();
                return false;
            }
        }else {//从文件中移出
            try {
                BufferedReader br = new BufferedReader(new FileReader(file));
                //临时文件
                File tempFile = new File(file.getAbsolutePath() + ".tmp");
                PrintWriter pw = new PrintWriter(new FileWriter(tempFile));
                String line = null;
                while ((line = br.readLine()) != null) {
                    //写入临时文件
                    if (!line.trim().equals(strings[0])) {
                        pw.println(line);
                        pw.flush();
                    }
                }

                //关闭流
                pw.close();
                br.close();

                //


                if (!file.delete()) {
                    return false;
                }

                if (!tempFile.renameTo(file)){
//                    System.out.println("Could not rename file");
                    return false;
                  }

            } catch (IOException e) {
                e.printStackTrace();
                return false;
            }
        }

        return true;
    }


    public boolean checkFile(File file){
        if(!file.exists()){
            try {
                FileUtils.forceMkdirParent(file);
            } catch (IOException e) {
                e.printStackTrace();
                return false;
            }
        }
        return true;
    }


    /***
     * 根据图片src获取图片的名称
     */

    public String getImageNameBySrc(String removeImageSrc, String curDir, String skn) {
        String paramStr = removeImageSrc.substring( removeImageSrc.indexOf("?") + 1);
        String s[] = paramStr.split( "&");
       /* String curDir = "";
        String skn = "";*/
        String order = "";
        for( int i = 0 ; i < s.length ; i ++){
            if(StringUtils.startsWith(s[i], "order")){
                order = s[i].split("=")[1];
                break;
            }
        }


        String sknDir = curDir + "/" + skn + "/";
        Iterator<File> photoIterator = FileUtils.iterateFiles(new File(sknDir), null, false);
        String photoName = "";
        while (photoIterator.hasNext()) {
            File onePhoto = photoIterator.next();
            String onePhotoName = onePhoto.getName();
            if (onePhotoName.startsWith(order +"-")) {
                photoName = onePhotoName;
                break;

            }
        }

//        return photoName.substring(2);//头部不显示
        return photoName;
    }

    //统计文件的行
    public int  countFileLines(File file)  {
        int count = 0;
        LineIterator lineIterator = null;
        try {
            lineIterator = FileUtils.lineIterator(file, "UTF-8");
            while (lineIterator.hasNext()) {
                lineIterator.next();
                count ++;
            }
        } catch (IOException e) {
            e.printStackTrace();
        }

        return count;

    }

    /**
     *
     * @param file
     * @param iDisplayStart --起始行
     * @param iDisplayLength --总行数（默认值是1）
     * @return
     */
    public List<ImageAndSimilarResp> getImagesPyPage(File file, Integer iDisplayStart, Integer iDisplayLength) {
        List<ImageAndSimilarResp> result = new ArrayList<>();
        try {
            int currentLine = 0;//当前行
            int readTotal = 0;//已经读取的行数
            boolean readFlag = false;
            LineIterator lineIterator =  FileUtils.lineIterator(file, "UTF-8");
            while(lineIterator.hasNext()){
                String content = lineIterator.next();
                if(iDisplayStart == currentLine){//需要解析的起始行
                    readFlag = true;
                }

                if(readFlag){
                    ImageAndSimilarResp obj =  analyzeLineContent(content);
                    result.add(obj);
                    readTotal++;//已经读取的行数
                }

                if(readTotal == iDisplayLength){
                    break;
                }
                currentLine++;
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
        return  result;
    }


    private ImageAndSimilarResp analyzeLineContent(String content) {
        ImageAndSimilarResp obj = new ImageAndSimilarResp();
//        测试的
//        String[] s = StringUtils.split(content, "-");
        String[] s = StringUtils.split(content, ":");
        if(s.length == 2){//正确解析的情况
            obj.setSourceImage(s[0]);
            String[] similars = StringUtils.split(s[1],",");
            if(similars.length > 50){//前50
                String[] subStr = Arrays.copyOfRange(similars, 0 , 50);
                obj.setSimilarImages(Arrays.asList(subStr));
            }else{
                obj.setSimilarImages( Arrays.asList(similars));
            }

        }
        return obj;

    }
}