1
import type {
2
  Period,
3
  SiteLicenseDescriptionDB,
4
} from "@cocalc/util/upgrades/shopping";
5
import type { PurchaseInfo, StartEndDates, Subscription } from "./types";
6
import { CURRENT_VERSION } from "./consts";
7
import type { Date0 } from "@cocalc/util/types/store";
8
import dayjs from "dayjs";
9

10
// this ALWAYS returns purchaseInfo that is the *current* version.
11
export default function getPurchaseInfo(
12
  conf: SiteLicenseDescriptionDB,
13
): PurchaseInfo {
14
  conf.type = conf.type ?? "quota"; // backwards compatibility
15

16
  const { title, description } = conf;
17

18
  switch (conf.type) {
19
    case "quota":
20
      const {
21
        type,
22
        user,
23
        run_limit,
24
        period,
25
        ram,
26
        cpu,
27
        disk,
28
        member,
29
        uptime,
30
        boost = false,
31
      } = conf;
32
      return {
33
        version: CURRENT_VERSION,
34
        type, // "quota"
35
        user,
36
        upgrade: "custom" as "custom",
37
        quantity: run_limit,
38
        subscription: (period == "range" ? "no" : period) as Subscription,
39
        ...fixRange(conf.range, conf.period),
40
        custom_ram: ram,
41
        custom_dedicated_ram: 0,
42
        custom_cpu: cpu,
43
        custom_dedicated_cpu: 0,
44
        custom_disk: disk,
45
        custom_member: member,
46
        custom_uptime: uptime,
47
        boost,
48
        title,
49
        description,
50
      };
51

52
    case "vm":
53
      return {
54
        version: CURRENT_VERSION,
55
        type: "vm",
56
        quantity: 1,
57
        dedicated_vm: conf.dedicated_vm,
58
        subscription: "no",
59
        ...fixRange(conf.range, conf.period),
60
        title,
61
        description,
62
      };
63

64
    case "disk":
65
      return {
66
        version: CURRENT_VERSION,
67
        type: "disk",
68
        quantity: 1,
69
        dedicated_disk: conf.dedicated_disk,
70
        subscription: conf.period,
71
        title,
72
        description,
73
        ...fixRange(null, conf.period),
74
      };
75
  }
76
}
77

78
// Make sure both start and end dates defined as Date.  For all licenses we
79
// always require that both are defined.  For a subscription, the end date must
80
// be defined, but it will get periodically moved forward as the subscription
81
// is updated.
82
// Later, when actually saving the range to the database, we will maybe append
83
// a portion of the start which is in the past.
84
export function fixRange(
85
  rangeOrig: readonly [Date0 | string, Date0 | string] | undefined | null,
86
  period: Period,
87
  noRangeShift?: boolean,
88
): StartEndDates {
89
  if (period != "range" && !noRangeShift) {
90
    // ATTN! -- we messed up and didn't deal with this case before, and a user
91
    // could in theory:
92
    //  1. set the period to 'range', and put in a week period via start and end
93
    //  2. set the period to 'yearly'.
94
    // Then rangeOrig is still a week, so they pay for one week instead of one year!
95
    // Instead, in whenever the period is 'monthly' or 'yearly' (anything but 'range',
96
    // we unset rangeOrig here so we use start=now and end=now + a year (say) for
97
    // the price computation.
98
    rangeOrig = null;
99
  }
100
  const now = new Date();
101
  if (rangeOrig == null) {
102
    if (period == "range") {
103
      throw Error(
104
        "if period is 'range', then start and end dates must be explicitly given",
105
      );
106
    }
107
    // we expand the dates to be as inclusive as possible for subscriptions, since
108
    // that doesn't result in any more charge to the user.
109
    return {
110
      start: dayjs(now).startOf("day").toDate(),
111
      end: dayjs(addPeriod(now, period)).endOf("day").toDate(),
112
    };
113
  }
114

115
  return {
116
    start: rangeOrig?.[0] ? new Date(rangeOrig?.[0]) : now,
117
    end: rangeOrig?.[1] ? new Date(rangeOrig?.[1]) : addPeriod(now, period),
118
  };
119
}
120

121
function addPeriod(date: Date, period: Period): Date {
122
  if (period == "range") {
123
    throw Error("period must not be range");
124
  } else if (period == "monthly") {
125
    return dayjs(date).add(1, "month").toDate();
126
  } else if (period == "yearly") {
127
    return dayjs(date).add(1, "year").toDate();
128
  } else {
129
    throw Error(`unsupported period ${period}`);
130
  }
131
}
132

133