ntpa/Ntp.Analyzer/Export/DirectoryStreamDestination.cs

//
// DirectoryStreamDestination.cs
//
// Author:
//       Carsten Sonne Larsen <cs@innolan.dk>
//
// Copyright (c) 2013-2016 Carsten Sonne Larsen
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.

using System;
using System.IO;
using Ntp.Analyzer.Log;
using Ntp.System;

namespace Ntp.Analyzer.Export
{
    public class DirectoryStreamDestination : StreamDestination
    {
        /// <summary>
        /// Initializes a new instance of the <see cref="Ntp.Analyzer.Export.DirectoryStreamDestination" /> class.
        /// </summary>
        /// <param name="basePath">Base path.</param>
        /// <param name="path">Path.</param>
        /// <param name="prefix">Prefix.</param>
        public DirectoryStreamDestination(string basePath, string path, string prefix)
        {
            path = (path != null ? path.Trim () : String.Empty);
            basePath = (basePath != null ? basePath.Trim () : String.Empty);

            if (basePath.Length > 1 && path [path.Length - 1] != '/')
                basePath = basePath + "/";

            if (path.Length > 1 && path [path.Length - 1] != '/')
                path = path + "/";

            this.path = basePath + path;
            this.prefix = prefix;
        }

        private readonly string path;
        private readonly string prefix;

        public override string Location
        {
            get { return path; }
        }

        public override string AbsoluteLocation
        {
            get { return path + prefix; }
        }

        public override void Test()
        {
            const string testContent = "FileStreamDestinationTest";
            string testFile = AbsoluteLocation + "configtest";

            if (File.Exists(testFile))
                File.Delete(testFile);

            File.WriteAllText(testFile, testContent);
            File.Delete(testFile);
        }

        public override void Write(Stream stream, LogBase log)
        {
            string file = path + prefix + Param;

            string created = null;
            try {
                created = DirectoryCommand.CreateRecursiveFromFile (file);
            } catch (Exception e) {
                log.WriteLine (String.Format (
                    "Could create path for file {0}", file),
                    Severity.Error);
                log.WriteLine (e, Severity.Trace);
            }

            if (created != null) {
                log.WriteLine (String.Format (
                    "Created new directory {0}", created),
                    Severity.Notice);
            }

            if (File.Exists (file))
                File.Delete (file);

            FileStream fileStream = null;
            try {
                fileStream = new FileStream (file, FileMode.CreateNew);
                stream.CopyTo (fileStream);
            } catch (Exception e) {
                log.WriteLine (String.Format (
                    "Could not write file {0}", file),
                    Severity.Error);
                log.WriteLine (e, Severity.Trace);
            } finally {
                // Release memory
                if (fileStream != null) {
                    fileStream.Close ();
                    fileStream = null;
                }
            }
        }
    }
}