add more override methods that handle project files

[jSite2.git] / src / net / pterodactylus / jsite / project / Project.java

diff --git a/src/net/pterodactylus/jsite/project/Project.java b/src/net/pterodactylus/jsite/project/Project.java
index 305e8ef..65f28a1 100644 (file)
--- a/src/net/pterodactylus/jsite/project/Project.java
+++ b/src/net/pterodactylus/jsite/project/Project.java
@@ -250,6 +250,18 @@ public class Project extends AbstractBean {
        /**
         * Adds a file override for the given file.
         * 
+        * @param projectFile
+        *            The file
+        * @param override
+        *            The override for the file
+        */
+       public void addFileOverride(ProjectFile projectFile, FileOverride override) {
+               addFileOverride(projectFile.getCompletePath(), override);
+       }
+
+       /**
+        * Adds a file override for the given file.
+        * 
         * @param filePath
         *            The file path
         * @param override
@@ -262,6 +274,16 @@ public class Project extends AbstractBean {
        /**
         * Removes the file override for the given file.
         * 
+        * @param projectFile
+        *            The file for which to remove the override
+        */
+       public void removeFileOverride(ProjectFile projectFile) {
+               removeFileOverride(projectFile.getCompletePath());
+       }
+
+       /**
+        * Removes the file override for the given file.
+        * 
         * @param filePath
         *            The file path for which to remove the override
         */
@@ -278,7 +300,7 @@ public class Project extends AbstractBean {
         *         not have an override
         */
        public FileOverride getFileOverride(ProjectFile projectFile) {
-               return fileOverrides.get(projectFile.getCompletePath());
+               return getFileOverride(projectFile.getCompletePath());
        }
 
        /**