Rewrite URLs for Salesforce Sites

1global class yourClass implements Site.UrlRewriter {
2    global PageReference mapRequestUrl(PageReference
3            yourFriendlyUrl)
4    global PageReference[] generateUrlFor(PageReference[]
5            yourSalesforceUrls);
6}

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