PKHeX/PKHeX.WinForms/Misc/QR.cs

202 lines
8 KiB
C#
Raw Normal View History

using System;
using System.Drawing;
2016-11-17 03:04:34 +00:00
using System.Linq;
using System.Text;
using System.Text.RegularExpressions;
using System.Web;
using System.Windows.Forms;
using PKHeX.Core;
using QRCoder;
using static PKHeX.Core.MessageStrings;
namespace PKHeX.WinForms
{
public partial class QR : Form
{
private readonly PKM pkm;
private readonly Image icon;
private Image qr;
2016-11-17 03:47:41 +00:00
private readonly string[] Lines;
private string extraText;
2016-11-17 03:47:41 +00:00
public QR(Image qr, Image icon, PKM pk, params string[] lines)
2015-02-26 07:12:38 +00:00
{
InitializeComponent();
2016-11-17 03:47:41 +00:00
pkm = pk;
// Layer on Text
const int stretch = 50;
Height += stretch;
if (pkm?.Format == 7)
2016-11-17 03:47:41 +00:00
Height += 40;
this.qr = qr;
this.icon = icon;
Lines = lines;
2016-11-17 03:47:41 +00:00
2018-11-23 07:46:55 +00:00
if (pkm?.Format == 7 && pkm is PK7)
2019-02-02 18:19:41 +00:00
UpdateBoxSlotCopies(null, EventArgs.Empty);
2016-11-17 03:47:41 +00:00
else
RefreshImage();
}
private void RefreshImage()
2016-11-17 03:47:41 +00:00
{
Font font = !Main.Unicode ? Font : FontUtil.GetPKXFont((float)8.25);
2015-02-26 07:12:38 +00:00
Image preview = new Bitmap(45, 45);
using (Graphics gfx = Graphics.FromImage(preview))
{
gfx.FillRectangle(new SolidBrush(Color.White), 0, 0, preview.Width, preview.Height);
int x = (preview.Width / 2) - (icon.Width / 2);
int y = (preview.Height / 2) - (icon.Height / 2);
gfx.DrawImage(icon, x, y);
2015-02-26 07:12:38 +00:00
}
2015-03-14 02:59:51 +00:00
// Layer on Preview Image
Image pic;
{
int x = (qr.Width / 2) - (preview.Width / 2);
int y = (qr.Height / 2) - (preview.Height / 2);
pic = ImageUtil.LayerImage(qr, preview, x, y);
}
2016-11-17 03:47:41 +00:00
2015-03-14 02:59:51 +00:00
Image newpic = new Bitmap(PB_QR.Width, PB_QR.Height);
using (Graphics g = Graphics.FromImage(newpic))
{
g.FillRectangle(new SolidBrush(Color.White), 0, 0, newpic.Width, newpic.Height);
g.DrawImage(pic, 0, 0);
g.DrawString(GetLine(0), font, Brushes.Black, new PointF(18, qr.Height - 5));
g.DrawString(GetLine(1), font, Brushes.Black, new PointF(18, qr.Height + 8));
g.DrawString(GetLine(2).Replace(Environment.NewLine, "/").Replace("//", " ").Replace(":/", ": "), font, Brushes.Black, new PointF(18, qr.Height + 20));
g.DrawString(GetLine(3) + extraText, font, Brushes.Black, new PointF(18, qr.Height + 32));
}
2015-03-14 02:59:51 +00:00
PB_QR.BackgroundImage = newpic;
}
2016-11-17 03:47:41 +00:00
private string GetLine(int line) => Lines.Length <= line ? string.Empty : Lines[line];
2015-02-25 04:42:05 +00:00
private void PB_QR_Click(object sender, EventArgs e)
{
if (DialogResult.Yes != WinFormsUtil.Prompt(MessageBoxButtons.YesNo, MsgQRClipboardImage)) return;
2015-02-25 04:42:05 +00:00
try { Clipboard.SetImage(PB_QR.BackgroundImage); }
catch { WinFormsUtil.Alert(MsgQRClipboardFail); }
2015-02-25 04:42:05 +00:00
}
2018-05-12 15:13:39 +00:00
// QR Utility
private const string QR6PathBad = "null/#"; // prefix to prevent URL from loading
private const string QR6Path = "http://lunarcookies.github.io/b1s1.html#";
private const string DecodeAPI = "http://api.qrserver.com/v1/read-qr-code/?fileurl=";
private const int QRSize = 365;
private static readonly string EncodeAPI = $"http://chart.apis.google.com/chart?chs={QRSize}x{QRSize}&cht=qr&chl=";
internal static byte[] GetQRData(string address)
{
// Fetch data from QR code...
try { if (address.Length < 4 || !address.StartsWith("http")) { WinFormsUtil.Alert(MsgQRUrlFailPath, address); return Array.Empty<byte>(); } }
catch { WinFormsUtil.Alert(MsgQRUrlFailPath, address); return Array.Empty<byte>(); }
string webURL = DecodeAPI + HttpUtility.UrlEncode(address);
string data;
try
{
data = NetUtil.GetStringFromURL(webURL);
if (data.Contains("could not find")) { WinFormsUtil.Alert(MsgQRUrlFailImage); return Array.Empty<byte>(); }
if (data.Contains("filetype not supported")) { WinFormsUtil.Alert(MsgQRUrlFailType, address); return Array.Empty<byte>(); }
}
catch { WinFormsUtil.Alert(MsgQRUrlFailConnection); return Array.Empty<byte>(); }
// Quickly convert the json response to a data string
try { return DecodeQRJson(data); }
catch (Exception e) { WinFormsUtil.Alert(MsgQRUrlFailConvert, e.Message); return Array.Empty<byte>(); }
}
private static byte[] DecodeQRJson(string data)
{
const string cap = "\",\"error\":null}]}]";
const string intro = "[{\"type\":\"qrcode\",\"symbol\":[{\"seq\":0,\"data\":\"";
const string qrcode = "nQR-Code:";
if (!data.StartsWith(intro))
throw new FormatException();
string pkstr = data.Substring(intro.Length);
if (pkstr.Contains(qrcode)) // Remove multiple QR codes in same image
pkstr = pkstr.Substring(0, pkstr.IndexOf(qrcode, StringComparison.Ordinal));
pkstr = pkstr.Substring(0, pkstr.IndexOf(cap, StringComparison.Ordinal)); // Trim outro
if (!pkstr.StartsWith("http") && !pkstr.StartsWith(QR6PathBad)) // G7
{
string fstr = Regex.Unescape(pkstr);
byte[] raw = Encoding.Unicode.GetBytes(fstr);
// Remove 00 interstitials and retrieve from offset 0x30, take PK7 Stored Size (always)
return raw.Where((_, i) => i % 2 == 0).Skip(0x30).Take(0xE8).ToArray();
}
// All except G7
pkstr = pkstr.Substring(pkstr.IndexOf('#') + 1); // Trim URL
pkstr = pkstr.Replace("\\", string.Empty); // Rectify response
return Convert.FromBase64String(pkstr);
}
internal static Image GetQRImage(byte[] data, string server)
{
string qrdata = Convert.ToBase64String(data);
string message = server + qrdata;
string webURL = EncodeAPI + HttpUtility.UrlEncode(message);
try
{
return NetUtil.GetImageFromURL(webURL);
}
catch
{
if (DialogResult.Yes != WinFormsUtil.Prompt(MessageBoxButtons.YesNo, MsgQRUrlFailConnection, MsgQRClipboardUrl))
return null;
try { Clipboard.SetText(webURL); }
catch { WinFormsUtil.Alert(MsgClipboardFailWrite); }
}
return null;
}
2016-11-17 03:47:41 +00:00
private void UpdateBoxSlotCopies(object sender, EventArgs e)
2016-11-17 03:47:41 +00:00
{
if (pkm == null || pkm.Format != 7)
throw new ArgumentException("Can't update QR7 if pkm isn't a PK7!");
var box = (int) NUD_Box.Value - 1;
var slot = (int) NUD_Slot.Value - 1;
var copies = (int) NUD_Copies.Value;
var new_qr = GenerateQRCode7((PK7)pkm, box, slot, copies);
2016-11-17 03:47:41 +00:00
qr = new_qr;
SuspendLayout();
extraText = $" (Box {box+1}, Slot {slot+1}, {copies} cop{(copies > 1 ? "ies" : "y")})";
2016-11-17 03:47:41 +00:00
RefreshImage();
ResumeLayout();
2016-11-17 03:47:41 +00:00
}
// QR7 Utility
public static Image GenerateQRCode7(PK7 pk7, int box = 0, int slot = 0, int num_copies = 1)
{
byte[] data = QR7.GenerateQRData(pk7, box, slot, num_copies);
return GenerateQRCode(data, ppm: 4);
}
private static Image GenerateQRCode(byte[] data, int ppm = 4)
{
using (var generator = new QRCodeGenerator())
using (var qr_data = generator.CreateQRCode(data))
using (var qr_code = new QRCode(qr_data))
return qr_code.GetGraphic(ppm);
}
public static string GetQRServer(int format)
{
switch (format)
{
case 6: return QR6Path;
default: return QR6PathBad;
}
}
}
}