forked from denoland/std
-
Notifications
You must be signed in to change notification settings - Fork 0
/
cookie.ts
138 lines (126 loc) · 3.59 KB
/
cookie.ts
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
// Copyright 2018-2019 the Deno authors. All rights reserved. MIT license.
import { ServerRequest, Response } from "./server.ts";
import { assert } from "../testing/asserts.ts";
import { pad } from "../strings/pad.ts";
const SETCOOKIE = "Set-Cookie";
const COOKIE = "Cookie";
export interface Cookie {
[key: string]: string;
}
export interface CookieValue {
name: string;
value: string;
}
export interface CookieOptions {
Expires?: Date;
MaxAge?: number;
Domain?: string;
Path?: string;
Secure?: boolean;
HttpOnly?: boolean;
SameSite?: SameSite;
}
export type SameSite = "Strict" | "Lax";
function cookieStringFormat(cookie: CookieValue, opt: CookieOptions): string {
function dtPad(v: string, lPad: number = 2): string {
return pad(v, lPad, { char: "0" });
}
const out: string[] = [];
out.push(`${cookie.name}=${cookie.value}`);
// Fallback for invalid Set-Cookie
// ref: https://tools.ietf.org/html/draft-ietf-httpbis-cookie-prefixes-00#section-3.1
if (cookie.name.startsWith("__Secure")) {
opt.Secure = true;
}
if (cookie.name.startsWith("__Host")) {
opt.Path = "/";
opt.Secure = true;
delete opt.Domain;
}
if (opt.Secure) {
out.push("Secure");
}
if (opt.HttpOnly) {
out.push("HttpOnly");
}
if (Number.isInteger(opt.MaxAge)) {
assert(opt.MaxAge > 0, "Max-Age must be an integer superior to 0");
out.push(`Max-Age=${opt.MaxAge}`);
}
if (opt.Domain) {
out.push(`Domain=${opt.Domain}`);
}
if (opt.SameSite) {
out.push(`SameSite=${opt.SameSite}`);
}
if (opt.Path) {
out.push(`Path=${opt.Path}`);
}
if (opt.Expires) {
let dateString = "";
let d = dtPad(opt.Expires.getUTCDate().toString());
const h = dtPad(opt.Expires.getUTCHours().toString());
const min = dtPad(opt.Expires.getUTCMinutes().toString());
const s = dtPad(opt.Expires.getUTCSeconds().toString());
const y = opt.Expires.getUTCFullYear();
// See Date format: https://tools.ietf.org/html/rfc7231#section-7.1.1.1
const days = ["Sun", "Mon", "Tue", "Wed", "Thus", "Fri", "Sat"];
const months = [
"Jan",
"Feb",
"Mar",
"May",
"Jun",
"Jul",
"Aug",
"Sep",
"Oct",
"Nov",
"Dec"
];
dateString += `${days[opt.Expires.getDay()]}, ${d} ${
months[opt.Expires.getUTCMonth()]
} ${y} ${h}:${min}:${s} GMT`;
out.push(`Expires=${dateString}`);
}
return out.join("; ");
}
/* Parse the cookie of the Server Request */
export function getCookie(rq: ServerRequest): Cookie {
if (rq.headers.has(COOKIE)) {
const out: Cookie = {};
const c = rq.headers.get(COOKIE).split(";");
for (const kv of c) {
const cookieVal = kv.split("=");
const key = cookieVal.shift().trim();
out[key] = cookieVal.join("");
}
return out;
}
return {};
}
/* Set the cookie header properly in the Response */
export function setCookie(
res: Response,
cookie: CookieValue,
opt: CookieOptions = {}
): void {
if (!res.headers) {
res.headers = new Headers();
}
// TODO (zekth) : Add proper parsing of Set-Cookie headers
// Parsing cookie headers to make consistent set-cookie header
// ref: https://tools.ietf.org/html/rfc6265#section-4.1.1
res.headers.set(SETCOOKIE, cookieStringFormat(cookie, opt));
}
/* Set the cookie header properly in the Response to delete it */
export function delCookie(res: Response, CookieName: string): void {
if (!res.headers) {
res.headers = new Headers();
}
const c: CookieValue = {
name: CookieName,
value: ""
};
res.headers.set(SETCOOKIE, cookieStringFormat(c, { Expires: new Date(0) }));
}