summaryrefslogtreecommitdiffstats
path: root/testing/web-platform/tests/service-workers/service-worker/resources/redirect-worker.js
blob: 82e21fc26fdb94236020df968eb9fa3b286d9e19 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
// We store an empty response for each fetch event request we see
// in this Cache object so we can get the list of urls in the
// message event.
var cacheName = 'urls-' + self.registration.scope;

var waitUntilPromiseList = [];

// Sends the requests seen by this worker. The output is:
// {
//   requestInfos: [
//     {url: url1, resultingClientId: id1},
//     {url: url2, resultingClientId: id2},
//   ]
// }
async function getRequestInfos(event) {
  // Wait for fetch events to finish.
  await Promise.all(waitUntilPromiseList);
  waitUntilPromiseList = [];

  // Generate the message.
  const cache = await caches.open(cacheName);
  const requestList = await cache.keys();
  const requestInfos = [];
  for (let i = 0; i < requestList.length; i++) {
    const response = await cache.match(requestList[i]);
    const body = await response.json();
    requestInfos[i] = {
      url: requestList[i].url,
      resultingClientId: body.resultingClientId
    };
  }
  await caches.delete(cacheName);

  event.data.port.postMessage({requestInfos});
}

// Sends the results of clients.get(id) from this worker. The
// input is:
// {
//   actual_ids: {a: id1, b: id2, x: id3}
// }
//
// The output is:
// {
//   clients: {
//     a: {found: false},
//     b: {found: false},
//     x: {
//       id: id3,
//       url: url1,
//       found: true
//    }
//   }
// }
async function getClients(event) {
  // |actual_ids| is like:
  // {a: id1, b: id2, x: id3}
  const actual_ids = event.data.actual_ids;
  const result = {}
  for (let key of Object.keys(actual_ids)) {
    const id = actual_ids[key];
    const client = await self.clients.get(id);
    if (client === undefined)
      result[key] = {found: false};
    else
      result[key] = {found: true, url: client.url, id: client.id};
  }
  event.data.port.postMessage({clients: result});
}

self.addEventListener('message', async function(event) {
  if (event.data.command == 'getRequestInfos') {
    event.waitUntil(getRequestInfos(event));
    return;
  }

  if (event.data.command == 'getClients') {
    event.waitUntil(getClients(event));
    return;
  }
});

function get_query_params(url) {
  var search = (new URL(url)).search;
  if (!search) {
    return {};
  }
  var ret = {};
  var params = search.substring(1).split('&');
  params.forEach(function(param) {
      var element = param.split('=');
      ret[decodeURIComponent(element[0])] = decodeURIComponent(element[1]);
    });
  return ret;
}

self.addEventListener('fetch', function(event) {
    var waitUntilPromise = caches.open(cacheName).then(function(cache) {
      const responseBody = {};
      responseBody['resultingClientId'] = event.resultingClientId;
      const headers = new Headers({'Content-Type': 'application/json'});
      const response = new Response(JSON.stringify(responseBody), {headers});
      return cache.put(event.request, response);
    });
    event.waitUntil(waitUntilPromise);

    var params = get_query_params(event.request.url);
    if (!params['sw']) {
      // To avoid races, add the waitUntil() promise to our global list.
      // If we get a message event before we finish here, it will wait
      // these promises to complete before proceeding to read from the
      // cache.
      waitUntilPromiseList.push(waitUntilPromise);
      return;
    }

    event.respondWith(waitUntilPromise.then(async () => {
      if (params['sw'] == 'gen') {
        return Response.redirect(params['url']);
      } else if (params['sw'] == 'gen-manual') {
        // Note this differs from Response.redirect() in that relative URLs are
        // preserved.
        return new Response("", {
          status: 301,
          headers: {location: params['url']},
        });
      } else if (params['sw'] == 'fetch') {
        return fetch(event.request);
      } else if (params['sw'] == 'fetch-url') {
        return fetch(params['url']);
      } else if (params['sw'] == 'follow') {
        return fetch(new Request(event.request.url, {redirect: 'follow'}));
      } else if (params['sw'] == 'manual') {
        return fetch(new Request(event.request.url, {redirect: 'manual'}));
      } else if (params['sw'] == 'manualThroughCache') {
        const url = event.request.url;
        await caches.delete(url)
        const cache = await self.caches.open(url);
        const response = await fetch(new Request(url, {redirect: 'manual'}));
        await cache.put(event.request, response);
        return cache.match(url);
      }
      // unexpected... trigger an interception failure
    }));
  });