sm64-roblox-liberty-prime/tools/UploadAnimations.lua
2022-10-13 03:53:17 -05:00

316 lines
7.1 KiB
Lua

------------------------------------------------------------------------------------------------------------------------------------------------------------------------
-- SHA256
------------------------------------------------------------------------------------------------------------------------------------------------------------------------
--!strict
local band = bit32.band
local bnot = bit32.bnot
local bxor = bit32.bxor
local rrotate = bit32.rrotate
local rshift = bit32.rshift
local primes = {
0x428a2f98,
0x71374491,
0xb5c0fbcf,
0xe9b5dba5,
0x3956c25b,
0x59f111f1,
0x923f82a4,
0xab1c5ed5,
0xd807aa98,
0x12835b01,
0x243185be,
0x550c7dc3,
0x72be5d74,
0x80deb1fe,
0x9bdc06a7,
0xc19bf174,
0xe49b69c1,
0xefbe4786,
0x0fc19dc6,
0x240ca1cc,
0x2de92c6f,
0x4a7484aa,
0x5cb0a9dc,
0x76f988da,
0x983e5152,
0xa831c66d,
0xb00327c8,
0xbf597fc7,
0xc6e00bf3,
0xd5a79147,
0x06ca6351,
0x14292967,
0x27b70a85,
0x2e1b2138,
0x4d2c6dfc,
0x53380d13,
0x650a7354,
0x766a0abb,
0x81c2c92e,
0x92722c85,
0xa2bfe8a1,
0xa81a664b,
0xc24b8b70,
0xc76c51a3,
0xd192e819,
0xd6990624,
0xf40e3585,
0x106aa070,
0x19a4c116,
0x1e376c08,
0x2748774c,
0x34b0bcb5,
0x391c0cb3,
0x4ed8aa4a,
0x5b9cca4f,
0x682e6ff3,
0x748f82ee,
0x78a5636f,
0x84c87814,
0x8cc70208,
0x90befffa,
0xa4506ceb,
0xbef9a3f7,
0xc67178f2,
}
local function toHex(str: string): string
local result = str:gsub(".", function(char)
return string.format("%02x", char:byte())
end)
return result
end
local function toBytes(value: number, length: number)
local str = ""
for i = 1, length do
local rem = value % 256
str = string.char(rem) .. str
value = (value - rem) / 256
end
return str
end
local function digestBlock(msg: string, i: number, hash: { number }, digest: { number })
for j = 1, 16 do
local offset = i + (j - 1) * 4
local a, b, c, d = string.byte(msg, offset, offset + 3)
digest[j] = ((a * 256 + b) * 256 + c) * 256 + d
end
for j = 17, 64 do
local v = digest[j - 15]
local s0 = bxor(rrotate(v, 7), rrotate(v, 18), rshift(v, 3))
v = digest[j - 2]
digest[j] = digest[j - 16] + s0 + digest[j - 7] + bxor(rrotate(v, 17), rrotate(v, 19), rshift(v, 10))
end
local a, b, c, d, e, f, g, h = table.unpack(hash)
for i = 1, 64 do
local s0 = bxor(rrotate(a, 2), rrotate(a, 13), rrotate(a, 22))
local maj = bxor(band(a, b), band(a, c), band(b, c))
local t2 = s0 + maj
local s1 = bxor(rrotate(e, 6), rrotate(e, 11), rrotate(e, 25))
local ch = bxor(band(e, f), band(bnot(e), g))
local t1 = h + s1 + ch + primes[i] + digest[i]
h, g, f, e, d, c, b, a = g, f, e, d + t1, c, b, a, t1 + t2
end
hash[1] = band(hash[1] + a)
hash[2] = band(hash[2] + b)
hash[3] = band(hash[3] + c)
hash[4] = band(hash[4] + d)
hash[5] = band(hash[5] + e)
hash[6] = band(hash[6] + f)
hash[7] = band(hash[7] + g)
hash[8] = band(hash[8] + h)
end
local function sha256(msg: string): string
local extra = 64 - ((#msg + 9) % 64)
local len = toBytes(8 * #msg, 8)
msg ..= "\128" .. string.rep("\0", extra) .. len
assert(#msg % 64 == 0)
local hash = {
0x6a09e667,
0xbb67ae85,
0x3c6ef372,
0xa54ff53a,
0x510e527f,
0x9b05688c,
0x1f83d9ab,
0x5be0cd19,
}
local digest = {}
for i = 1, #msg, 64 do
digestBlock(msg, i, hash, digest)
end
local result = ""
for i = 1, 8 do
local value = hash[i]
result = result .. toBytes(value, 4)
end
return toHex(result)
end
------------------------------------------------------------------------------------------------------------------------------------------------------------------------
-- Upload Animations
------------------------------------------------------------------------------------------------------------------------------------------------------------------------
local StudioAssetService = game:GetService("StudioAssetService")
local ContentProvider = game:GetService("ContentProvider")
local HttpService = game:GetService("HttpService")
local FLIP = CFrame.Angles(0, math.pi, 0)
type Params = { [string]: any }
local function makeQueryString(data: Params)
-- NOTE - This function can be used to create a query string of parameters
-- at the end of url query, or create a application/form-url-encoded post body string
local params = {}
-- NOTE - Arrays are handled, but generally data is expected to be flat.
for key, value in data do
if value ~= nil then -- for optional params
if type(value) == "table" then
for i = 1, #value do
local str = HttpService:UrlEncode(value[i])
table.insert(params, key .. "=" .. str)
end
else
local str = HttpService:UrlEncode(value)
table.insert(params, key .. "=" .. str)
end
end
end
return table.concat(params, "&")
end
local function uploadAnimation(anim: Animation)
local loop: boolean = not not (anim:GetAttribute("Loop"))
local flip: boolean = not not (anim:GetAttribute("Flip"))
local kfs: KeyframeSequence = game.ServerStorage.AnimSaves:FindFirstChild(anim.Name)
local startFrame: number = tonumber(anim:GetAttribute("StartFrame") or nil) or -1
if kfs and kfs:IsA("KeyframeSequence") then
kfs = kfs:Clone()
else
return
end
local id = tonumber(anim.AnimationId:match("%d+$"))
local params: Params = {}
local uploadUrl: string
if id then
uploadUrl = "ide/publish/uploadexistinganimation?"
params.assetID = id
else
uploadUrl = "ide/publish/uploadnewanimation?"
params.allowComments = "True"
params.assetTypeName = "Animation"
params.ispublic = "False"
params.isGamesAsset = "False"
params.description = ""
params.name = anim.Name
params.groupId = ""
params.AllID = 1
end
local baseUrl = ContentProvider.BaseUrl
uploadUrl = baseUrl .. uploadUrl .. makeQueryString(params)
if not loop then
local endAt = kfs:FindFirstChild("End")
if endAt and endAt:IsA("Keyframe") then
local stall = endAt:Clone()
stall.Name = "Freeze"
stall.Parent = kfs
stall.Time = 9999
end
end
if startFrame then
local timeDiff = startFrame / 30
for i, kf in kfs:GetChildren() do
if kf:IsA("Keyframe") then
kf.Time -= timeDiff
if kf.Time < 0 then
kf:Destroy()
end
end
end
end
if flip then
for i, kf in kfs:GetChildren() do
if kf:IsA("Keyframe") then
local lowerTorso = kf:FindFirstChild("LowerTorso", true)
if lowerTorso and lowerTorso:IsA("Pose") then
local cf = lowerTorso.CFrame
lowerTorso.CFrame = FLIP * cf
end
end
end
end
local tmp = Instance.new("Folder")
tmp.Parent = workspace
tmp.Name = "Upload"
kfs.Parent = tmp
local body = StudioAssetService:SerializeInstances(tmp:GetChildren())
local newHash = sha256(body)
tmp:Destroy()
if anim:GetAttribute("UploadHash") ~= newHash then
warn("Uploading Animation", anim.Name .. "...")
local success, response = pcall(function()
print("POST", uploadUrl)
return game:HttpPostAsync(uploadUrl, body, "application/octet-stream")
end)
if success then
anim.AnimationId = "rbxassetid://" .. response
anim:SetAttribute("UploadHash", newHash)
print("\tSuccess!")
else
print("\tError uploading", anim.Name .. ": ", response)
end
else
warn(anim.Name, "is up to date!")
end
end
for i, anim in game.ReplicatedFirst.SM64.Assets.Animations:GetChildren() do
if anim:IsA("Animation") then
uploadAnimation(anim)
end
end