/*
 * Copyright 2013 Websquared, Inc.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *   http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.fastcatsearch.datasource.reader;

import org.fastcatsearch.ir.common.IRException;
import org.fastcatsearch.ir.document.Document;
import org.fastcatsearch.ir.field.Field;
import org.fastcatsearch.ir.settings.FieldSetting;
import org.fastcatsearch.ir.settings.SchemaSetting;
import org.fastcatsearch.util.HTMLTagRemover;

import java.util.Map;

/**
 * 데이터소스 리더.
 * 
 * */
public class DefaultDataSourceReader extends AbstractDataSourceReader<Map<String, Object>> {

	// DataSourceConfig안에는 SingleSourceConfig가 여러개 들어있다.
	public DefaultDataSourceReader(SchemaSetting schemaSetting) throws IRException {
		super(schemaSetting);
	}

	public Document createDocument(Map<String, Object> map) throws IRException {
		FieldSetting fs = null;
		Object data = null;
		try {

            /*
            * 루프를 돌기전에 먼저 target속성이 존재하면 그 필드를 먼저수행해준다. target은 transform이 있어야만 한다.
            * 왜냐하면, 다른 필드에 값을 넣어주기 때문에..
            * 또한 해당 값이 Multivalue라면 먼저 파싱해서 transform을 해야한다.
            *
            * */
//			logger.debug("doc >> {}", map);
			// Schema를 기반으로 Document로 만든다.
			Document document = new Document(fieldSettingList.size());
			for (int i = 0; i < fieldSettingList.size(); i++) {
				fs = fieldSettingList.get(i);
				
				String key = fs.getId();
				String source = fs.getSource();
				if(source != null && source.length() > 0){
					//source가 있다면 source에서 데이터를 가져온다.
					key = source;
				}
				data = map.get(key);
				//데이터가 없으면 소문자로 시도. 퍼블릭 REST API를 통해 받은 json등은 키를 변경하기가 어려우므로 추가된 기능.
				if(data == null) {
					data = map.get(key.toLowerCase());
				}
				//null이면 공백문자로 치환.
				if(data == null) {
					data = "";
				} else if (data instanceof String) {
					data = ((String) data).trim();
				}


//				logger.debug("Get {} : {}", key, data);
				String multiValueDelimiter = fs.getMultiValueDelimiter();
				Field f = fs.createIndexableField(data, multiValueDelimiter);
				document.set(i, f);
//				logger.debug("doc [{}]{}:{}", i, fs.getId(), f);
			}
			return document;
		} catch (Throwable e) {
			close();
			if(fs!=null) {
				logger.error("", e);
				throw new IRException("Exception At Field ["+fs.getName()+"] in \""+data+"\"", e);
			} else {
				logger.error("", e);
				throw new IRException(e.getMessage());
			}
		}
	}

}