#!/usr/bin/env python
# coding=utf-8
#
# Copyright ? 2011-2015 Splunk, 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.

from __future__ import absolute_import, division, print_function, unicode_literals
# import app
import sys
import os

import logging as logger

#logger.basicConfig(level=logger.INFO, format='%(asctime)s %(levelname)s %(message)s',
#                   filename=os.path.join('/opt/splunk/', 'test.log'), filemode='a')

from splunklib.searchcommands import dispatch, EventingCommand, Configuration, Option, validators
from splunklib import six


def formattsinge(sourcedata, filterarray, max_order, results, src, dest,iter_mode):
    filterarray2 = []
    sourcedata2 = []
    unsourcedata2 = {}
    limit_bool = True if max_order <= -1 or max_order > 0 else False
    if filterarray and sourcedata and limit_bool:
        for row in sourcedata:
            haved_bool = True
            for filter in filterarray:
                if haved_bool:
                    if iter_mode == 'src' or iter_mode == 'all':
                        if filter == row[src]:
                            filterarray2.append(row[dest])
                            results.append(row)
                            unsourcedata2[row[src] + row[dest]] = row
                            haved_bool = False

                    if iter_mode == 'dest' or iter_mode == 'all':
                        if filter == row[dest]:
                            filterarray2.append(row[src])
                            results.append(row)
                            unsourcedata2[row[src] + row[dest]] = row
                            haved_bool = False

        max_order -= 1
        unsourcedata2_keys = unsourcedata2.keys()
        for row1 in sourcedata:
            if row1[src] + row1[dest] not in unsourcedata2_keys:
                sourcedata2.append(row1)
        formattsinge(sourcedata2, filterarray2, max_order, results, src, dest,iter_mode)
    else:
        return results
    return results

def formattsinge_sort(sourcedata, filterarray, max_order, results, src, dest, sort_field,iter_mode):
    filterarray2 = []
    sourcedata2 = []
    unsourcedata2 = {}
    limit_bool = True if max_order == -1 or max_order > 0 else False
    if filterarray and sourcedata and limit_bool and len(filterarray) < max_order:
        for row in sourcedata:
            haved_bool = True
            for filter in filterarray:
                if haved_bool:
                    filters = filter.split('_') if len(filter.split('_')) > 1 else [filter, 0]
                    if iter_mode == 'src' or iter_mode == 'all':
                        if filters[0] == row[src] and filters[1] < row[sort_field]:
                            filterarray2.append(row[dest] + '_' + row[sort_field])
                            results.append(row)
                            unsourcedata2[row[src] + row[dest]] = row
                            haved_bool = False
                    if iter_mode == 'dest' or iter_mode == 'all':
                        if filters[0] == row[dest] and filters[1] > row[sort_field]:
                            filterarray2.append(row[src] + '_' + row[sort_field])
                            results.append(row)
                            unsourcedata2[row[src] + row[dest]] = row
                            haved_bool = False

        max_order -= 1
        unsourcedata2_keys = unsourcedata2.keys()
        for row1 in sourcedata:
            if row1[src] + row1[dest] not in unsourcedata2_keys:
                sourcedata2.append(row1)
        formattsinge_sort(sourcedata2, filterarray2, max_order, results, src, dest, sort_field,iter_mode)
    else:
        return results
    return results

@Configuration()
class DiagramSingeCommand(EventingCommand):
    # self.logger.debug("command start run ...")
    limit = Option(require=False, validate=validators.Integer())
    src_field = Option(require=True, validate=validators.Fieldname())
    dest_field = Option(require=True, validate=validators.Fieldname())
    sort_field = Option(require=False, validate=validators.Fieldname())
    value = Option(require=True)
    iter_mode = Option(require=False)

    def transform(self, records):
        self.logger.debug('CountMatchesCommand: %s', records)  # logs command line

        src_field = self.src_field
        dest_field = self.dest_field
        sort_field = self.sort_field
        value = [str(self.value)]
        results_default = []
        iter_mode = self.iter_mode if self.iter_mode and self.iter_mode in ['src', 'dest', 'all'] else 'src'
        limit = self.limit if self.limit else 1000
        records_list = []
        for r in records:
            records_list.append(r)

        if sort_field:
            records = formattsinge_sort(records_list, value, limit, results_default, src_field, dest_field,sort_field,iter_mode)
        else :
            records = formattsinge(records_list, value, limit, results_default, src_field, dest_field,iter_mode)

        for record in records:
            yield record


dispatch(DiagramSingeCommand, sys.argv, sys.stdin, sys.stdout, __name__)

[diagramsinge]
filename = diagramsinge.py
supports_getinfo = false
supports_rawargs = true
chunked = true

[diagramsinge-command]
syntax = diagramsinge src_field=<field_name> dest_field=<field_name> value=<value> limit=<int> sort_field=<field_name> iter_mode=<mode>
alias =
shortdesc = 格式化統計結果.
description = \
    追溯與指定源關聯的結果
comment1 = \
    運行此命令后會修改事件告警狀態.
example1 = | stats count by src dest \
    | diagramsinge src_field="src" dest_field="dest" value="127.0.0.1"
appears-in = 1.5
maintainer = dnoble
usage = public
tags = stats