Chameleon Applications

Chameleon Applications Svn Source Tree

Root/trunk/ChameleonPrefPane/Sources/BootSetupController.mm

1//␊
2// BootSetupController.mm␊
3// ChameleonPrefPane␊
4//␊
5// Created by Rekursor on 1/22/10.␊
6//␊
7␊
8#import "BootSetupController.h"␊
9#import "ChameleonPrefPane.h"␊
10#import "TableViewsController.h"␊
11#import "ShellProcess.h"␊
12#import "PartitionInfoManager.h"␊
13#include "ChameleonPropertyList.h"␊
14␊
15static const char cPartDescSep = ';'; // partition descriptor separator␊
16␊
17static BootSetupController *gInstance = NULL;␊
18␊
19@implementation BootSetupController␊
20␊
21//--------------------------------------------------------------------------␊
22- (id) init␊
23{␊
24␉self = [super init];␊
25␉return (gInstance = self);␊
26}␉␊
27␊
28//--------------------------------------------------------------------------␊
29- (void) addOptionsDesc␊
30{␊
31 BootProp::instance().addOptionDesc(mDefaultPartition, mDefaultPartitionText, OptionString, "Default Partition", "");␊
32 BootProp::instance().addOptionDesc(mHidePartition , mHidePartitionText, OptionString, "Hide Partition", "");␊
33 BootProp::instance().addOptionDesc(mRenamePartition , mRenamePartitionText, OptionString, "Rename Partition", "");␊
34}␊
35␊
36//--------------------------------------------------------------------------␊
37- (void) refreshLockStates␊
38{␊
39␉// non boot config options (not automatized)␊
40 [PreferencesControllerBase refreshLockState: mSwapHD01 ];␊
41 [PreferencesControllerBase refreshLockState: mSwapHD02 ];␊
42 [PreferencesControllerBase refreshLockState: mFreezeParts ];␊
43 [PreferencesControllerBase refreshLockState: mInjectFrozenParts ];␊
44 [PreferencesControllerBase refreshLockState: mBootConfigPath ];␊
45 [PreferencesControllerBase refreshLockState: mBootConfigPathText ];␊
46}␊
47␊
48//--------------------------------------------------------------------------␊
49- (void) setDefaultsValues: (NSMutableDictionary*) dict␊
50{␊
51␉[dict setObject:[[[NSNumber alloc] initWithBool: false] autorelease] forKey: keySwapHD01];␊
52␉[dict setObject:[[[NSNumber alloc] initWithBool: false] autorelease] forKey: keySwapHD02];␊
53␉[dict setObject:[[[NSNumber alloc] initWithBool: false] autorelease] forKey: keyUseFrozenParts];␊
54␉[dict setObject:[[[NSNumber alloc] initWithBool: false] autorelease] forKey: keyUseFrozenParts];␊
55␉[dict setObject:[[[NSString alloc] initWithString: @""] autorelease] forKey: keyForceBootConfigPath];␊
56}␊
57//--------------------------------------------------------------------------␊
58-(void) loadOptionsFromPreferencesFile: (NSMutableDictionary*) dict␊
59{␊
60␉[mSwapHD01 setIntValue: [[dict objectForKey:␉keySwapHD01] intValue]];␊
61␉[mSwapHD02 setIntValue: [[dict objectForKey:␉keySwapHD02] intValue]];␊
62␉[mFreezeParts setIntValue: [[dict objectForKey: keyUseFrozenParts] intValue] ];␊
63␉id obj = [dict objectForKey: keyForceBootConfigPath];␊
64␉[mBootConfigPathText setStringValue: obj ? obj : @"" ];␊
65␉int val = (obj && [[mBootConfigPathText stringValue] length] >0 ? 1 : 0);␊
66␉[mBootConfigPath setIntValue: val];␊
67␉[dict setObject:[NSString stringWithString: [mBootConfigPathText stringValue]] ␊
68␉␉␉ forKey: keyForceBootConfigPath];␊
69␉[mBootConfigPathText setEnabled: val ? true : false]; ␊
70␉[mBootConfigPathText setEditable: val ? true : false];␊
71␉[mBootExtraPathText setStringValue: [[mBootConfigPathText stringValue] stringByDeletingLastPathComponent]];␊
72␉[mBootCdbootPathText setStringValue: [[mBootConfigPathText stringValue] stringByDeletingLastPathComponent]];␊
73}␊
74␊
75//--------------------------------------------------------------------------␊
76- (void) loadFrozenParts␊
77{␊
78␉// iterate for all entries to add␊
79␉char keyPartN[32] = ""; ␊
80␉int k=0;␊
81␉␊
82␉NSMutableArray* partList = (NSMutableArray*) [PartsInfoMgr parts];␊
83␉[partList removeAllObjects];␊
84␉␊
85␉for (int i=0; i< [[self preferencesParts] count]; i++)␊
86␉{␊
87␉␉// get the key␊
88␉␉snprintf(keyPartN, sizeof(keyPartN)-1, "partition%02d",i);␊
89␉␉NSString* obj = [[self preferencesParts] objectForKey: [NSString stringWithUTF8String: keyPartN] ];␊
90␉␉// assign this key if valid␊
91␉␉if (obj!=nil && [obj length]>0)␊
92␉␉{␊
93␉␉␉PartitionInfoElement* p = [[[PartitionInfoElement alloc] init] autorelease];␊
94␉␉␉// parse string␊
95␉␉␉NSArray* words = [obj componentsSeparatedByCharactersInSet: [NSCharacterSet characterSetWithCharactersInString: @";"]];␊
96␉␉␉k=0;␊
97␉␉␉int diskNum=0, partNum=0;␊
98␉␉␉for (NSString * word in words)␊
99␉␉␉{␊
100␉␉␉␉switch (k) {␊
101␉␉␉␉␉case 0: // disk number␊
102␉␉␉␉␉␉diskNum = [word intValue] ;␊
103␉␉␉␉␉␉[ p setDiskNumber:diskNum];␊
104␉␉␉␉␉␉break;␊
105␉␉␉␉␉case 1: // partition number␊
106␉␉␉␉␉␉partNum = [word intValue] ;␊
107␉␉␉␉␉␉[ p setPartitionNumber: partNum];␉␉␉␉␉␉␉␊
108␉␉␉␉␉␉break;␊
109␉␉␉␉␉case 2: // volume label␊
110␉␉␉␉␉␉[p setVName: word];␊
111␉␉␉␉␉␉[p setVAliasName: word];␊
112␉␉␉␉␉␉break;␊
113␉␉␉␉␉case 3: // fstype␊
114␉␉␉␉␉␉[p setVKind: word];␊
115␉␉␉␉␉␉break;␊
116␉␉␉␉␉case 4: // vUUID␊
117␉␉␉␉␉␉[p setVUUID: word];␊
118␉␉␉␉␉␉break;␊
119␉␉␉␉␉default:␊
120␉␉␉␉␉␉break;␊
121␉␉␉␉}␊
122␉␉␉␉k++;␊
123␉␉␉}␊
124␉␉␉␊
125␉␉␉[partList addObject:p];␊
126␉␉}␊
127␉}␊
128␉// for mgr to refresh hide and rename attributes␊
129␉[PartsInfoMgr hideParts:nil];␊
130␉[PartsInfoMgr renameParts:nil];␊
131␉␊
132}␊
133␊
134//--------------------------------------------------------------------------␊
135- (void) swapDisks: (bool) bSwap src: (int) iSrc dst: (int) iDst;␊
136{␊
137␉//if(!BootProp::instance().isValid()) return;␊
138␉if ([mFreezeParts intValue]==0) ␊
139␉␉[PartsInfoMgr reload];␊
140␉else ␊
141␉␉[self loadFrozenParts ];␊
142␉␊
143␉if (bSwap)␊
144␉{␊
145␉␉[PartsInfoMgr swapHD:iSrc with: iDst ];␉␊
146␉}␊
147␉␊
148␉[ [self chameleon] selectDefaultPartition];␊
149␉␊
150}␊
151␊
152//--------------------------------------------------------------------------␊
153- (void) doSwapHD: (int) val save: (bool) doSave src: (int) isrc dst: (int) idst␊
154{␊
155␉␊
156␉if( val>0 && ![[BootSetupController instance]->mFreezeParts intValue]) //on␊
157␉{␊
158␉␉[self swapDisks: true src:isrc dst:idst ];␊
159␉}␊
160␉else␊
161␉{␊
162␉␉[self swapDisks: false src:isrc dst:idst ];␊
163␉}␊
164␉␊
165␉if(doSave)␊
166␉{␊
167␉␉if (isrc==0 && idst==1)␊
168␉␉␉[[self preferencesFile] setObject: [NSNumber numberWithBool: val ? true : false] forKey: keySwapHD01];␊
169␉␉if (isrc==0 && idst==2)␊
170␉␉␉[[self preferencesFile] setObject: [NSNumber numberWithBool: val ? true : false] forKey: keySwapHD02];␊
171␉␉[ [self chameleon] savePreferences:[self preferencesFile] ];␊
172␉}␊
173␉␊
174␉[[[TableViewsController instance] partitionsTable] reloadData];␊
175␉[[[TableViewsController instance] partitionsTable] scrollRowToVisible: 0];␊
176␉//[self tableViewSelectionDidChange: nil];␊
177␉␊
178}␊
179␊
180␊
181//--------------------------------------------------------------------------␊
182- (IBAction)onInjectPartsToFreeze: (id)sender␊
183{␊
184␉int size = [[PartsInfoMgr parts] count ];␊
185␉NSString * title = GetLocStrDef(@"Freeze_Dialog", @"msgbox", @"Inject Partitions to Freeze Configuration");␊
186␉if (!size) ␊
187␉{ // nothing to inject␊
188␉␉NSRunAlertPanel(title, ␊
189␉␉␉␉␉␉GetLocStrDef(@"Freeze_NoPart", @"msgbox", @"No current partitions to inject, did you check boot config file ?"),␊
190␉␉␉␉␉␉@"OK", nil,nil);␊
191␉␉return;␊
192␉}␊
193␉// generate the parts list in preferences proplist␊
194␉NSInteger n = NSRunAlertPanel(title, ␊
195␉␉␉␉␉␉␉␉ GetLocStrDef(@"Freeze_Overwrite", @"msgbox",␊
196␉␉␉␉␉␉␉␉␉␉␉ @"Are you sure you want to overwrite your Freeze settings with current partition list ?"),␊
197␉␉␉␉␉␉␉␉ @"OK", @"Cancel",nil);␊
198␉if (n==1)␊
199␉{␊
200␉␉// populate the dictionary with the current partitions␊
201␉␉␊
202␉␉// empty dictionary and update key in parent:␊
203␉␉[[self preferencesFile] removeObjectForKey: keyPartitionsList];␊
204␉␉[[self preferencesParts] removeAllObjects ];␊
205␉␉␊
206␉␉// iterate for all entries to add␊
207␉␉char partDesc[256]="";␊
208␉␉char keyPartN[32] = "";␊
209␉␉for (int i=0; i< size; i++)␊
210␉␉{␊
211␉␉␉␊
212␉␉␉PartitionInfoElement* p = [PartsInfoMgr objectAtIndex:i];␊
213␉␉␉␊
214␉␉␉// format the partition key and descriptor string␊
215␉␉␉snprintf(keyPartN, sizeof(keyPartN)-1, "partition%02d",i);␊
216␉␉␉␊
217␉␉␉snprintf(partDesc, sizeof(partDesc)-1, "%d%c%d%c%s%c%s%c%s",␊
218␉␉␉␉␉ [p diskNumber], cPartDescSep,␊
219␉␉␉␉␉ [p partitionNumber], cPartDescSep,␊
220␉␉␉␉␉ [[p vAliasName] UTF8String], cPartDescSep,␊
221␉␉␉␉␉ [[p vKind] UTF8String], cPartDescSep, ␊
222␉␉␉␉␉ [[p vUUID] UTF8String]);␊
223␉␉␉␊
224␉␉␉// write it to the dictionary␊
225␉␉␉NSString * key = [NSString stringWithUTF8String: keyPartN];␊
226␉␉␉NSString * desc = [NSString stringWithUTF8String: partDesc]; ␊
227␉␉␉[[self preferencesParts] setObject: desc forKey: key];␊
228␉␉}␊
229␉␉[[self preferencesFile] setObject: [self preferencesParts] forKey: keyPartitionsList];␊
230␉␉[self savePreferences ];␊
231␉}␊
232}␊
233␊
234//--------------------------------------------------------------------------␊
235- (IBAction) onForceBootConfigPath: (id) sender␊
236{␊
237 NSString * f;␊
238 ␊
239␉if (sender == mBootConfigPath)␊
240␉{␊
241␉␉int val = [mBootConfigPath intValue];␊
242␉␉[mBootConfigPathText setEnabled: val ? true : false]; ␊
243␉␉[mBootConfigPathText setEditable: val ? true : false];␊
244␉␉␊
245␉␉if (val) ␊
246␉␉{␊
247␉␉␉f = [self selectPlistFile: @"org.chameleon.Boot.plist"];␊
248␉␉␉if(f)␊
249␉␉␉{␊
250␉␉␉␉[mBootConfigPathText setStringValue:f];␊
251␉␉␉␉[mBootExtraPathText setStringValue: [f stringByDeletingLastPathComponent] ];␊
252␉␉␉␉[mBootCdbootPathText setStringValue: [f stringByDeletingLastPathComponent]];␊
253␉␉␉}␊
254␉␉}␊
255␉␉␉␊
256␊
257␉␉[[self preferencesFile] setObject: ␊
258␉␉ val ? [mBootConfigPathText stringValue] : [NSString stringWithUTF8String: ""] ␊
259␉␉␉␉␉␉␉␉ forKey: keyForceBootConfigPath];␊
260␉␉␊
261␉}␊
262␉else{␊
263␉␉f = [mBootConfigPathText stringValue];␊
264␉␉[[self preferencesFile] setObject:f forKey: keyForceBootConfigPath];␊
265␉␉[mBootExtraPathText setStringValue: [f stringByDeletingLastPathComponent] ];␊
266␉␉[mBootCdbootPathText setStringValue: [f stringByDeletingLastPathComponent]];␊
267␉}␊
268␉[self savePreferences ];␊
269␉BootProp::instance().cleanup();␊
270␉[[ChameleonPrefPane instance ] initBootConfig];␊
271}␊
272//--------------------------------------------------------------------------␊
273-(IBAction) onCheckButtonChange: (NSButton*) sender␊
274{␊
275␉// IMPROVE ME: Should automatize the callback/load/save mechanism ␊
276␉// for the preferences file like the bootConfig file␊
277␉if (sender == mSwapHD01 || sender == mSwapHD02)␊
278␉{␊
279␉␉[PartsInfoMgr resetSwapping];␊
280␉␉[self doSwapHD: [mSwapHD01 intValue] save:true src:0 dst:1];␊
281␉␉[self doSwapHD: [mSwapHD02 intValue] save:true src:0 dst:2];␊
282␉}␊
283␉else if (sender == mFreezeParts)␊
284␉{␊
285␉␉bool val = !![sender intValue];␊
286␉␉[[self preferencesFile] setObject: [NSNumber numberWithBool: val] forKey: keyUseFrozenParts];␊
287␉␉[[self chameleon] savePreferences: [self preferencesFile]];␊
288␉␉[ self onCheckButtonChange: mSwapHD01];␊
289␉}␊
290␉else if (sender == mInjectFrozenParts)␊
291␉{␊
292␉␉[self onInjectPartsToFreeze: mInjectFrozenParts];␊
293␉}␉␊
294␉else if (sender == mDefaultPartition || sender == mHidePartition␊
295␉␉␉ || sender == mRenamePartition )␊
296␉{ // sync with other panels␊
297␉␉[self handleSender:sender];␊
298␉}␉␊
299␉else if (sender == mBootConfigPath || (NSTextField*)sender == mBootConfigPathText)␊
300␉{ // sync with other panels␊
301␉␉[self onForceBootConfigPath: sender];␊
302␉}␉␊
303␉else if (sender == mBootExtraPath || (NSTextField*)sender == mBootExtraPathText)␊
304␉{ // sync with other panels␊
305␉␉[self onForceBootConfigPath: sender];␊
306␉}␉␊
307␉else if (sender == mBootCdbootPath || (NSTextField*)sender == mBootCdbootPathText)␊
308␉{ // sync with other panels␊
309␉␉[self onForceBootConfigPath: sender];␊
310␉}␉␊
311␉// Handle BootOptions generically:␊
312␉else ␊
313␉␉[self handleSender:sender];␊
314}␊
315//--------------------------------------------------------------------------␊
316-(IBAction) onTextFiedChange: (NSTextField*) sender␊
317{␊
318␉if ( sender == mDefaultPartitionText || sender == mHidePartitionText ||␊
319␉␉␉ sender == mRenamePartitionText )␊
320␉{␊
321␉␉[self handleSender:sender];␊
322␊
323␉␉[PreferencesControllerBase loadOptionsFromBootFile ];␊
324␉␉[PartsInfoMgr reloadWithHideSpec: [mHidePartitionText stringValue] andRenameSpec: [mRenamePartitionText stringValue] ];␊
325␊
326␉␉[self doSwapHD: [mSwapHD01 intValue] save:true src:0 dst:1];␊
327␉␉[self doSwapHD: [mSwapHD02 intValue] save:true src:0 dst:2];␊
328␉}␊
329␉else if (sender == mBootConfigPathText)␊
330␉{␊
331␉␉[self onForceBootConfigPath: sender];␊
332␉}␊
333}␊
334//--------------------------------------------------------------------------␊
335- (IBAction) onCdBootConfigPath: (id) sender␊
336{␊
337␉if ( sender == mBootCdbootPath)␊
338␉{␊
339␉␉NSString * dir = [self selectDirectory:@"/"];␊
340␉␉if (dir!=nil)␊
341␉␉␉[mBootCdbootPathText setStringValue: dir];␊
342␉}␊
343␉␉␉␊
344}␊
345//--------------------------------------------------------------------------␊
346- (IBAction) onExtraConfigPath: (id) sender␊
347{␊
348␉if ( sender == mBootExtraPath)␊
349␉{␊
350␉␉NSString * dir = [self selectDirectory:@"/"];␊
351␉␉if (dir!=nil)␊
352␉␉␉[mBootExtraPathText setStringValue: dir];␊
353␉}␊
354}␊
355␊
356//--------------------------------------------------------------------------␊
357- (IBAction) onCreateBootCD: (id) sender␊
358{␊
359␉// using rek modified rohan script from Lizard boot cd creator:␊
360␉NSString* destination = @"~/Desktop/Chameleon";␊
361␉NSString* pathTask = [self getResourcePath:@"cdBootCreator" ofType:@"sh"];␊
362␉NSString* pathExtra = [mBootExtraPathText stringValue];␊
363␉NSString* pathCdboot = [mBootCdbootPathText stringValue];␊
364␉NSMutableArray* arr = [[NSMutableArray alloc] init];␊
365␉bool result;␊
366␉␊
367␉if ( ␊
368␉␉pathTask!=nil && [pathTask length] > 0 ␊
369␉␉&& pathExtra !=nil && [pathExtra length] > 0 ␊
370␉␉&& pathCdboot !=nil && [pathCdboot length] > 0 ␊
371␉ )␊
372␉{␊
373␉␉[arr addObject: pathExtra];␊
374␉␉[arr addObject: pathCdboot];␉␊
375␉␉result = [self executeTaskAndWaitForTermination: pathTask withArgs: arr];␊
376␉␉if (result)␊
377␉␉{␊
378␉␉␉[self msgBoxInfo:␊
379␉␉␉ [NSString stringWithFormat: ␊
380␉␉␉ GetLocStrDef(@"CD_Success", @"msgbox", @"Boot cd creator script successfully executed. BootCD.iso created in %@."), ␊
381␉␉␉ destination]];␊
382␉␉}␊
383␉␉else␊
384␉␉␉[self msgBoxError:␊
385␉␉␉␉GetLocStrDef(@"CD_Failure", @"msgbox", @"ERROR: Couldn't execute boot cd creator script successfully!")];␊
386␉}␊
387␉␊
388␉[arr release];␊
389}␊
390//--------------------------------------------------------------------------␊
391+ (BootSetupController *)instance { return(gInstance);}␊
392␊
393@end␊
394

Archive Download this file

Revision: 451