# Copyright (C) 2009-2016 CS-SI. All Rights Reserved.
# Author: Yoann Vandoorselaere <yoann.v@prelude-ids.com>
#
# This file is part of the Prelude-Correlator program.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2, or (at your option)
# any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

try:
    import os
    from preludecorrelator import siteconfig

    def get_config_filename(fname, module=None, package="prelude-correlator"):
        return os.path.join(siteconfig.conf_dir, fname)

    def get_data_filename(fname, module=None, package="prelude-correlator", profile=None):
        return os.path.join(siteconfig.lib_dir, profile or "", fname)

except:
    import pkg_resources

    def get_config_filename(fname, module=None, package="prelude-correlator"):
        if module is None:
            module = pkg_resources.Requirement.parse(package)

        return pkg_resources.resource_filename(module, fname)

    def get_data_filename(fname, module=None, package="prelude-correlator", profile=None):
        if module is None:
            module = pkg_resources.Requirement.parse(package)

        return pkg_resources.resource_filename(module, os.path.join(profile or "", fname))

