Util/FSpotTools.cs

   1 //
   2 // FSpotTools.cs
   3 //
   4 // Copyright (C) 2004 Novell, Inc.
   5 //
   6
   7 //
   8 // Permission is hereby granted, free of charge, to any person obtaining a
   9 // copy of this software and associated documentation files (the "Software"),
  10 // to deal in the Software without restriction, including without limitation
  11 // the rights to use, copy, modify, merge, publish, distribute, sublicense,
  12 // and/or sell copies of the Software, and to permit persons to whom the
  13 // Software is furnished to do so, subject to the following conditions:
  14 //
  15 // The above copyright notice and this permission notice shall be included in
  16 // all copies or substantial portions of the Software.
  17 //
  18 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  19 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  20 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  21 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  22 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  23 // FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  24 // DEALINGS IN THE SOFTWARE.
  25 //
  26
  27 using System;
  28 using System.Collections;
  29 using System.IO;
  30 using Mono.Data.SqliteClient;
  31
  32 namespace Beagle.Util {
  33
  34         public class FSpotTools {
  35
  36                 public class Tag {
  37                         public uint   Id;
  38                         public string Name;
  39                         public uint   CategoryId;
  40                         public bool   IsCategory;
  41                         public int    SortPriority;
  42                         public Tag    Category;
  43                         // FIXME: Icon
  44                 }
  45
  46                 public class Photo {
  47                         public uint   Id;
  48                         public string Path;
  49                         public string Description;
  50                         public Tag[]  Tags;
  51                 }
  52
  53
  54                 static private SqliteConnection connection;
  55                 static Hashtable tagCache = null;
  56                 static Hashtable directoryCache = null;
  57
  58
  59                 static private string PhotoStorePath {
  60                         get {
  61                                 string home = Environment.GetEnvironmentVariable ("HOME");
  62                                 return Path.Combine (home, ".gnome2/f-spot/photos.db");
  63                         }
  64                 }
  65
  66                 static private SqliteConnection PhotoStoreConnection {
  67                         get {
  68                                 if (connection == null && File.Exists (PhotoStorePath)) {
  69                                         connection = new SqliteConnection ();
  70                                         connection.ConnectionString = "URI=file:" + PhotoStorePath;
  71                                         connection.Open ();
  72                                 }
  73
  74                                 return connection;
  75                         }
  76                 }
  77
  78                 static private bool HavePhotoStore {
  79                         get {
  80                                 return PhotoStoreConnection != null;
  81                         }
  82                 }
  83
  84                 // FIXME: We should expire this cache if the underlying db has changed.
  85                 static private Tag GetTagById (uint id)
  86                 {
  87                         if (! HavePhotoStore)
  88                                 return null;
  89
  90                         if (tagCache == null) {
  91                                 tagCache = new Hashtable ();
  92
  93                                 SqliteCommand command = new SqliteCommand ();
  94                                 command.Connection = PhotoStoreConnection;
  95                                 command.CommandText = "SELECT id, name, category_id, is_category, sort_priority FROM tags";
  96
  97                                 SqliteDataReader reader = command.ExecuteReader ();
  98                                 while (reader.Read ()) {
  99                                         Tag tag = new Tag ();
 100                                         tag.Id = Convert.ToUInt32 (reader [0]);
 101                                         tag.Name = (string) reader [1];
 102                                         tag.CategoryId = Convert.ToUInt32 (reader [2]);
 103                                         tag.IsCategory = (reader [3] == "1");
 104                                         tag.SortPriority = Convert.ToInt32 (reader [4]);
 105                                         tagCache [tag.Id] = tag;
 106                                 }
 107
 108                                 // Walk across all tags, linking to the category's Tag
 109                                 // object.  Since the tagCache is fully populated, it is
 110                                 // safe to call GetTagById here.
 111                                 foreach (Tag tag in tagCache.Values)
 112                                         tag.Category = GetTagById (tag.CategoryId);
 113
 114                                 command.Dispose ();
 115                         }
 116
 117                         return (Tag) tagCache [id];
 118                 }
 119
 120                 // FIXME: We should expire this cache if the underlying db has changed.
 121                 static bool IsPossibleDirectory (string directory)
 122                 {
 123                         if (! HavePhotoStore)
 124                                 return false;
 125
 126                         if (directoryCache == null) {
 127                                 directoryCache = new Hashtable ();
 128
 129                                 SqliteCommand command = new SqliteCommand ();
 130                                 command.Connection = PhotoStoreConnection;
 131                                 command.CommandText = "SELECT DISTINCT directory_path FROM photos";
 132
 133                                 SqliteDataReader reader = command.ExecuteReader ();
 134                                 while (reader.Read ()) {
 135                                         directoryCache [reader [0]] = true;
 136                                 }
 137
 138                                 command.Dispose ();
 139                         }
 140
 141                         return directoryCache.Contains (directory);
 142                 }
 143
 144                 static public Photo GetPhoto (string path)
 145                 {
 146                         if (! HavePhotoStore)
 147                                 return null;
 148
 149                         path = Path.GetFullPath (path);
 150                         string dir = Path.GetDirectoryName (path);
 151                         string name = Path.GetFileName (path);
 152
 153                         if (! IsPossibleDirectory (dir))
 154                                 return null;
 155
 156                         SqliteCommand command;
 157                         SqliteDataReader reader;
 158
 159                         command = new SqliteCommand ();
 160                         command.Connection = PhotoStoreConnection;
 161                         command.CommandText = String.Format ("SELECT id, description         " +
 162                                                              "FROM photos                    " +
 163                                                              "WHERE directory_path = \"{0}\" " +
 164                                                              "  AND name = \"{1}\"",
 165                                                              dir, name);
 166
 167                         Photo photo = null;
 168                         reader = command.ExecuteReader ();
 169                         if (reader.Read ()) {
 170                                 photo = new Photo ();
 171                                 photo.Path = path;
 172                                 photo.Id = Convert.ToUInt32 (reader [0]);
 173                                 photo.Description = (string) reader [1];
 174                         }
 175
 176                         command.Dispose ();
 177
 178                         if (photo != null) {
 179                                 command = new SqliteCommand ();
 180                                 command.Connection = PhotoStoreConnection;
 181                                 command.CommandText = String.Format ("SELECT tag_id       " +
 182                                                                      "FROM photo_tags     " +
 183                                                                      "WHERE photo_id = {0}",
 184                                                                      photo.Id);
 185
 186                                 Hashtable tagHash = new Hashtable ();
 187
 188                                 // Mark the photo with both any tags and all parents.
 189                                 // Maybe this isn't the right thing to do, but it seems
 190                                 // to most closely mirror the tag semantics implied by f-spot.
 191                                 reader = command.ExecuteReader ();
 192                                 while (reader.Read ()) {
 193                                         uint id = Convert.ToUInt32 (reader [0]);
 194                                         Tag tag = GetTagById (id);
 195                                         while (tag != null) {
 196                                                 tagHash [tag] = tag;
 197                                                 tag = tag.Category;
 198                                         }
 199                                 }
 200
 201                                 photo.Tags = new Tag [tagHash.Count];
 202                                 int i = 0;
 203                                 foreach (Tag t in tagHash.Values) {
 204                                         photo.Tags [i] = t;
 205                                         ++i;
 206                                 }
 207                         }
 208
 209                         return photo;
 210                 }
 211
 212         }
 213 }