source: OniSplit/InstanceFileOperations.cs@ 1117

Last change on this file since 1117 was 1114, checked in by iritscen, 5 years ago

Adding OniSplit source code (v0.9.99.0). Many thanks to Neo for all his work over the years.

File size: 6.4 KB
Line 
1using System;
2using System.Collections.Generic;
3using System.IO;
4
5namespace Oni
6{
7 internal sealed class InstanceFileOperations
8 {
9 private InstanceFileManager fileManager;
10 private string destinationDir;
11 private readonly Dictionary<string, string> fileNames = new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase);
12 private Dictionary<string, string> referencedFiles;
13 private readonly Dictionary<string, string> instances = new Dictionary<string, string>(StringComparer.Ordinal);
14
15 public void Copy(InstanceFileManager fileManager, List<string> sourceFiles, string destinationDir)
16 {
17 Initialize(fileManager, sourceFiles, destinationDir);
18
19 foreach (KeyValuePair<string, string> pair in referencedFiles)
20 {
21 if (File.Exists(pair.Value))
22 {
23 if (!Utils.AreFilesEqual(pair.Key, pair.Value))
24 Console.WriteLine("File {0} already exists at destination and it is different. File not copied.", pair.Value);
25 }
26 else
27 {
28 File.Copy(pair.Key, pair.Value);
29 }
30 }
31 }
32
33 public void Move(InstanceFileManager fileManager, List<string> sourceFilePaths, string outputDirPath)
34 {
35 Initialize(fileManager, sourceFilePaths, outputDirPath);
36
37 foreach (KeyValuePair<string, string> pair in referencedFiles)
38 {
39 if (File.Exists(pair.Value))
40 {
41 if (Utils.AreFilesEqual(pair.Key, pair.Value))
42 File.Delete(pair.Key);
43 else
44 Console.WriteLine("File {0} already exists at destination and it is different. Source file not moved.", pair.Value);
45 }
46 else
47 {
48 File.Move(pair.Key, pair.Value);
49 }
50 }
51 }
52
53 public void MoveOverwrite(InstanceFileManager fileManager, List<string> sourceFilePaths, string outputDirPath)
54 {
55 Initialize(fileManager, sourceFilePaths, outputDirPath);
56
57 foreach (KeyValuePair<string, string> pair in referencedFiles)
58 {
59 if (File.Exists(pair.Value))
60 File.Delete(pair.Value);
61
62 File.Move(pair.Key, pair.Value);
63 }
64 }
65
66 public void MoveDelete(InstanceFileManager fileManager, List<string> sourceFilePaths, string outputDirPath)
67 {
68 Initialize(fileManager, sourceFilePaths, outputDirPath);
69
70 foreach (KeyValuePair<string, string> pair in referencedFiles)
71 {
72 if (File.Exists(pair.Value))
73 File.Delete(pair.Key);
74 else
75 File.Move(pair.Key, pair.Value);
76 }
77 }
78
79 public void GetDependencies(InstanceFileManager fileManager, List<string> sourceFilePaths)
80 {
81 Initialize(fileManager, sourceFilePaths, null);
82
83 foreach (string filePath in referencedFiles.Keys)
84 {
85 Console.WriteLine(filePath);
86 }
87 }
88
89 private void Initialize(InstanceFileManager fileManager, List<string> inputFiles, string destinationDir)
90 {
91 this.fileManager = fileManager;
92 this.destinationDir = destinationDir;
93 this.referencedFiles = new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase);
94
95 if (destinationDir != null)
96 {
97 if (Directory.Exists(destinationDir))
98 {
99 //
100 // Get a list of existing files to avoid name conflicts due to instance names
101 // that differ only in case.
102 //
103
104 foreach (string existingFilePath in Directory.GetFiles(destinationDir, "*.oni"))
105 {
106 string fileName = Path.GetFileNameWithoutExtension(existingFilePath);
107 string instanceName = Importer.DecodeFileName(existingFilePath);
108
109 fileNames[fileName] = fileName;
110 instances[instanceName] = existingFilePath;
111 }
112 }
113 else
114 {
115 Directory.CreateDirectory(destinationDir);
116 }
117 }
118
119 var sourceFiles = new Dictionary<string, string>(StringComparer.Ordinal);
120 string lastSourceDir = null;
121
122 foreach (string inputFile in inputFiles)
123 {
124 string sourceDir = Path.GetDirectoryName(inputFile);
125
126 if (sourceDir != lastSourceDir)
127 {
128 lastSourceDir = sourceDir;
129 sourceFiles.Clear();
130
131 foreach (string file in Directory.GetFiles(sourceDir, "*.oni"))
132 sourceFiles[Importer.DecodeFileName(file)] = file;
133 }
134
135 GetReferencedFiles(inputFile, sourceFiles);
136 }
137 }
138
139 private void GetReferencedFiles(string sourceFile, Dictionary<string, string> sourceFiles)
140 {
141 AddReferencedFile(sourceFile);
142
143 var instanceFile = fileManager.OpenFile(sourceFile);
144
145 foreach (var descriptor in instanceFile.GetPlaceholders())
146 {
147 string referencedSourceFile;
148
149 if (!sourceFiles.TryGetValue(descriptor.FullName, out referencedSourceFile)
150 || referencedFiles.ContainsKey(referencedSourceFile))
151 continue;
152
153 GetReferencedFiles(referencedSourceFile, sourceFiles);
154 }
155 }
156
157 private void AddReferencedFile(string filePath)
158 {
159 if (referencedFiles.ContainsKey(filePath))
160 return;
161
162 string instanceName = Importer.DecodeFileName(filePath);
163 string destinationFile;
164
165 if (!instances.TryGetValue(instanceName, out destinationFile))
166 {
167 if (destinationDir != null)
168 destinationFile = Path.Combine(destinationDir, Importer.EncodeFileName(instanceName, fileNames) + ".oni");
169 }
170
171 referencedFiles.Add(filePath, destinationFile);
172 }
173 }
174}
Note: See TracBrowser for help on using the repository browser.