# -*- coding: utf-8 -*-
'''
Writes matches to disk to verify activity, helpful when testing
'''

# import python libs
from __future__ import absolute_import
import os
import json

# Import salt libs
import salt.utils


def save(name):
    '''
    Save the register to <salt cachedir>/thorium/saves/<name>
    '''
    ret = {'name': name,
           'changes': {},
           'comment': '',
           'result': True}
    tgt_dir = os.path.join(__opts__['cachedir'], 'thorium', 'saves')
    fn_ = os.path.join(tgt_dir, name)
    if not os.path.isdir(tgt_dir):
        os.makedirs(tgt_dir)
    with salt.utils.fopen(fn_, 'w+') as fp_:
        fp_.write(json.dumps(__reg__))
    return ret

# -*- coding: utf-8 -*-
'''
Writes matches to disk to verify activity, helpful when testing

Normally this is used by giving the name of the file (without a path) that the
data will be saved to. If for instance you use ``foo`` as the name:

.. code-block: yaml

    foo:
      file.save

Then the file will be saved to:

.. code-block: bash

    <salt cachedir>/thorium/saves/foo

You may also provide an absolute path for the file to be saved to:

    /tmp/foo.save:
        file.save

Files will be saved in JSON format. However, JSON does not support ``set()``s.
If you are saving a register entry that contains a ``set()``, then it will fail
to save to JSON format. However, you may pass data through a filter which makes
it JSON compliant:

    foo:
      file.save:
        filter: True

Be warned that if you do this, then the file will be saved, but not in a format
that can be re-imported into Python.
'''

# import python libs
from __future__ import absolute_import
import os
import json

# Import salt libs
import salt.utils
from salt.utils import simple_types_filter


def save(name, filter=False):
    '''
    Save the register to <salt cachedir>/thorium/saves/<name>
    '''
    ret = {'name': name,
           'changes': {},
           'comment': '',
           'result': True}
    if name.startswith('/'):
        tgt_dir = name
    else:
        tgt_dir = os.path.join(__opts__['cachedir'], 'thorium', 'saves')
    fn_ = os.path.join(tgt_dir, name)
    if not os.path.isdir(tgt_dir):
        os.makedirs(tgt_dir)
    with salt.utils.fopen(fn_, 'w+') as fp_:
        if filter is True:
            fp_.write(json.dumps(simple_types_filter(__reg__)))
        else:
            fp_.write(json.dumps(__reg__))
    return ret

Allow filtering in Thorium file.save