summaryrefslogtreecommitdiffstats
path: root/modules/libjar/nsJARURI.h
blob: 7be600f0216ac5c209035166cb72939c3bfb7db6 (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
146
147
148
149
150
151
152
153
154
155
156
/* -*- Mode: c++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef nsJARURI_h__
#define nsJARURI_h__

#include "nsIJARURI.h"
#include "nsISerializable.h"
#include "nsCOMPtr.h"
#include "nsString.h"
#include "nsINestedURI.h"
#include "nsIURIMutator.h"

#define NS_THIS_JARURI_IMPL_CID                      \
  { /* 9a55f629-730b-4d08-b75b-fa7d9570a691 */       \
    0x9a55f629, 0x730b, 0x4d08, {                    \
      0xb7, 0x5b, 0xfa, 0x7d, 0x95, 0x70, 0xa6, 0x91 \
    }                                                \
  }

#define NS_JARURI_CID                                \
  { /* 245abae2-b947-4ded-a46d-9829d3cca462 */       \
    0x245abae2, 0xb947, 0x4ded, {                    \
      0xa4, 0x6d, 0x98, 0x29, 0xd3, 0xcc, 0xa4, 0x62 \
    }                                                \
  }

#define NS_JARURIMUTATOR_CID                         \
  { /* 19d9161b-a2a9-4518-b2c9-fcb8296d6dcd */       \
    0x19d9161b, 0xa2a9, 0x4518, {                    \
      0xb2, 0xc9, 0xfc, 0xb8, 0x29, 0x6d, 0x6d, 0xcd \
    }                                                \
  }

class nsJARURI final : public nsIJARURI,
                       public nsISerializable,
                       public nsINestedURI {
 public:
  NS_DECL_THREADSAFE_ISUPPORTS
  NS_DECL_NSIURI
  NS_DECL_NSIURL
  NS_DECL_NSIJARURI
  NS_DECL_NSISERIALIZABLE
  NS_DECL_NSINESTEDURI

  NS_DECLARE_STATIC_IID_ACCESSOR(NS_THIS_JARURI_IMPL_CID)

  // nsJARURI
  nsresult FormatSpec(const nsACString& entryPath, nsACString& result,
                      bool aIncludeScheme = true);
  nsresult CreateEntryURL(const nsACString& entryFilename, const char* charset,
                          nsIURL** url);

 protected:
  nsJARURI();
  virtual ~nsJARURI();
  nsresult SetJAREntry(const nsACString& entryPath);
  nsresult Init(const char* charsetHint);
  nsresult SetSpecWithBase(const nsACString& aSpec, nsIURI* aBaseURL);

  // enum used in a few places to specify how .ref attribute should be handled
  enum RefHandlingEnum { eIgnoreRef, eHonorRef, eReplaceRef };

  // Helper to share code between Equals methods.
  virtual nsresult EqualsInternal(nsIURI* other,
                                  RefHandlingEnum refHandlingMode,
                                  bool* result);

  nsCOMPtr<nsIURI> mJARFile;
  // mJarEntry stored as a URL so that we can easily access things
  // like extensions, refs, etc.
  nsCOMPtr<nsIURL> mJAREntry;
  nsCString mCharsetHint;

 private:
  nsresult Clone(nsIURI** aURI);
  nsresult SetSpecInternal(const nsACString& input);
  nsresult SetScheme(const nsACString& input);
  nsresult SetUserPass(const nsACString& input);
  nsresult SetUsername(const nsACString& input);
  nsresult SetPassword(const nsACString& input);
  nsresult SetHostPort(const nsACString& aValue);
  nsresult SetHost(const nsACString& input);
  nsresult SetPort(int32_t port);
  nsresult SetPathQueryRef(const nsACString& input);
  nsresult SetRef(const nsACString& input);
  nsresult SetFilePath(const nsACString& input);
  nsresult SetQuery(const nsACString& input);
  nsresult SetQueryWithEncoding(const nsACString& input,
                                const mozilla::Encoding* encoding);
  bool Deserialize(const mozilla::ipc::URIParams&);
  nsresult ReadPrivate(nsIObjectInputStream* aStream);

  nsresult SetFileNameInternal(const nsACString& fileName);
  nsresult SetFileBaseNameInternal(const nsACString& fileBaseName);
  nsresult SetFileExtensionInternal(const nsACString& fileExtension);

 public:
  class Mutator final : public nsIURIMutator,
                        public BaseURIMutator<nsJARURI>,
                        public nsIURLMutator,
                        public nsISerializable,
                        public nsIJARURIMutator {
    NS_DECL_ISUPPORTS
    NS_FORWARD_SAFE_NSIURISETTERS_RET(mURI)
    NS_DEFINE_NSIMUTATOR_COMMON
    NS_DECL_NSIURLMUTATOR

    NS_IMETHOD
    Write(nsIObjectOutputStream* aOutputStream) override {
      return NS_ERROR_NOT_IMPLEMENTED;
    }

    [[nodiscard]] NS_IMETHOD Read(nsIObjectInputStream* aStream) override {
      return InitFromInputStream(aStream);
    }

    NS_IMETHOD
    SetSpecBaseCharset(const nsACString& aSpec, nsIURI* aBaseURI,
                       const char* aCharset) override {
      RefPtr<nsJARURI> uri;
      if (mURI) {
        mURI.swap(uri);
      } else {
        uri = new nsJARURI();
      }

      nsresult rv = uri->Init(aCharset);
      NS_ENSURE_SUCCESS(rv, rv);

      rv = uri->SetSpecWithBase(aSpec, aBaseURI);
      if (NS_FAILED(rv)) {
        return rv;
      }

      mURI.swap(uri);
      return NS_OK;
    }

    explicit Mutator() {}

   private:
    virtual ~Mutator() {}

    friend class nsJARURI;
  };

  friend BaseURIMutator<nsJARURI>;
};

NS_DEFINE_STATIC_IID_ACCESSOR(nsJARURI, NS_THIS_JARURI_IMPL_CID)

#endif  // nsJARURI_h__