Synnefo » Pithos MS Client

using System;

using System.Collections.Generic;

using System.ComponentModel.Composition;

using System.Diagnostics;

using System.Diagnostics.Contracts;

using System.IO;

using Castle.ActiveRecord;

using Castle.ActiveRecord.Framework.Config;

using Pithos.Interfaces;

namespace Pithos.Core

{

    [Export(typeof(IStatusChecker)),Export(typeof(IStatusKeeper))]

    public class StatusChecker:IStatusChecker,IStatusKeeper

    {

        [System.ComponentModel.Composition.Import]

        public IPithosSettings Settings { get; set; }

        public StatusChecker()

        {

            var source = new XmlConfigurationSource("DbConfig.xml");

            ActiveRecordStarter.Initialize(source,typeof(FileState));            

            if (!File.Exists("pithos.db"))

                ActiveRecordStarter.CreateSchema();            

        }

        public FileOverlayStatus GetFileOverlayStatus(string path)

        {

            try

            {

                var state = FileState.TryFind(path);

                return state == null ? FileOverlayStatus.NA : state.OverlayStatus;

            }

            catch (Exception exc)

            {

                Trace.TraceError(exc.ToString());

                return FileOverlayStatus.NA;

            }

        }

        private PithosStatus _pithosStatus=PithosStatus.InSynch;

        public void SetPithosStatus(PithosStatus status)

        {

            _pithosStatus = status;

        }

        public PithosStatus GetPithosStatus()

        {

            return _pithosStatus;

        }

        public void SetFileOverlayStatus(string path, FileOverlayStatus overlayStatus)

        {

            var state = FileState.TryFind(path);

            if (state != null)

            {

                state.OverlayStatus = overlayStatus;

                state.Update();

            }

            else

            {

                state=new FileState{FilePath=path,OverlayStatus=overlayStatus};

                state.Save();

            }

        }

        public void RemoveFileOverlayStatus(string path)

        {

            FileState.DeleteAll(new[] {path});            

        }

        public void RenameFileOverlayStatus(string oldPath, string newPath)

        {

            var state = FileState.Find(oldPath);

            //TODO: This will cause problems if path is used as a key in relationships

            state.FilePath = newPath;

            state.Update();

        }

        public void SetFileStatus(string path, FileStatus status)

        {

            var state = FileState.Find(path);

            state.FileStatus = status;

        }

        public FileStatus GetFileStatus(string path)

        {

            var state = FileState.TryFind(path);

            return (state==null)?FileStatus.Missing:state.FileStatus ;

        }

        public void ClearFileStatus(string path)

        {

            //TODO:SHOULDN'T need both clear file status and remove overlay status

            FileState.DeleteAll(new[] { path });   

        }

        public void UpdateFileChecksum(string path, string checksum)

        {

            var state = FileState.Find(path);

            state.Checksum = checksum;

            state.Update();

        }

    }

}