Rewriting URLs for Force.com Sites

1swfobject.registerObject("clippy.codeblock-5", "9");
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17global with sharing class myRewriter implements Site.UrlRewriter {
18
19    //Variables to represent the user-friendly URLs for
20    //account and contact pages
21    String ACCOUNT_PAGE = '/myaccount/';
22    String CONTACT_PAGE = '/mycontact/';
23    //Variables to represent my custom Visualforce pages
24    //that display account and contact information
25    String ACCOUNT_VISUALFORCE_PAGE = '/myaccount?id=';
26    String CONTACT_VISUALFORCE_PAGE = '/mycontact?id=';
27
28    global PageReference mapRequestUrl(PageReference
29            myFriendlyUrl){
30        String url = myFriendlyUrl.getUrl();
31
32        if(url.startsWith(CONTACT_PAGE)){
33            //Extract the name of the contact from the URL
34            //For example: /mycontact/Ryan returns Ryan
35            String name = url.substring(CONTACT_PAGE.length(),
36                    url.length());
37
38            //Select the ID of the contact that matches
39            //the name from the URL
40            Contact con = [SELECT Id FROM Contact WHERE Name =:
41                    name LIMIT 1];
42
43            //Construct a new page reference in the form
44            //of my Visualforce page
45            return new PageReference(CONTACT_VISUALFORCE_PAGE + con.id);
46        }
47        if(url.startsWith(ACCOUNT_PAGE)){
48            //Extract the name of the account
49            String name = url.substring(ACCOUNT_PAGE.length(),
50                    url.length());
51
52            //Query for the ID of an account with this name
53            Account acc = [SELECT Id FROM Account WHERE Name =:name LIMIT 1];
54
55           //Return a page in Visualforce format
56            return new PageReference(ACCOUNT_VISUALFORCE_PAGE + acc.id);
57        }
58        //If the URL isn't in the form of a contact or
59        //account page, continue with the request
60        return null;
61    }
62    global List<PageReference> generateUrlFor(List<PageReference> 
63            mySalesforceUrls){
64        //A list of pages to return after all the links 
65        //have been evaluated
66        List<PageReference> myFriendlyUrls = new List<PageReference>();
67        
68        //a list of all the ids in the urls
69        List<id> accIds = new List<id>();
70        
71        // loop through all the urls once, finding all the valid ids
72        for(PageReference mySalesforceUrl : mySalesforceUrls){
73        //Get the URL of the page
74        String url = mySalesforceUrl.getUrl();
75 
76            //If this looks like an account page, transform it
77            if(url.startsWith(ACCOUNT_VISUALFORCE_PAGE)){
78                //Extract the ID from the query parameter
79                //and store in a list
80                //for querying later in bulk.
81                        String id= url.substring(ACCOUNT_VISUALFORCE_PAGE.length(),
82                        url.length());
83                        accIds.add(id);
84            }
85        }
86
87    // Get all the account names in bulk
88    List <account> accounts = [SELECT Name FROM Account WHERE Id IN :accIds];
89
90    // make the new urls
91    Integer counter = 0;
92
93    // it is important to go through all the urls again, so that the order
94    // of the urls in the list is maintained. 
95    for(PageReference mySalesforceUrl : mySalesforceUrls) {
96
97       //Get the URL of the page
98       String url = mySalesforceUrl.getUrl();
99
100       if(url.startsWith(ACCOUNT_VISUALFORCE_PAGE)){
101         myFriendlyUrls.add(new PageReference(ACCOUNT_PAGE + accounts.get(counter).name));
102         counter++;
103       } else {
104         //If this doesn't start like an account page,
105         //don't do any transformations
106         myFriendlyUrls.add(mySalesforceUrl);
107       }
108    }
109   
110    //Return the full list of pages
111    return myFriendlyUrls;
112  }
113
114}